Edit File: prettytable.cpython-37.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�lines�height�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�_format�_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 rV � 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 �align�excessr 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�rowcount�colcountr )r r> r: �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? rA �setattr�getattrrc �slicer> �add_row�int� Exceptionre )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. r6 r$ r& r" r# r )r r! r% r7 r5 r r* r+ )r/ r1 r3 r8 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_attributesry )r^ r` �valr r r rC 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!)r: r �AssertionErrorry 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� ry )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)rh �crj z'Alignment %s is invalid, use l, c or r!)r� ry )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� ry )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!)rx r� ry rV )r^ rq 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� ry )r^ rq 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.)�typere rf r� �isdigitry )r^ rq 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� re rf r� r r r� ry )r^ rq 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� ry )r^ rq 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.)rL rJ �HEADER�NONEr� ry )r^ rq 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.)rL rJ r� r� ry )r^ rq 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!)r: r� ry )r^ rq 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� ry )r^ rq 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� ry )r^ rq 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!)rc �dictr� ry )r^ rq r� r r r r� � s z PrettyTable._validate_attributesc C s | j S )N)r: )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 )rV )r r' )r^ r r r � s z0PrettyTable._set_field_names.<locals>.<listcomp>r? r� r� )rC r: r; �zip�popr< )r^ r� � old_names�old_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� r: 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� r: 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� )rC r: 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)rF )r^ r r r � _get_fields� s zPrettyTable._get_fieldsc C s | � d|� || _d S )Nr )rC rF )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)rD )r^ r r r � _get_start� s zPrettyTable._get_startc C s | � d|� || _d S )Nr )rC rD )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))rE )r^ r r r �_get_end� s zPrettyTable._get_endc C s | � d|� || _d S )Nr )rC rE )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)rN )r^ r r r � _get_sortby� s zPrettyTable._get_sortbyc C s | � d|� || _d S )Nr$ )rC rN )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)rO )r^ r r r �_get_reversesort� s z PrettyTable._get_reversesortc C s | � d|� || _d S )Nr% )rC rO )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)rP )r^ r r r � _get_sort_key s zPrettyTable._get_sort_keyc C s | � d|� || _d S )Nr& )rC rP )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))rG )r^ r r r � _get_header s zPrettyTable._get_headerc C s | � d|� || _d S )Nr )rC rG )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))rH )r^ r r r �_get_header_style! s z PrettyTable._get_header_stylec C s | � |� || _d S )N)r� rH )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))rI )r^ r r r � _get_border- s zPrettyTable._get_borderc C s | � d|� || _d S )Nr! )rC rI )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)rK )r^ r r r � _get_hrules9 s zPrettyTable._get_hrulesc C s | � d|� || _d S )Nr" )rC rK )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)rM )r^ r r r � _get_vrulesE s zPrettyTable._get_vrulesc C s | � d|� || _d S )Nr# )rC rM )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)rQ )r^ r r r �_get_int_formatQ s zPrettyTable._get_int_formatc C s x| j D ]}|| j|< qW d S )N)r: rQ )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)rR )r^ r r r �_get_float_format] s z PrettyTable._get_float_formatc C s x| j D ]}|| j|< qW d S )N)r: rR )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)rS )r^ r r r �_get_padding_widthi s z PrettyTable._get_padding_widthc C s | � d|� || _d S )Nr, )rC rS )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)rT )r^ r r r �_get_left_padding_widthu s z#PrettyTable._get_left_padding_widthc C s | � d|� || _d S )Nr- )rC rT )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)rU )r^ r r r �_get_right_padding_width� s z$PrettyTable._get_right_padding_widthc C s | � d|� || _d S )Nr. )rC rU )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)rW )r^ r r r �_get_vertical_char� s z PrettyTable._get_vertical_charc C s | � |�}| �d|� || _d S )Nr/ )rV rC rW )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)rX )r^ r r r �_get_horizontal_char� s z PrettyTable._get_horizontal_charc C s | � |�}| �d|� || _d S )Nr1 )rV rC rX )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)rY )r^ r r r �_get_junction_char� s z PrettyTable._get_junction_charc C s | � |�}| �d|� || _d S )Nr/ )rV rC rY )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)r[ )r^ r r r � _get_format� s zPrettyTable._get_formatc C s | � d|� || _d S )Nr6 )rC r[ )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)rZ )r^ r r r �_get_print_empty� s z PrettyTable._get_print_emptyc C s | � d|� || _d S )Nr5 )rC rZ )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 )Nr8 )rC 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 )Nrs )rA rC ru )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_stylery )r^ �styler r r � set_style� s zPrettyTable.set_stylec C s@ d| _ d| _t| _t| _d| _d| _d| _d| _ d| _ d| _ d S )NTr r0 r2 r4 ) r r! rJ rK rL rM r, r- r. r/ r1 r3 )r^ r r r r� � s z PrettyTable._set_default_stylec C s. d| _ d| _t| _d| _d| _d| _d| _d S )NTr r0 )r r! r� rK 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 ~!@#$%^&*()_+|-=\{}[];':",./;<>?)�random�choicer r! rL rJ r� r� rK rM �randintr- r. r/ r1 r3 )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)r: r ry �ranger? r> �append�list)r^ r} r r r rw ( 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> ry )r^ � 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� r: r� r; r< r� ry )r^ � fieldname�columnrk �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> r: 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)rc rx rQ rV �floatrR )r^ r� rg 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 ) r: r r? � enumerater� r �minr r@ )r^ �rowsr� �widthsr} rz rg 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 r} )� sortindexr r r � s z)PrettyTable._get_rows.<locals>.<listcomp>r% r& )�reverse�keyc S s g | ]}|d d� �qS )r Nr )r r} r r r r � s )r r r> r: rz �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� rg )r^ r r r � s z+PrettyTable._format_row.<locals>.<listcomp>)r� r: )r^ r} r� r )r^ r � _format_row� s zPrettyTable._format_rowc s � �fdd�|D �S )Nc s g | ]}�� |� ��qS r )r )r r} )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 r5 r! r� r r" r )r� rn r r r �_stringify_hrule�_hruler� �_stringify_headerrL rJ �_stringify_rowrV �join)r^ r_ r� r r �formatted_rowsr} 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# r3 r1 r ) r rL rJ r: 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/ ri r r� r� r� r� r� )r rL rJ r� r r: r� r@ rH � capitalizer� r� r� rm 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# ri r� r� r r� r r/ r" )r� r� r r: r@ r r �textwrap�fillr� r r r rL rJ r/ r< rx rm r; r� r )r^ r} r� rz r� rg r r � new_linesr � row_heightr� �hr� r r �yr� �dHeightrh 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 falser6 )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 )Nr7 z<br/>z<br>z<tabler8 z %s="%s"�>r� r z <tr>r z <th>%s</th>r z </tr>z <td>%s</td>z</table>) r� r r: r �replacer r r� rV ) r^ r� r � linebreak�open_tag� attr_namer� r r r} �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 )#Nr7 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"r8 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)rh rj r� �top�middle�bottom)r� r� r� zg <td style="padding-left: %dem; padding-right: %dem; text-align: %s; vertical-align: %s">%s</td>z</table>)r r� rL rJ r r: r r& r r r; r< r� rV )r^ r� r r r r' r( r) r� r r �aligns�valignsr} r* rk r� r r r r! � sd & " * z&PrettyTable._get_formatted_html_string)N)r� r� )��__name__� __module__� __qualname__ra rV rm rr r~ �py3kr� r rC r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �propertyr? r� r� rk 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� r3 r� r� r6 r� r� r5 r� r� r8 r� r� r� r� r� r� rw 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 � )� unicodedata� combining�uni_chr)�charr r r �_char_block_width s& .rA c C s t ttttt�d| ����S )Nr� )�sum�itermaprA �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 )rG )r r' r r r r D s c S s g | ] }|� � �qS r )rG )r r' r r r r G s ) �csv�Sniffer�sniff�read�seek�readerr r? r6 �nextrw )�fpr? r_ �dialectrM �tabler} r r r �from_csv7 s rR 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>)� descriptionr r? �fetchallrw )�cursorr_ rQ r} r r r �from_db_cursorK s rW 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 ra r_ �tables�last_rowr � max_row_width�active� last_content�is_last_row_header)r^ r_ r r r ra V s zTableHandler.__init__c C s || _ |dkrd| _d S )N�thT)r\ r^ )r^ �tag�attrsr 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)r_ �td�trFrQ ri ) r] rG rZ r� r r^ r r[ r �generate_tablerY r\ )r^ r` �stripped_contentrQ r r r � handle_endtagf s zTableHandler.handle_endtagc C s | j |7 _ d S )N)r] )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 r2 T) r r_ r r r[ r� r� �make_fields_uniquer? rw )r^ r rQ r} �appendsr� r r r re { 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 rj � s z TableHandler.make_fields_uniqueN) r3 r4 r5 ra rb rg ri re rj r r r r rX T s rX 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. )rX �feedrY )� 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� ry )ro r_ rY 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 name�Area� PopulationzAnnual RainfallT�04dz6.1frh �Adelaidei is� g Ă@�Brisbanei i:X g������@�Darwin�p iD� g�����ʚ@�HobartiM i�" g \�@�Sydneyi i�*B g33333��@� Melbournei i�: g333337�@�Perthi iQ� g33333+�@)r r$ r% r* r+ rk rw �print)r' r r r �main� s r �__main__)N)2� __version__r rH r� �re�sysr � itertoolsr= � version_infor6 re rf rd �maprC r� �iterzip�chrr? � html.parserr �imapZizip�unichr�htmlr �cgirJ rL r� r� r� r� r� r� �compilerE r �objectr rA r rR rW rX rq rr r r3 r r r r �<module> sj E! B