Edit File: prettytable.cpython-37.opt-1.pyc
B 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#e�$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, | � d�}t|�}tdd� |D ��}||fS )N� c S s g | ] }t |��qS � )�_str_block_width)�.0�liner r �</opt/alt/python37/lib/python3.7/site-packages/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 |� dd�| _g | _i | _i | _i | _g | _|r8|| _ng | _d� � | _ | j � d� � � | j � d� � � x0| j D ]&}||kr�| � ||| � 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�| � d �| _ |d �p�| � d!�| _!|d" �p| � 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�header)TFT� header_style�border�hrules�vrules�sortby� reversesortF�sort_keyc S s | S )Nr )�xr r r �<lambda>� � 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)'�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 rU � 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= r9 �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 ]}|�|� q`W n,t|t �r�|�| j| � nt dt |� ��|S )N�_r: z0Index %s is invalid, must be an integer or slice) r r> r@ �setattr�getattrra �slicer= �add_row�int� Exceptionrc )r\ �index�new�attr�rowr r r � __getitem__� s zPrettyTable.__getitem__c C s | � � S )N)� __unicode__)r\ r r r �__str__� s zPrettyTable.__str__c C s | � � �| j�S )N)r{ �encoder )r\ r r r r| � s c C s | � � S )N)� get_string)r\ r r r r{ � s zPrettyTable.__unicode__c C s2 |dkr| � |� �n|dkr.| �||� �n |dkrD| �||� n�|dkrZ| �||� n�|dkrp| �||� n�|dkr�| �||� n�|dkr�| �||� n�|dkr�| �||� n||d kr�| �|� nh|d kr�| � ||� nR|d kr�| � ||� n<|d k�r | � ||� n$|d k�r"| � ||� n t d| ��d S )Nr> )r r � max_widthr+ r, r- r5 r# r% r! r" r )r r r$ r6 r4 r r) r* )r. r0 r2 r7 zUnrecognised option: %s!)�_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_attributesru )r\ r^ �valr r r rB 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!)r9 r �AssertionErrorru 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�lowerNz;Invalid header style, use cap, title, upper, lower or None!)r� ru )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 )N)rf �crh z'Alignment %s is invalid, use l, c or r!)r� ru )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�bNz-Alignment %s is invalid, use t, m, b or None!)r� ru )r\ r� r r r �_validate_valign< s z PrettyTable._validate_valignc C sD yt |�dkst�W n* tk r> td|| �|�f ��Y nX d S )Nr zInvalid value for %s: %s!)rt r� ru rU )r\ rm 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 )N)TFz-Invalid value for %s! Must be True or False.)r� ru )r\ rm r� r r r r� H s z#PrettyTable._validate_true_or_falsec C sV |dkr d S y$t |�ttfks"t�|�� s.t�W n tk rP td| ��Y nX d S )N� z8Invalid value for %s! Must be an integer format string.)�typerc rd r� �isdigitru )r\ rm r� r r r r� N s z PrettyTable._validate_int_formatc C s� |dkr d S yvt |�ttfks"t�d|ks.t�|�d�}t|�dksHt�|d dksd|d �� sdt�|d dks�|d �� 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� rc rd r� r r r� ru )r\ rm 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� ru )r\ rm 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.)rK rI �HEADER�NONEr� ru )r\ rm 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.)rK rI r� r� ru )r\ rm 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!)r9 r� ru )r\ rm r� r r r r� v s z PrettyTable._validate_field_namec C s@ y x|D ]}| � ||� qW W n tk r: td��Y nX d S )Nz)fields must be a sequence of field names!)r� r� ru )r\ rm 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� ru )r\ rm 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!)ra �dictr� ru )r\ rm r� r r r r� � s z PrettyTable._validate_attributesc C s | j S )N)r9 )r\ r r r �_get_field_names� s z PrettyTable._get_field_namesc s& � fdd�|D �}� � d|� � jr2� jd d � }|� _� jr�|r�x&t||�D ]\}}� j| � j|< qNW x>|D ]}|� jkrp� j�|� qpW nx� jD ]}d� j|< q�W � j�r|�rx&t||�D ]\}}� j| � j|< q�W x@|D ]}|� jkr� j�|� q�W n x� jD ]}d� j|< �qW d S )Nc s g | ]}� � |��qS r )rU )r r& )r\ r r r � s z0PrettyTable._set_field_names.<locals>.<listcomp>r> r� r� )rB r9 r: �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)r: )r\ r r r � _get_align� s zPrettyTable._get_alignc C s( | � |� x| jD ]}|| j|< qW d S )N)r� r9 r: )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( | � |� x| jD ]}|| j|< qW d S )N)r� r9 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* | � d|� x| jD ]}|| j|< qW d S )Nr )rB r9 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)rE )r\ r r r � _get_fields� s zPrettyTable._get_fieldsc C s | � d|� || _d S )Nr )rB rE )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)rC )r\ r r r � _get_start� s zPrettyTable._get_startc C s | � d|� || _d S )Nr )rB rC )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))rD )r\ r r r �_get_end� s zPrettyTable._get_endc C s | � d|� || _d S )Nr )rB rD )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)rM )r\ r r r � _get_sortby� s zPrettyTable._get_sortbyc C s | � d|� || _d S )Nr# )rB rM )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)rN )r\ r r r �_get_reversesort� s z PrettyTable._get_reversesortc C s | � d|� || _d S )Nr$ )rB rN )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)rO )r\ r r r � _get_sort_key s zPrettyTable._get_sort_keyc C s | � d|� || _d S )Nr% )rB rO )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))rF )r\ r r r � _get_header s zPrettyTable._get_headerc C s | � d|� || _d S )Nr )rB rF )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))rG )r\ r r r �_get_header_style! s z PrettyTable._get_header_stylec C s | � |� || _d S )N)r� rG )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))rH )r\ r r r � _get_border- s zPrettyTable._get_borderc C s | � d|� || _d S )Nr )rB rH )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)rJ )r\ r r r � _get_hrules9 s zPrettyTable._get_hrulesc C s | � d|� || _d S )Nr! )rB rJ )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)rL )r\ r r r � _get_vrulesE s zPrettyTable._get_vrulesc C s | � d|� || _d S )Nr" )rB rL )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)rP )r\ r r r �_get_int_formatQ s zPrettyTable._get_int_formatc C s x| j D ]}|| j|< qW d S )N)r9 rP )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)rQ )r\ r r r �_get_float_format] s z PrettyTable._get_float_formatc C s x| j D ]}|| j|< qW d S )N)r9 rQ )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)rR )r\ r r r �_get_padding_widthi s z PrettyTable._get_padding_widthc C s | � d|� || _d S )Nr+ )rB rR )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)rS )r\ r r r �_get_left_padding_widthu s z#PrettyTable._get_left_padding_widthc C s | � d|� || _d S )Nr, )rB rS )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)rT )r\ r r r �_get_right_padding_width� s z$PrettyTable._get_right_padding_widthc C s | � d|� || _d S )Nr- )rB rT )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)rV )r\ r r r �_get_vertical_char� s z PrettyTable._get_vertical_charc C s | � |�}| �d|� || _d S )Nr. )rU rB rV )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)rW )r\ r r r �_get_horizontal_char� s z PrettyTable._get_horizontal_charc C s | � |�}| �d|� || _d S )Nr0 )rU rB rW )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)rX )r\ r r r �_get_junction_char� s z PrettyTable._get_junction_charc C s | � |�}| �d|� || _d S )Nr. )rU rB rX )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)rZ )r\ r r r � _get_format� s zPrettyTable._get_formatc C s | � d|� || _d S )Nr5 )rB rZ )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)rY )r\ r r r �_get_print_empty� s z PrettyTable._get_print_emptyc C s | � d|� || _d S )Nr4 )rB rY )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)r[ )r\ r r r �_get_attributes� s zPrettyTable._get_attributesc C s | � d|� || _d S )Nr7 )rB r[ )r\ r� r r r �_set_attributes� s zPrettyTable._set_attributesc C sP i }xF| j D ]<}||kr6| �||| � || ||< q t| d| �||< q W |S )Nro )r@ rB rq )r\ r] �optionsr^ r r r � _get_options� s zPrettyTable._get_optionsc C sT |t kr| �� n>|tkr$| �� n,|tkr6| �� n|tkrH| �� ntd��d S )NzInvalid pre-set style!) �DEFAULT�_set_default_style�MSWORD_FRIENDLY�_set_msword_style� PLAIN_COLUMNS�_set_columns_style�RANDOM�_set_random_styleru )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/ r1 r3 ) r r rI rJ rK rL r+ r, r- r. r0 r2 )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� rJ 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 �d�| _t �d�| _t �ttttf�| _t �tttf�| _ t � dd�| _ t � dd�| _ t �d�| _ t �d�| _t �d�| _d S )N)TFr � z ~!@#$%^&*()_+|-=\{}[];':",./;<>?)�randomZchoicer r rK rI r� r� rJ rL Zrandintr, r- r. r0 r2 )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 �| _| j�t|�� 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)r9 r ru �ranger> r= �append�list)r\ ry r r r rs ( 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= ru )r\ Z row_indexr r r �del_row7 s zPrettyTable.del_rowr� r� c C s� t | j�dt |�fkr�| �|� | �|� | j�|� || j|< || j|< xdtdt |��D ]6}t | j�|d k r|| j�g � | 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� r9 r� r: r; r� ru )r\ � fieldname�columnri �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= r9 r? )r\ r r r �clearb s zPrettyTable.clearc C s t �| �S )N)�copy�deepcopy)r\ r r r r� n s zPrettyTable.copyc C sd t |t�r.|| jkr.| �d| j| | �}n,t |t�rZ|| jkrZ| �d| j| | �}| �|�S )Nz%%%sdz%%%sf)ra rt rP rU �floatrQ )r\ r� re 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 ) r9 r r> � enumerater r �minr r? )r\ �rowsr� Zwidthsry rv re 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|d |d � �}|d rj| j�|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 ry )� sortindexr r r � s z)PrettyTable._get_rows.<locals>.<listcomp>r$ r% )�reverse�keyc S s g | ]}|d d� �qS )r Nr )r ry r r r r � s )r� r� r= r9 rv �sort)r\ r� r� r )r r � _get_rows� s zPrettyTable._get_rowsc s � fdd�t � j|�D �S )Nc s g | ]\}}� � ||��qS r )r� )r r� re )r\ r r r � s z+PrettyTable._format_row.<locals>.<listcomp>)r� r9 )r\ ry r� r )r\ r � _format_row� s zPrettyTable._format_rowc s � �fdd�|D �S )Nc s g | ]}�� |� ��qS r )r )r ry )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� | � |�}g }| jdkr,|d r(|d s,dS | �|�}| �||�}| �||� | �|�| _|d rt|�| �|�� n$|d r�|d t t fkr�|�| j� x |D ]}|�| � ||�� q�W |d r�|d t kr�|�| j� | � d�� |�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 r4 r r� r r! r )r� rk r r r� �_stringify_hrule�_hruler� �_stringify_headerrK rI �_stringify_rowrU �join)r\ r] r� r r� �formatted_rowsry r r r r~ � s" zPrettyTable.get_stringc C s |d s dS | � |�\}}|d ttfkr6|d g}n |d g}| js^|�|d � d�|�S xtt| j| j�D ]b\}}|d r�||d kr�qn|�|| | |d � |d tkr�|�|d � qn|�|d � qnW |d tkr�|�� |�|d � d�|�S )Nr r� r" r2 r0 r ) r rK rI r9 r� r r� r? r� )r\ r� r� r� r� r� r r r r r � s( z PrettyTable._stringify_hrulec C s� g }| � |�\}}|d rj|d ttfkr@|�| j� |�d� |d ttfkr`|�|d � n |�d� | js�|d ttfkr�|�|d � n |�d� x�t| j| j�D ]�\}}|d r�||d kr�q�| jdkr�|� � }nD| jd kr�|� � }n0| jd k�r|� � }n| jd k�r |� � }n|}|�d| | � ||| j| � d| � |d r�|d tk�rp|�|d � q�|�d� q�W |d �r�|d tk�r�|�� |�|d � |d �r�|d tk�r�|�d� |�| j� d �|�S ) Nr r! r r" r. rg r r� r� r� r� r� )r rK rI r� r r9 r� r? rG � capitalizer� r� r� rj r: 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 ]`\}}}}|�d�}g }x.|D ]&} t| �|krZt�| |�} |� | � q>W |}d� |�}|||< q W d} x$|D ] } t | �d } | | kr�| } q�W g } | � |�\}}xVtd| �D ]H}| � g � |d r�|d t tfk�r| | � | j� q�| | � d� q�W �xFt | j|| j�D �]0\}}}| j| }|�d�}| 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�| | � d| | �||| j| � d| � |d �rL|d t k�r>| | � | j� n| | � d� |d7 }�q�W �q*W xJtd| �D ]<}|d �rl|d tk�rl| | �� | | � |d � �qlW |d �r�|d t k�r�| | d � d� | | d � | j� x&td| �D ]}d� | | �| |< �q�W d� | �S )Nr r r r r" rg r� r� r r� r r. r! )r� r� r r9 r? r r �textwrapZfillr� r r r rK rI r. r; rt rj r: r� r )r\ ry r� rv r� re r r Z new_linesr Z row_heightr� �hr� r� r� �yr� ZdHeightrf r r r r : sj * . . zPrettyTable._stringify_rowc K s, | � |�}|d r | �|�}n | �|�}|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 falser5 )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 }|� d� |d rXx*|d D ] }|� d||d | f � q6W |� d� |� d�|�� |d r�|� d � x>| jD ]4}|d r�||d kr�q�|� d t|��d |� � q�W |� d� | �|�}| �||�}xr|D ]j} |� d � xPt| j| �D ]@\}} |d �r*||d k�r*�q|� dt| ��d |� � �qW |� d� q�W |� d� | �d ��|�S )Nr6 z<br/>z<br>z<tabler7 z %s="%s"�>r� r z <tr>r z <th>%s</th>r z </tr>z <td>%s</td>z</table>) r� r r9 r �replacer r r� rU ) r\ r� r � linebreak�open_tag� attr_namer� r� r ry �datumr r r r � s: z#PrettyTable._get_simple_html_stringc C s� g }| � |�\}}|d r d}nd}g }|�d� |d �r|d tkr`|d tkr`|�d� n�|d tkr�|d tkr�|�d � n�|d tkr�|d tkr�|�d � n`|d tkr�|�d � nH|d tkr�|�d � n0|d tkr�|�d � n|d tk�r|�d� |d �r@x,|d D ] }|�d||d | f � �q W |�d� |�d�|�� |d �r�|�d� xL| jD ]B}|d �r�||d k�r��qv|�d||t|��d|�f � �qvW |�d� | �|�} | � | |�} g } g } xH| jD ]>}| �dddd �| j | � | �d d d d �| j | � �q�W x�| D ]~} |�d� xbt | j| | | �D ]N\}}}}|d �r|||d k�r|�qT|�d!||||t|��d|�f � �qTW |�d� �q4W |�d"� | � d��|�S )#Nr6 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"r7 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)rf rh 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� rK rI r r9 r r r r r: r; r� rU )r\ r� r r� r� r r r r� r� r ZalignsZvalignsry r ri r� r r r r � sd & " * z&PrettyTable._get_formatted_html_string)N)r� r� )��__name__� __module__� __qualname__r_ rU rj rn rz �py3kr| r{ rB r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �propertyr> r� r� ri 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� r0 r� r� r2 r� r� r5 r� r� r4 r� r� r7 r� r� r� r� r� r� rs 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| krdkr n ndS d| kr0dkr8n ndS d| krLdkrTn ndS t �t| ��rfd S d | krzd ks�n d | kr�d kr�n ndS d| kr�dkr�n ndS d| kr�dkr�n ndS | 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 � )� unicodedataZ combining�uni_chr)�charr r r �_char_block_width s& .r- c C s t ttttt�d| ����S )Nr� )�sum�itermapr- �ord�_re�sub)r� r r r r / s r c K s� t �� �| �d��}| �d� t �| |�}tf |�}|r@||_n.trZdd� t |�D �|_ndd� |� � D �|_x |D ]}|� dd� |D �� qtW |S )Ni r c S s g | ] }|� � �qS r )�strip)r r& r r r r B s z from_csv.<locals>.<listcomp>c S s g | ] }|� � �qS r )r3 )r r& r r r r D s c S s g | ] }|� � �qS r )r3 )r r& r r r r G s ) �csvZSnifferZsniff�read�seek�readerr r> r# �nextrs )�fpr> r] Zdialectr7 �tablery r r r �from_csv7 s r; c K sF | j rBtf |�}dd� | j D �|_x| �� D ]}|�|� 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>)Z descriptionr r> Zfetchallrs )Zcursorr] r: ry 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 �| � || _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\ �tagZattrsr r r �handle_starttaga s z TableHandler.handle_starttagc C s� |dkr | j �� }| j�|� |dkrZ| j�| j| jf� t| jt| j��| _g | _d| _|dkr�| � | j�}| j �|� g | _d| _ d | _ d S )N)rE ZtdZtrFr: rg ) rC r3 r@ r� r� rD r rA r �generate_tabler? rB )r\ rF Zstripped_contentr: 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 �d� qHW |d dkr�| �|d � |d |_q|� |d � qW |S )zE Generates from a list of rows a PrettyTable object. r r r1 T) r r] r� r rA r� r� �make_fields_uniquer> rs )r\ r� r: ry Zappendsr� r r r rH { 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 rL � s z TableHandler.make_fields_uniqueN) r r! r" r_ rG rI rK rH rL r r r r r> T s r> c K s t f |�}|�| � |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 rQ 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 )rQ r r� ru )rO r] r? r r r � from_html_one� s rR c C s� t ddddg�} d| _d| _d| jd< d| _d| jd< | �d d d d g� | �d dddg� | �ddddg� | �ddddg� | �dddd g� | �d d d d g� | �d!d"d#d$g� t| � d S )%Nz City nameZAreaZ PopulationzAnnual RainfallTZ04dz6.1frf 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* ri rs �print)r& r r r �main� s rU �__main__)N)2� __version__r� r4 r� �re�sysr � itertoolsr* � version_infor# rc rd rb �mapr/ r� Ziterzip�chrr+ Z html.parserr ZimapZizipZunichrZhtmlr ZcgirI rK r� r� r� r� r� r� �compiler1 r �objectr r- r r; r= r> rQ rR rU r r r r r �<module> sj E! B