3
F�`Q�� � @ s` d Z ddlZddlZddlZddlZddlZddlZddlZddlZej d dkZ
e
rxe Z e Z
eZeZeZddlmZ n ejZejZeZddlmZ e
r�ej d dkr�ddlmZ n ddlmZ dZ dZ dZ dZ dZ d Z!d
Z"d Z#ej$d �Z%d
d� Z&G dd� de'�Z(dd� Z)dd� Z*d"dd�Z+dd� Z,G dd� de�Z-dd � Z.d d � Z/d d � Z0e1d!k�r\e0� dS )#z0.7.2� N� )�
HTMLParser� � )�escape�
� � � z \[[0-9;]*mc C s, | j d�}t|�}tdd� |D ��}||fS )N�
c S s g | ] }t |��qS � )�_str_block_width)�.0�liner r �!/usr/lib/python3.6/prettytable.py�
<listcomp>O s z _get_size.<locals>.<listcomp>)�split�len�max)�text�linesZheight�widthr r r � _get_sizeL s
r c @ sB e Zd Zd�dd�Zdd� Zdd� Zdd � Zd
d � Zer@d d
� Z ndd
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z e ee �Z d5d6� Z d7d8� Z e e e �Z!d9d:� Z"d;d<� Z#e e"e#�Z$d=d>� Z%d?d@� Z&e e%e&�Z'dAdB� Z(dCdD� Z)e e(e)�Z*dEdF� Z+dGdH� Z,e e+e,�Z-dIdJ� Z.dKdL� Z/e e.e/�Z0dMdN� Z1dOdP� Z2e e1e2�Z3dQdR� Z4dSdT� Z5e e4e5�Z6dUdV� Z7dWdX� Z8e e7e8�Z9dYdZ� Z:d[d\� Z;e e:e;�Z<d]d^� Z=d_d`� Z>e e=e>�Z?dadb� Z@dcdd� ZAe e@eA�ZBdedf� ZCdgdh� ZDe eCeD�ZEdidj� ZFdkdl� ZGe eFeG�ZHdmdn� ZIdodp� ZJe eIeJ�ZKdqdr� ZLdsdt� ZMe eLeM�ZNdudv� ZOdwdx� ZPe eOeP�ZQdydz� ZRd{d|� ZSe eReS�ZTd}d~� ZUdd�� ZVe eUeV�ZWd�d�� ZXd�d�� ZYe eXeY�ZZd�d�� Z[d�d�� Z\e e[e\�Z]d�d�� Z^d�d�� Z_e e^e_�Z`d�d�� Zad�d�� Zbe eaeb�Zcd�d�� Zdd�d�� Zee edee�Zfd�d�� Zgd�d�� Zhe egeh�Zid�d�� Zjd�d�� Zkd�d�� Zld�d�� Zmd�d�� Znd�d�� Zod�d�� Zpd�d�� Zqd�d�d��Zrd�d�� Zsd�d�� Ztd�d�� Zud�d�� Zvd�d�� Zwd�d�� Zxd�d�� Zyd�d�� Zzd�d�� Z{d�d�� Z|d�d Z}d�dĄ Z~d�dƄ Zd�dȄ Z�d�dʄ Z�d�d̄ Z�dS )�� PrettyTableNc K sZ |j dd�| _g | _i | _i | _i | _g | _|r8|| _ng | _dj � | _
| j
j dj � � | j
j dj � � x0| j
D ]&}||kr�| j ||| � qpd||< qpW |d p�d| _
|d p�d| _|d
p�d| _|d d'kr�|d | _nd | _|d p�d| _|d d(k�r
|d | _nd | _|d �p t| _|d �p,t| _|d �p<d| _|d d)k�rZ|d | _nd
| _|d �ppdd� | _|d �p�i | _|d �p�i | _|d �p�d| _ |d �p�d| _ |d �p�d| _ |d �p�| j d �| _ |d �p�| j d �| _!|d! �p| j d"�| _"|d# d*k�r |d# | _#nd | _#|d$ �p2d
| _$|d% �pBd
| _%|d& �pRi | _&dS )+a� Return a new PrettyTable instance
Arguments:
encoding - Unicode encoding scheme used to decode any encoded input
field_names - list or tuple of field names
fields - list or tuple of field names to include in displays
start - index of first data row to include in output
end - index of last data row to include in output PLUS ONE (list slice style)
header - print a header showing field names (True or False)
header_style - stylisation to apply to field names in header ("cap", "title", "upper", "lower" or None)
border - print a border around the table (True or False)
hrules - controls printing of horizontal rules after rows. Allowed values: FRAME, HEADER, ALL, NONE
vrules - controls printing of vertical rules between columns. Allowed values: FRAME, ALL, NONE
int_format - controls formatting of integer data
float_format - controls formatting of floating point data
padding_width - number of spaces on either side of column data (only used if left and right paddings are None)
left_padding_width - number of spaces on left hand side of column data
right_padding_width - number of spaces on right hand side of column data
vertical_char - single character string used to draw vertical lines
horizontal_char - single character string used to draw horizontal lines
junction_char - single character string used to draw line junctions
sortby - name of field to sort rows by
sort_key - sorting key function, applied to data points before sorting
valign - default valign for each row (None, "t", "m" or "b")
reversesort - True or False to sort in descending or ascending order�encodingzUTF-8zZstart end fields header border sortby reversesort sort_key attributes format hrules vruleszLint_format float_format padding_width left_padding_width right_padding_widthzQvertical_char horizontal_char junction_char header_style valign xhtml print_emptyN�startr �end�fields�headerTF� header_style�border�hrules�vrules�sortby� reversesort�sort_keyc S s | S )Nr )�xr r r �<lambda>� s z&PrettyTable.__init__.<locals>.<lambda>�
int_format� float_format�
padding_widthr �left_padding_width�right_padding_width�
vertical_char�|�horizontal_char�-�
junction_char�+� print_empty�format�xhtml�
attributes)TF)TF)TF)TF)'�getr � _field_names�_align�_valign�
_max_width�_rows� field_names�_widthsr �_options�extend�_validate_option�_start�_end�_fields�_header�
_header_style�_border�FRAME�_hrules�ALL�_vrules�_sortby� _reversesort� _sort_key� _int_format�
_float_format�_padding_width�_left_padding_width�_right_padding_width�_unicode�_vertical_char�_horizontal_char�_junction_char� _print_empty�_formatZ_xhtml� _attributes)�selfr= �kwargs�optionr r r �__init__T s^
zPrettyTable.__init__c C s. t |t�st|�}t |t�s*t|| jd�}|S )N�strict)�
isinstance�
basestring�str�unicoder )r[ �valuer r r rT � s
zPrettyTable._unicodec C s� |t |� }|dkr ||d S |dkr4|d | S |d r�t |�d rh|d d | |d d d S |d d d | |d d S n |d d | |d d S d S )N�l� �rr r )r
)r[ r r �alignZexcessr r r �_justify� s "zPrettyTable._justifyc C sP |dkrt | j�S |dkrD| jr*t | j�S | jr>t | jd �S dS nt|��d S )N�rowcountZcolcountr )r r< r8 �AttributeError)r[ �namer r r � __getattr__� s
zPrettyTable.__getattr__c C s� t � }| j|_x(| jD ] }t|d| t| d| �� qW t|dt| d�� t|t�rtxJ| j| D ]}|j|� q`W n,t|t �r�|j| j| � nt
dt |� ��|S )N�_r9 z0Index %s is invalid, must be an integer or slice) r r= r? �setattr�getattrr` �slicer< �add_row�int� Exceptionrb )r[ �index�new�attr�rowr r r � __getitem__� s
zPrettyTable.__getitem__c C s | j � S )N)� __unicode__)r[ r r r �__str__� s zPrettyTable.__str__c C s | j � j| j�S )N)rz �encoder )r[ r r r r{ � s c C s | j � S )N)�
get_string)r[ r r r rz � s zPrettyTable.__unicode__c C s4 |dkr| j |� �n|dkr.| j||� �n|d krD| j||� n�|d
krZ| j||� n�|d krp| j||� n�|d kr�| j||� n�|d
kr�| j||� n�|d kr�| j||� n~|dkr�| j|� nj|dkr�| j ||� nT|dk� r�| j
||� n<|d k�r | j ||� n$|dk�r$| j ||� n t
d| ��d S ) Nr= r r � max_widthr* r+ r, r4 r# r% r! r" r r r r$ r5 r3 r r( r) r- r/ r1 r6 zUnrecognised option: %s!)r r r~ r* r+ r, r4 )r r r$ r5 r3 )r- r/ r1 )�_validate_field_names�_validate_nonnegative_int�_validate_field_name�_validate_function�_validate_hrules�_validate_vrules�_validate_all_field_names�_validate_true_or_false�_validate_header_style�_validate_int_format�_validate_float_format�_validate_single_char�_validate_attributesrt )r[ r] �valr r r rA s6
z PrettyTable._validate_optionc C s� | j rPyt|�t| j �ks t�W n. tk
rN tdt|�t| j �f ��Y nX | jr�y t|�t| jd �ksrt�W n2 tk
r� tdt|�t| jd �f ��Y nX y t|�tt|��ks�t�W n tk
r� td��Y nX d S )NzJField name list has incorrect number of values, (actual) %d!=%d (expected)r zField names must be unique!)r8 r �AssertionErrorrt r< �set)r[ r� r r r r s $ z!PrettyTable._validate_field_namesc C s2 y|dkst �W n t k
r, td��Y nX d S )N�cap�title�upper�lowerz;Invalid header style, use cap, title, upper, lower or None!)r� r� r� r� N)r� rt )r[ r� r r r r� 0 s z"PrettyTable._validate_header_stylec C s6 y|dkst �W n t k
r0 td| ��Y nX d S )Nre �crg z'Alignment %s is invalid, use l, c or r!)re r� rg )r� rt )r[ r� r r r �_validate_align6 s zPrettyTable._validate_alignc C s6 y|dkst �W n t k
r0 td| ��Y nX d S )N�t�m�bz-Alignment %s is invalid, use t, m, b or None!)r� r� r� N)r� rt )r[ r� r r r �_validate_valign< s z PrettyTable._validate_valignc C sD yt |�dkst�W n* tk
r> td|| j|�f ��Y nX d S )Nr zInvalid value for %s: %s!)rs r� rt rT )r[ rl r� r r r r� B s z%PrettyTable._validate_nonnegative_intc C s6 y|dkst �W n t k
r0 td| ��Y nX d S )NTFz-Invalid value for %s! Must be True or False.)TF)r� rt )r[ rl r� r r r r� H s z#PrettyTable._validate_true_or_falsec C sV |dkr d S y$t |�ttfks"t�|j� s.t�W n tk
rP td| ��Y nX d S )N� z8Invalid value for %s! Must be an integer format string.)�typerb rc r� �isdigitrt )r[ rl r� r r r r� N s z PrettyTable._validate_int_formatc C s� |dkr d S yvt |�ttfks"t�d|ks.t�|jd�}t|�dksHt�|d dksd|d j� sdt�|d dks�|d j� s�t�W n tk
r� td| ��Y nX d S )Nr� �.r r r z5Invalid value for %s! Must be a float format string.)r� rb rc r� r r r� rt )r[ rl r� �bitsr r r r� W s
z"PrettyTable._validate_float_formatc C s8 yt |d�st�W n tk
r2 td| ��Y nX d S )N�__call__z*Invalid value for %s! Must be a function.)�hasattrr� rt )r[ rl r� r r r r� d s z PrettyTable._validate_functionc C s> y|t tttfkst�W n tk
r8 td| ��Y nX d S )Nz:Invalid value for %s! Must be ALL, FRAME, HEADER or NONE.)rJ rH �HEADER�NONEr� rt )r[ rl r� r r r r� j s z PrettyTable._validate_hrulesc C s< y|t ttfkst�W n tk
r6 td| ��Y nX d S )Nz3Invalid value for %s! Must be ALL, FRAME, or NONE.)rJ rH r� r� rt )r[ rl r� r r r r� p s z PrettyTable._validate_vrulesc C s@ y|| j ks|d kst�W n tk
r: td| ��Y nX d S )NzInvalid field name: %s!)r8 r� rt )r[ rl r� r r r r� v s z PrettyTable._validate_field_namec C s@ y x|D ]}| j ||� qW W n tk
r: td��Y nX d S )Nz)fields must be a sequence of field names!)r� r� rt )r[ rl r� r& r r r r� | s
z%PrettyTable._validate_all_field_namesc C s: yt |�dkst�W n tk
r4 td| ��Y nX d S )Nr z4Invalid value for %s! Must be a string of length 1.)r
r� rt )r[ rl r� r r r r� � s z!PrettyTable._validate_single_charc C s4 yt |t�st�W n tk
r. td��Y nX d S )Nz4attributes must be a dictionary of name/value pairs!)r` �dictr� rt )r[ rl r� r r r r� � s z PrettyTable._validate_attributesc C s | j S )N)r8 )r[ r r r �_get_field_names� s z PrettyTable._get_field_namesc s$ � fdd�|D �}� j d|� � jr2� jd d � }|� _� jr�|r�x&t||�D ]\}}� j| � j|< qNW x>|D ]}|� jkrp� jj|� qpW nx� jD ]}d� j|< q�W � jo�|�rx&t||�D ]\}}� j| � j|< q�W x@|D ]}|� jkr� jj|� q�W n x� jD ]}d� j|< �q W d S )Nc s g | ]}� j |��qS r )rT )r r& )r[ r r r � s z0PrettyTable._set_field_names.<locals>.<listcomp>r= r� r� )rA r8 r9 �zip�popr: )r[ r� Z old_namesZold_name�new_name�fieldr )r[ r �_set_field_names� s*
z PrettyTable._set_field_namesc C s | j S )N)r9 )r[ r r r �
_get_align� s zPrettyTable._get_alignc C s( | j |� x| jD ]}|| j|< qW d S )N)r� r8 r9 )r[ r� r� r r r �
_set_align� s
zPrettyTable._set_alignc C s | j S )N)r: )r[ r r r � _get_valign� s zPrettyTable._get_valignc C s( | j |� x| jD ]}|| j|< qW d S )N)r� r8 r: )r[ r� r� r r r � _set_valign� s
zPrettyTable._set_valignc C s | j S )N)r; )r[ r r r �_get_max_width� s zPrettyTable._get_max_widthc C s* | j d|� x| jD ]}|| j|< qW d S )Nr~ )rA r8 r; )r[ r� r� r r r �_set_max_width� s zPrettyTable._set_max_widthc C s | j S )z�List or tuple of field names to include in displays
Arguments:
fields - list or tuple of field names to include in displays)rD )r[ r r r � _get_fields� s zPrettyTable._get_fieldsc C s | j d|� || _d S )Nr )rA rD )r[ r� r r r � _set_fields� s zPrettyTable._set_fieldsc C s | j S )z{Start index of the range of rows to print
Arguments:
start - index of first data row to include in output)rB )r[ r r r �
_get_start� s zPrettyTable._get_startc C s | j d|� || _d S )Nr )rA rB )r[ r� r r r �
_set_start� s zPrettyTable._set_startc C s | j S )z�End index of the range of rows to print
Arguments:
end - index of last data row to include in output PLUS ONE (list slice style))rC )r[ r r r �_get_end� s zPrettyTable._get_endc C s | j d|� || _d S )Nr )rA rC )r[ r� r r r �_set_end� s zPrettyTable._set_endc C s | j S )z_Name of field by which to sort rows
Arguments:
sortby - field name to sort by)rL )r[ r r r � _get_sortby� s zPrettyTable._get_sortbyc C s | j d|� || _d S )Nr# )rA rL )r[ r� r r r � _set_sortby� s zPrettyTable._set_sortbyc C s | j S )z�Controls direction of sorting (ascending vs descending)
Arguments:
reveresort - set to True to sort by descending order, or False to sort by ascending order)rM )r[ r r r �_get_reversesort� s z PrettyTable._get_reversesortc C s | j d|� || _d S )Nr$ )rA rM )r[ r� r r r �_set_reversesort s z PrettyTable._set_reversesortc C s | j S )z�Sorting key function, applied to data points before sorting
Arguments:
sort_key - a function which takes one argument and returns something to be sorted)rN )r[ r r r �
_get_sort_key s zPrettyTable._get_sort_keyc C s | j d|� || _d S )Nr% )rA rN )r[ r� r r r �
_set_sort_key s zPrettyTable._set_sort_keyc C s | j S )z�Controls printing of table header with field names
Arguments:
header - print a header showing field names (True or False))rE )r[ r r r � _get_header s zPrettyTable._get_headerc C s | j d|� || _d S )Nr )rA rE )r[ r� r r r � _set_header s zPrettyTable._set_headerc C s | j S )z�Controls stylisation applied to field names in header
Arguments:
header_style - stylisation to apply to field names in header ("cap", "title", "upper", "lower" or None))rF )r[ r r r �_get_header_style! s z PrettyTable._get_header_stylec C s | j |� || _d S )N)r� rF )r[ r� r r r �_set_header_style( s
z PrettyTable._set_header_stylec C s | j S )z~Controls printing of border around table
Arguments:
border - print a border around the table (True or False))rG )r[ r r r � _get_border- s zPrettyTable._get_borderc C s | j d|� || _d S )Nr )rA rG )r[ r� r r r � _set_border4 s zPrettyTable._set_borderc C s | j S )z�Controls printing of horizontal rules after rows
Arguments:
hrules - horizontal rules style. Allowed values: FRAME, ALL, HEADER, NONE)rI )r[ r r r � _get_hrules9 s zPrettyTable._get_hrulesc C s | j d|� || _d S )Nr! )rA rI )r[ r� r r r � _set_hrules@ s zPrettyTable._set_hrulesc C s | j S )z�Controls printing of vertical rules between columns
Arguments:
vrules - vertical rules style. Allowed values: FRAME, ALL, NONE)rK )r[ r r r � _get_vrulesE s zPrettyTable._get_vrulesc C s | j d|� || _d S )Nr" )rA rK )r[ r� r r r � _set_vrulesL s zPrettyTable._set_vrulesc C s | j S )zbControls formatting of integer data
Arguments:
int_format - integer format string)rO )r[ r r r �_get_int_formatQ s zPrettyTable._get_int_formatc C s x| j D ]}|| j|< qW d S )N)r8 rO )r[ r� r� r r r �_set_int_formatW s zPrettyTable._set_int_formatc C s | j S )zrControls formatting of floating point data
Arguments:
float_format - floating point format string)rP )r[ r r r �_get_float_format] s z PrettyTable._get_float_formatc C s x| j D ]}|| j|< qW d S )N)r8 rP )r[ r� r� r r r �_set_float_formatc s z PrettyTable._set_float_formatc C s | j S )z�The number of empty spaces between a column's edge and its content
Arguments:
padding_width - number of spaces, must be a positive integer)rQ )r[ r r r �_get_padding_widthi s z PrettyTable._get_padding_widthc C s | j d|� || _d S )Nr* )rA rQ )r[ r� r r r �_set_padding_widthp s z PrettyTable._set_padding_widthc C s | j S )z�The number of empty spaces between a column's left edge and its content
Arguments:
left_padding - number of spaces, must be a positive integer)rR )r[ r r r �_get_left_padding_widthu s z#PrettyTable._get_left_padding_widthc C s | j d|� || _d S )Nr+ )rA rR )r[ r� r r r �_set_left_padding_width| s z#PrettyTable._set_left_padding_widthc C s | j S )z�The number of empty spaces between a column's right edge and its content
Arguments:
right_padding - number of spaces, must be a positive integer)rS )r[ r r r �_get_right_padding_width� s z$PrettyTable._get_right_padding_widthc C s | j d|� || _d S )Nr, )rA rS )r[ r� r r r �_set_right_padding_width� s z$PrettyTable._set_right_padding_widthc C s | j S )z�The charcter used when printing table borders to draw vertical lines
Arguments:
vertical_char - single character string used to draw vertical lines)rU )r[ r r r �_get_vertical_char� s z PrettyTable._get_vertical_charc C s | j |�}| jd|� || _d S )Nr- )rT rA rU )r[ r� r r r �_set_vertical_char� s
z PrettyTable._set_vertical_charc C s | j S )z�The charcter used when printing table borders to draw horizontal lines
Arguments:
horizontal_char - single character string used to draw horizontal lines)rV )r[ r r r �_get_horizontal_char� s z PrettyTable._get_horizontal_charc C s | j |�}| jd|� || _d S )Nr/ )rT rA rV )r[ r� r r r �_set_horizontal_char� s
z PrettyTable._set_horizontal_charc C s | j S )z�The charcter used when printing table borders to draw line junctions
Arguments:
junction_char - single character string used to draw line junctions)rW )r[ r r r �_get_junction_char� s z PrettyTable._get_junction_charc C s | j |�}| jd|� || _d S )Nr- )rT rA rW )r[ r� r r r �_set_junction_char� s
z PrettyTable._set_junction_charc C s | j S )z~Controls whether or not HTML tables are formatted to match styling options
Arguments:
format - True or False)rY )r[ r r r � _get_format� s zPrettyTable._get_formatc C s | j d|� || _d S )Nr4 )rA rY )r[ r� r r r � _set_format� s zPrettyTable._set_formatc C s | j S )z�Controls whether or not empty tables produce a header and frame or just an empty string
Arguments:
print_empty - True or False)rX )r[ r r r �_get_print_empty� s z PrettyTable._get_print_emptyc C s | j d|� || _d S )Nr3 )rA rX )r[ r� r r r �_set_print_empty� s z PrettyTable._set_print_emptyc C s | j S )z�A dictionary of HTML attribute name/value pairs to be included in the <table> tag when printing HTML
Arguments:
attributes - dictionary of attributes)rZ )r[ r r r �_get_attributes� s zPrettyTable._get_attributesc C s | j d|� || _d S )Nr6 )rA rZ )r[ r� r r r �_set_attributes� s zPrettyTable._set_attributesc C sP i }xF| j D ]<}||kr6| j||| � || ||< q t| d| �||< q W |S )Nrn )r? rA rp )r[ r\ �optionsr] r r r � _get_options� s zPrettyTable._get_optionsc C sT |t kr| j� n>|tkr$| j� n,|tkr6| j� n|tkrH| j� ntd��d S )NzInvalid pre-set style!) �DEFAULT�_set_default_style�MSWORD_FRIENDLY�_set_msword_style�
PLAIN_COLUMNS�_set_columns_style�RANDOM�_set_random_stylert )r[ Zstyler r r � set_style� s
zPrettyTable.set_stylec C s@ d| _ d| _t| _t| _d| _d| _d| _d| _ d| _
d| _ d S )NTr r. r0 r2 ) r r rH rI rJ rK r* r+ r, r- r/ r1 )r[ r r r r� � s z PrettyTable._set_default_stylec C s. d| _ d| _t| _d| _d| _d| _d| _d S )NTr r. )r r r� rI r* r+ r, r- )r[ r r r r� s z PrettyTable._set_msword_stylec C s" d| _ d| _d| _d| _d| _d S )NTFr r � )r r r* r+ r, )r[ r r r r� s
z PrettyTable._set_columns_stylec C s� t jd�| _t jd�| _t jttttf�| _t jtttf�| _ t j
dd�| _ t j
dd�| _ t jd�| _
t jd�| _t jd�| _d S )NTFr � z ~!@#$%^&*()_+|-=\{}[];':",./;<>?)TF)TF)�randomZchoicer r rJ rH r� r� rI rK Zrandintr+ r, r- r/ r1 )r[ r r r r� s z PrettyTable._set_random_stylec C sf | j r2t|�t| j �kr2tdt|�t| j �f ��| j sRdd� tdt|��D �| _| jjt|�� dS )z�Add a row to the table
Arguments:
row - row of data, should be a list with as many elements as the table
has fieldsz>Row has incorrect number of values, (actual) %d!=%d (expected)c S s g | ]}d |d �qS )zField %dr r )r �nr r r r 4 s z'PrettyTable.add_row.<locals>.<listcomp>r N)r8 r rt �ranger= r< �append�list)r[ rx r r r rr ( s
zPrettyTable.add_rowc C s4 |t | j�d kr(td|t | j�f ��| j|= dS )z�Delete a row to the table
Arguments:
row_index - The index of the row you want to delete. Indexing starts at 0.r z4Cant delete row at index %d, table only has %d rows!N)r r< rt )r[ Z row_indexr r r �del_row7 s zPrettyTable.del_rowr� r� c C s� t | j�dt |�fkr�| j|� | j|� | jj|� || j|< || j|< xdtdt |��D ]6}t | j�|d k r|| jjg � | j| j|| � qZW nt dt |�t | j�f ��dS )a� Add a column to the table.
Arguments:
fieldname - name of the field to contain the new column of data
column - column of data, should be a list with as many elements as the
table has rows
align - desired alignment for this column - "l" for left, "c" for centre and "r" for right
valign - desired vertical alignment for new columns - "t" for top, "m" for middle and "b" for bottomr r z2Column length %d does not match number of rows %d!N)
r r< r� r� r8 r� r9 r: r� rt )r[ � fieldname�columnrh �valign�ir r r �
add_columnC s
zPrettyTable.add_columnc C s
g | _ dS )z?Delete all rows from the table but keep the current field namesN)r< )r[ r r r �
clear_rows\ s zPrettyTable.clear_rowsc C s g | _ g | _g | _dS )zWDelete all rows and field names from the table, maintaining nothing but styling optionsN)r< r8 r> )r[ r r r �clearb s zPrettyTable.clearc C s
t j| �S )N)�copy�deepcopy)r[ r r r r� n s zPrettyTable.copyc C sd t |t�r.|| jkr.| jd| j| | �}n,t |t�rZ|| jkrZ| jd| j| | �}| j|�S )Nz%%%sdz%%%sf)r` rs rO rT �floatrP )r[ r� rd r r r �
_format_valueu s
zPrettyTable._format_valuec C s� |d rdd� | j D �}nt| j�dg }xx|D ]p}xjt|�D ]^\}}| j| }|| jkr�t|| tt|�d | j| ��||< q>t|| t|�d �||< q>W q0W || _d S )Nr c S s g | ]}t |�d �qS )r )r )r r� r r r r ~ s z/PrettyTable._compute_widths.<locals>.<listcomp>r ) r8 r r= � enumerater~ r �minr r> )r[ �rowsr� Zwidthsrx ru rd r� r r r �_compute_widths| s
("zPrettyTable._compute_widthsc C sD |d d k r|d }n|d }|d d k r4|d }n|d }||fS )Nr+ r* r, r )r[ r� �lpad�rpadr r r �_get_padding_widths� s
z PrettyTable._get_padding_widthsc sn t j| j|d |d � �}|d rj| jj|d �� � fdd�|D �}|j|d |d d� d d� |D �}|S )
z�Return only those data rows that should be printed, based on slicing and sorting.
Arguments:
options - dictionary of option settings.r r r# c s g | ]}|� g| �qS r r )r rx )� sortindexr r r � s z)PrettyTable._get_rows.<locals>.<listcomp>r$ r% )�reverse�keyc S s g | ]}|d d� �qS )r Nr )r rx r r r r � s )r� r� r< r8 ru �sort)r[ r� r� r )r r � _get_rows� s zPrettyTable._get_rowsc s � fdd�t � j|�D �S )Nc s g | ]\}}� j ||��qS r )r� )r r� rd )r[ r r r � s z+PrettyTable._format_row.<locals>.<listcomp>)r� r8 )r[ rx r� r )r[ r � _format_row� s zPrettyTable._format_rowc s � �fdd�|D �S )Nc s g | ]}�j |� ��qS r )r )r rx )r� r[ r r r � s z,PrettyTable._format_rows.<locals>.<listcomp>r )r[ r� r� r )r� r[ r � _format_rows� s zPrettyTable._format_rowsc K s� | j |�}g }| jdkr0|d s,|d r0dS | j|�}| j||�}| j||� | j|�| _|d rx|j| j|�� n$|d r�|d t t
fkr�|j| j� x |D ]}|j| j ||�� q�W |d r�|d t
kr�|j| j� | j d�j
|�S )a" Return string representation of table in current state.
Arguments:
start - index of first data row to include in output
end - index of last data row to include in output PLUS ONE (list slice style)
fields - names of fields (columns) to include
header - print a header showing field names (True or False)
border - print a border around the table (True or False)
hrules - controls printing of horizontal rules after rows. Allowed values: ALL, FRAME, HEADER, NONE
vrules - controls printing of vertical rules between columns. Allowed values: FRAME, ALL, NONE
int_format - controls formatting of integer data
float_format - controls formatting of floating point data
padding_width - number of spaces on either side of column data (only used if left and right paddings are None)
left_padding_width - number of spaces on left hand side of column data
right_padding_width - number of spaces on right hand side of column data
vertical_char - single character string used to draw vertical lines
horizontal_char - single character string used to draw horizontal lines
junction_char - single character string used to draw line junctions
sortby - name of field to sort rows by
sort_key - sorting key function, applied to data points before sorting
reversesort - True or False to sort in descending or ascending order
print empty - if True, stringify just the header for an empty table, if False return an empty string r r3 r r� r r! r )r� rj r r r� �_stringify_hrule�_hruler� �_stringify_headerrJ rH �_stringify_rowrT �join)r[ r\ r� r r� �formatted_rowsrx r r r r} � s"
zPrettyTable.get_stringc C s |d s dS | j |�\}}|d ttfkr6|d g}n
|d g}| js^|j|d � dj|�S xtt| j| j�D ]b\}}|d r�||d kr�qn|j|| | |d � |d tkr�|j|d � qn|j|d � qnW |d tkr�|j� |j|d � dj|�S )Nr r� r" r1 r/ r ) r� rJ rH r8 r� r r� r> r� )r[ r� r� r� r� r� r r r r r � s(
z PrettyTable._stringify_hrulec C s� g }| j |�\}}|d rj|d ttfkr@|j| j� |jd� |d ttfkr`|j|d � n
|jd� | js�|d ttfkr�|j|d � n
|jd� x�t| j| j�D ]�\}}|d r�||d kr�q�| jdkr�|j � }nD| jd kr�|j
� }n0| jd
k�r|j � }n| jd k�r |j � }n|}|jd| | j
||| j| � d| � |d r�|d tk�rp|j|d � q�|jd� q�W |d �r�|d tk�r�|j� |j|d � |d �r�|d tk�r�|jd� |j| j� d j|�S )
Nr r! r r" r- rf r r� r� r� r� r� )r� rJ rH r� r r8 r� r> rF �
capitalizer� r� r� ri r9 r� r� r )r[ r� r� r� r� r� r r� r r r r sJ
*
z PrettyTable._stringify_headerc C s x�t tdt|��| j|| j�D ]`\}}}}|jd�}g }x.|D ]&} t| �|krZtj| |�} |j | � q>W |}dj
|�}|||< q W d}
x$|D ] } t | �d } | |
kr�| }
q�W g }
| j |�\}}xVtd|
�D ]H}|
j g � |d r�|d t
tfk�r|
| j | j� q�|
| j d� q�W �xFt | j|| j�D �]0\}}}| j| }|jd�}|
t|� }|�r�|dk�r�dgt|d � | dg|t|d � }n(|d
k�r�dg| | }n|dg| }d}x�|D ]�}|d �r�||d k�r�q�|
| j d| | j||| j| � d| � |d �rL|d t
k�r>|
| j | j� n|
| j d� |d7 }�q�W �q*W xJtd|
�D ]<}|d �rl|d tk�rl|
| j� |
| j |d � �qlW |d �r�|d
t
k�r�|
|
d j d� |
|
d j | j� x&td|
�D ]}dj
|
| �|
|< �q�W dj
|
�S )Nr r r r r" rf r� r� r r� r r- r! )r� r� r r8 r> r r
�textwrapZfillr� r r r� rJ rH r- r: rs ri r9 r� r )r[ rx r� ru r� rd r r Z new_linesr Z
row_heightr� �hr� r� r� �yr� ZdHeightre r r r r
: sj *
.
.
zPrettyTable._stringify_rowc K s, | j |�}|d r | j|�}n
| j|�}|S )a; Return string representation of HTML formatted version of table in current state.
Arguments:
start - index of first data row to include in output
end - index of last data row to include in output PLUS ONE (list slice style)
fields - names of fields (columns) to include
header - print a header showing field names (True or False)
border - print a border around the table (True or False)
hrules - controls printing of horizontal rules after rows. Allowed values: ALL, FRAME, HEADER, NONE
vrules - controls printing of vertical rules between columns. Allowed values: FRAME, ALL, NONE
int_format - controls formatting of integer data
float_format - controls formatting of floating point data
padding_width - number of spaces on either side of column data (only used if left and right paddings are None)
left_padding_width - number of spaces on left hand side of column data
right_padding_width - number of spaces on right hand side of column data
sortby - name of field to sort rows by
sort_key - sorting key function, applied to data points before sorting
attributes - dictionary of name/value pairs to include as HTML attributes in the <table> tag
xhtml - print <br/> tags if True, <br> tags if falser4 )r� �_get_formatted_html_string�_get_simple_html_string)r[ r\ r� �stringr r r �get_html_string� s
zPrettyTable.get_html_stringc C sr g }|d rd}nd}g }|j d� |d rXx*|d D ] }|j d||d | f � q6W |j d� |j dj|�� |d r�|j d
� x>| jD ]4}|d r�||d kr�q�|j d t|�jd
|� � q�W |j d� | j|�}| j||�}xr|D ]j} |j d
� xPt| j| �D ]@\}}
|d �r*||d k�r*�q|j dt|
�jd
|� � �qW |j d� q�W |j d� | jd
�j|�S )Nr5 z<br/>z<br>z<tabler6 z %s="%s"�>r� r z <tr>r z <th>%s</th>r z </tr>z <td>%s</td>z</table>) r� r r8 r �replacer r r� rT ) r[ r� r � linebreak�open_tag� attr_namer� r� r rx �datumr r r r � s:
z#PrettyTable._get_simple_html_stringc C s� g }| j |�\}}|d r d}nd}g }|jd� |d �r|d tkr`|d tkr`|jd� n�|d tkr�|d tkr�|jd � n�|d tkr�|d tkr�|jd
� n`|d tkr�|jd � nH|d tkr�|jd � n0|d tkr�|jd
� n|d tk�r|jd� |d �r@x,|d D ] }|jd||d | f � �q W |jd� |jdj|�� |d �r�|jd� xL| jD ]B}|d �r�||d k�r��qv|jd||t|�jd|�f � �qvW |jd� | j|�} | j | |�}
g } g } xH| jD ]>}| jdddd �| j
| � | jd d d d �| j | � �q�W x�|
D ]~}
|jd� xbt | j|
| | �D ]N\}}}}|d �r|||d k�r|�qT|jd!||||t|�jd|�f � �qTW |jd� �q4W |jd"� | j
d�j|�S )#Nr5 z<br/>z<br>z<tabler r! r" z frame="box" rules="all"z frame="box"z frame="box" rules="cols"z frame="hsides"z frame="hsides" rules="rows"z frame="vsides"z frame="vsides" rules="cols"r6 z %s="%s"r r� r z <tr>r zW <th style="padding-left: %dem; padding-right: %dem; text-align: center">%s</th>r z </tr>�left�right�center)re rg r� �topZmiddleZbottom)r� r� r� zg <td style="padding-left: %dem; padding-right: %dem; text-align: %s; vertical-align: %s">%s</td>z</table>)r� r� rJ rH r r8 r r r r r9 r: r� rT )r[ r� r r� r� r r r r� r� r ZalignsZvalignsrx r rh r� r r r r � sd
&
"
*
z&PrettyTable._get_formatted_html_string)N)r� r� )��__name__�
__module__� __qualname__r^ rT ri rm ry �py3kr{ rz rA r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �propertyr= r� r� rh r� r� r� r� r� r~ r� r� r r� r� r r� r� r r� r� r# r� r� r$ r� r� r% r� r� r r� r� r r� r� r r� r� r! r� r� r" r� r� r( r� r� r) r� r� r* r� r� r+ r� r� r, r� r� r- r� r� r/ r� r� r1 r� r� r4 r� r� r3 r� r� r6 r� r� r� r� r� r� rr r� r� r� r� r� r� r� r� r r r r} r r r
r r r r r r r r R s
]
=/L (r c C s� d| kodkn r dS d| ko.dkn r8dS d| koJdkn rTdS t jt| ��rfd S d
| koxd kn s�d | ko�d
kn r�dS d| ko�dkn r�dS d| ko�dkn r�dS | dkr�dS | dkr�d S dS )N�! �~ r i N i�� r i � i�� r i@0 i�0 i�0 i�0 i� i`� i 0 i>0 r� � � )r� r& ���)r r' )� unicodedataZ combining�uni_chr)�charr r r �_char_block_width s& 0r, c C s t tttttjd| ����S )Nr� )�sum�itermapr, �ord�_re�sub)r� r r r r
/ s r
c K s� t j� j| jd��}| jd� t j| |�}tf |�}|r@||_n.trZdd� t |�D �|_ndd� |j � D �|_x |D ]}|j
dd� |D �� qtW |S )Ni r c S s g | ] }|j � �qS r )�strip)r r& r r r r B s z from_csv.<locals>.<listcomp>c S s g | ] }|j � �qS r )r2 )r r& r r r r D s c S s g | ] }|j � �qS r )r2 )r r& r r r r G s ) �csvZSnifferZsniff�read�seek�readerr r= r" �nextrr )�fpr= r\ Zdialectr6 �tablerx r r r �from_csv7 s
r: c K sF | j rBtf |�}dd� | j D �|_x| j� D ]}|j|� q,W |S d S )Nc S s g | ] }|d �qS )r r )r �colr r r r O s z"from_db_cursor.<locals>.<listcomp>)� descriptionr r= Zfetchallrr )Zcursorr\ r9 rx r r r �from_db_cursorK s
r= c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� TableHandlerc K s> t j| � || _g | _g | _g | _d| _d | _d| _d| _ d S )Nr r� F)
r r^ r\ �tables�last_rowr� �
max_row_width�active� last_content�is_last_row_header)r[ r\ r r r r^ V s
zTableHandler.__init__c C s || _ |dkrd| _d S )N�thT)rB rD )r[ �tag�attrsr r r �handle_starttaga s z TableHandler.handle_starttagc C s� |dkr | j j� }| jj|� |dkrZ| jj| j| jf� t| jt| j��| _g | _d| _|dkr�| j | j�}| j
j|� g | _d| _ d | _ d S )NrE �tdZtrFr9 rf )rE rI ) rC r2 r@ r� r� rD r rA r �generate_tabler? rB )r[ rF Zstripped_contentr9 r r r �
handle_endtagf s
zTableHandler.handle_endtagc C s | j |7 _ d S )N)rC )r[ �datar r r � handle_datax s zTableHandler.handle_datac C s� t f | j�}x�| jD ]~}t|d �| jk r^| jt|d � }x td|�D ]}|d jd� qHW |d dkr�| j|d � |d |_q|j |d � qW |S )zE
Generates from a list of rows a PrettyTable object.
r r r0 T)
r r\ r� r rA r� r� �make_fields_uniquer= rr )r[ r� r9 rx Zappendsr� r r r rJ { s zTableHandler.generate_tablec C sX xRt dt|��D ]@}x:t |d t|��D ]$}|| || kr(|| d7 < q(W qW dS )zB
iterates over the row and make each field unique
r r �'N)r� r )r[ r r� �jr r r rN � s z TableHandler.make_fields_uniqueN) r r r! r^ rH rK rM rJ rN r r r r r> T s r> c K s t f |�}|j| � |jS )z�
Generates a list of PrettyTables from a string of HTML code. Each <table> in
the HTML becomes one PrettyTable object.
)r> Zfeedr? )� html_coder\ �parserr r r � from_html� s
rS c K sF t | f|�}yt|�dks t�W n tk
r< td��Y nX |d S )zf
Generates a PrettyTables from a string of HTML code which contains only a
single <table>
r zDMore than one <table> in provided HTML code! Use from_html instead.r )rS r r� rt )rQ r\ r? r r r �
from_html_one� s rT c C s� t ddddg�} d| _d| _d| jd< d| _d| jd< | jd d
d d g� | jd
dddg� | jddddg� | jddddg� | jdddd g� | jd d d d g� | jd!d"d#d$g� t| � d S )%Nz City nameZAreaZ
PopulationzAnnual RainfallTZ04dz6.1fre ZAdelaidei is� g Ă@ZBrisbanei i:X g������@ZDarwin�p iD� g�����ʚ@ZHobartiM i�" g \�@ZSydneyi
i�*B g33333��@Z Melbournei i�: g333337�@ZPerthi
iQ� g33333+�@)r r# r$ r( r) rh rr �print)r& r r r �main� s
rW �__main__)N)2� __version__r� r3 r� �re�sysr � itertoolsr) � version_infor" rb rc ra �mapr. r� Ziterzip�chrr* Z html.parserr ZimapZizipZunichrZhtmlr ZcgirH rJ r� r� r� r� r� r� �compiler0 r �objectr r, r
r: r= r> rS rT rW r r r r r �<module> sj
E!
B