Google
Edit File: prettytable.cpython-39.pyc
a 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 �//usr/lib/python3.9/site-packages/prettytable.py� <listcomp>O � 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 sV |� dd�| _g | _i | _i | _i | _g | _|r8|| _ng | _d� � | _ | j � d� � � | j � d� � � | j D ]&}||v r�| � ||| � qnd||< qn|d p�d| _ |d p�d| _|d p�d| _|d d v r�|d | _nd | _|d p�d| _|d d v �r|d | _nd | _|d �pt| _|d �p(t| _|d �p8d| _|d d v �rV|d | _nd| _|d �pldd� | _|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 v �r |d$ | _#nd | _#|d% �p.d| _$|d& �p>d| _%|d' �pNi | _&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>� r 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)�selfrB �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 rY � 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 rA r= �AttributeError)r` �namer r r � __getattr__� s zPrettyTable.__getattr__c C s� t � }| j|_| jD ] }t|d| t| d| �� qt|dt| d�� t|t�rl| j| D ]}|�|� qZn,t|t �r�|�| j| � nt dt |� ��|S )N�_r> z0Index %s is invalid, must be an integer or slice) r rB rD �setattr�getattrre �slicerA �add_row�int� Exceptionrg )r` �index�new�attr�rowr r r � __getitem__� s zPrettyTable.__getitem__c C s | � � S r) )� __unicode__�r` r r r �__str__� s zPrettyTable.__str__c C s | � � �| j�S r) )r �encoder r� r r r r� � s c C s | � � S r) )� get_stringr� r r r r � s zPrettyTable.__unicode__c C s2 |dv r| � |� �n|dv r.| �||� �n |dv rD| �||� n�|dv rZ| �||� n�|dv rp| �||� n�|dv r�| �||� n�|dv r�| �||� n�|dv r�| �||� n||d v r�| �|� nh|d v r�| � ||� nR|d v r�| � ||� n<|d v �r | � ||� n$|d v �r"| � ||� n t d| ��d S )NrB )r r � max_widthr/ r0 r1 r9 r& r( r$ r% r )r r# r' r: r8 r" r- r. )r2 r4 r6 r; 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` rb �valr r r rF s6 z PrettyTable._validate_optionc C s� | j rNzt|�t| j �ks J �W n, tyL tdt|�t| j �f ��Y n0 | jr�z t|�t| jd �kspJ �W n0 ty� tdt|�t| jd �f ��Y n0 z t|�tt|��ks�J �W n ty� td��Y n0 d S )NzJField name list has incorrect number of values, (actual) %d!=%d (expected)r zField names must be unique!)r= r �AssertionErrorry rA �set�r` r� r r r r� s $ z!PrettyTable._validate_field_namesc C s0 z|dv sJ �W n t y* td��Y n0 d S )N)�cap�title�upper�lowerNz;Invalid header style, use cap, title, upper, lower or None!�r� ry r� r r r r� 0 s z"PrettyTable._validate_header_stylec C s4 z|dv sJ �W n t y. td| ��Y n0 d S )N)rj �crl z'Alignment %s is invalid, use l, c or r!r� r� r r r �_validate_align6 s zPrettyTable._validate_alignc C s4 z|dv sJ �W n t y. td| ��Y n0 d S )N)�t�m�bNz-Alignment %s is invalid, use t, m, b or None!r� r� r r r �_validate_valign< s z PrettyTable._validate_valignc C sB zt |�dksJ �W n( ty< td|| �|�f ��Y n0 d S )Nr zInvalid value for %s: %s!)rx r� ry rY �r` rq r� r r r r� B s z%PrettyTable._validate_nonnegative_intc C s4 z|dv sJ �W n t y. td| ��Y n0 d S )Nr! z-Invalid value for %s! Must be True or False.r� r� r r r r� H s z#PrettyTable._validate_true_or_falsec C sT |dkr d S z$t |�ttfv s"J �|�� s.J �W n tyN td| ��Y n0 d S )N� z8Invalid value for %s! Must be an integer format string.)�typerg rh �isdigitr� ry r� r r r r� N s z PrettyTable._validate_int_formatc C s� |dkr d S zvt |�ttfv s"J �d|v s.J �|�d�}t|�dksHJ �|d dksd|d �� sdJ �|d dks�|d �� s�J �W n ty� td| ��Y n0 d S )Nr� �.r r r z5Invalid value for %s! Must be a float format string.)r� rg rh r r r� r� ry )r` rq r� �bitsr r r r� W s z"PrettyTable._validate_float_formatc C s6 zt |d�sJ �W n ty0 td| ��Y n0 d S )N�__call__z*Invalid value for %s! Must be a function.)�hasattrr� ry r� r r r r� d s z PrettyTable._validate_functionc C s< z|t tttfv sJ �W n ty6 td| ��Y n0 d S )Nz:Invalid value for %s! Must be ALL, FRAME, HEADER or NONE.)rO rM �HEADER�NONEr� ry r� r r r r� j s z PrettyTable._validate_hrulesc C s: z|t ttfv sJ �W n ty4 td| ��Y n0 d S )Nz3Invalid value for %s! Must be ALL, FRAME, or NONE.)rO rM r� r� ry r� r r r r� p s z PrettyTable._validate_vrulesc C s> z|| j v s|d u sJ �W n ty8 td| ��Y n0 d S )NzInvalid field name: %s!)r= r� ry r� r r r r� v s z PrettyTable._validate_field_namec C s: z|D ]}| � ||� qW n ty4 td��Y n0 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 s8 zt |�dksJ �W n ty2 td| ��Y n0 d S )Nr z4Invalid value for %s! Must be a string of length 1.)r r� ry r� r r r r� � s z!PrettyTable._validate_single_charc C s2 zt |t�sJ �W n ty, td��Y n0 d S )Nz4attributes must be a dictionary of name/value pairs!)re �dictr� ry r� r r r r� � s z PrettyTable._validate_attributesc C s | j S r) )r= r� r r r �_get_field_names� s z PrettyTable._get_field_namesc s � fdd�|D �}� � d|� � jr2� jd d � }|� _� jr�|r�t||�D ]\}}� j| � j|< qL|D ]}|� jvrj� j�|� qjn� jD ]}d� j|< q�� jr�|r�t||�D ]\}}� j| � j|< q�|D ]}|� jvrЈ j�|� q�n� jD ]}d� j|< q�d S )Nc s g | ]}� � |��qS r )rY �r r+ r� r r r � r z0PrettyTable._set_field_names.<locals>.<listcomp>rB r� r� )rF r= 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 r) )r> r� r r r � _get_align� s zPrettyTable._get_alignc C s$ | � |� | jD ]}|| j|< qd S r) )r� r= r> �r` r� r� r r r � _set_align� s zPrettyTable._set_alignc C s | j S r) )r? r� r r r � _get_valign� s zPrettyTable._get_valignc C s$ | � |� | jD ]}|| j|< qd S r) )r� r= r? r� r r r � _set_valign� s zPrettyTable._set_valignc C s | j S r) )r@ r� r r r �_get_max_width� s zPrettyTable._get_max_widthc C s& | � d|� | jD ]}|| j|< qd S )Nr� )rF 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)rI r� r r r � _get_fields� s zPrettyTable._get_fieldsc C s | � d|� || _d S )Nr )rF rI 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)rG r� r r r � _get_start� s zPrettyTable._get_startc C s | � d|� || _d S )Nr )rF rG 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))rH r� r r r �_get_end� s zPrettyTable._get_endc C s | � d|� || _d S )Nr )rF rH 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)rQ r� r r r � _get_sortby� s zPrettyTable._get_sortbyc C s | � d|� || _d S )Nr& )rF rQ 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)rR r� r r r �_get_reversesort� s z PrettyTable._get_reversesortc C s | � d|� || _d S )Nr' )rF rR 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)rS r� r r r � _get_sort_key s zPrettyTable._get_sort_keyc C s | � d|� || _d S )Nr( )rF rS 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))rJ r� r r r � _get_header s zPrettyTable._get_headerc C s | � d|� || _d S )Nr )rF rJ 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))rK r� r r r �_get_header_style! s z PrettyTable._get_header_stylec C s | � |� || _d S r) )r� rK 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))rL r� r r r � _get_border- s zPrettyTable._get_borderc C s | � d|� || _d S )Nr# )rF rL 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)rN r� r r r � _get_hrules9 s zPrettyTable._get_hrulesc C s | � d|� || _d S )Nr$ )rF rN 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)rP r� r r r � _get_vrulesE s zPrettyTable._get_vrulesc C s | � d|� || _d S )Nr% )rF rP 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)rT r� r r r �_get_int_formatQ s zPrettyTable._get_int_formatc C s | j D ]}|| j|< qd S r) )r= rT 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)rU r� r r r �_get_float_format] s z PrettyTable._get_float_formatc C s | j D ]}|| j|< qd S r) )r= rU 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)rV r� r r r �_get_padding_widthi s z PrettyTable._get_padding_widthc C s | � d|� || _d S )Nr/ )rF rV 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)rW r� r r r �_get_left_padding_widthu s z#PrettyTable._get_left_padding_widthc C s | � d|� || _d S )Nr0 )rF rW 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)rX r� r r r �_get_right_padding_width� s z$PrettyTable._get_right_padding_widthc C s | � d|� || _d S )Nr1 )rF rX 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)rZ r� r r r �_get_vertical_char� s z PrettyTable._get_vertical_charc C s | � |�}| �d|� || _d S �Nr2 )rY rF rZ 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)r[ r� r r r �_get_horizontal_char� s z PrettyTable._get_horizontal_charc C s | � |�}| �d|� || _d S )Nr4 )rY rF 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)r\ r� r r r �_get_junction_char� s z PrettyTable._get_junction_charc C s | � |�}| �d|� || _d S r� )rY rF 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 )Nr9 )rF 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)r] r� r r r �_get_print_empty� s z PrettyTable._get_print_emptyc C s | � d|� || _d S )Nr8 )rF 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 )Nr; )rF r_ r� r r r �_set_attributes� s zPrettyTable._set_attributesc C sL i }| j D ]<}||v r4| �||| � || ||< q t| d| �||< q |S )Nrs )rD rF ru )r` ra �optionsrb 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` Zstyler r r � set_style� s zPrettyTable.set_stylec C s@ d| _ d| _t| _t| _d| _d| _d| _d| _ d| _ d| _ d S )NTr r3 r5 r7 ) r r# rM rN rO rP r/ r0 r1 r2 r4 r6 r� r r r r� � s z PrettyTable._set_default_stylec C s. d| _ d| _t| _d| _d| _d| _d| _d S )NTr r3 )r r# r� rN r/ r0 r1 r2 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/ r0 r1 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 )Nr! r � z ~!@#$%^&*()_+|-=\{}[];':",./;<>?)�random�choicer r# rO rM r� r� rN rP Zrandintr0 r1 r2 r4 r6 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 r z'PrettyTable.add_row.<locals>.<listcomp>r N)r= r ry �rangerB rA �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 rA ry )r` Z row_indexr r r �del_row7 s zPrettyTable.del_rowr� r� c C s� t | j�dt |�fv r�| �|� | �|� | j�|� || j|< || j|< tdt |��D ]6}t | j�|d k rz| j�g � | j| �|| � qXnt 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 rA r� r� r= r� r> r? r� ry )r` � fieldname�columnrm �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)rA 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)rA r= rC r� r r r �clearb s zPrettyTable.clearc C s t �| �S r) )�copy�deepcopyr� r r r r n s zPrettyTable.copyc C sd t |t�r.|| jv r.| �d| j| | �}n,t |t�rZ|| jv rZ| �d| j| | �}| �|�S )Nz%%%sdz%%%sf)re rx rT rY �floatrU )r` r� ri r r r � _format_valueu s zPrettyTable._format_valuec C s� |d rdd� | j D �}nt| j�dg }|D ]l}t|�D ]^\}}| j| }|| jv r~t|| tt|�d | j| ��||< q:t|| t|�d �||< q:q.|| _d S )Nr c S s g | ]}t |�d �qS �r )r )r r� r r r r ~ r z/PrettyTable._compute_widths.<locals>.<listcomp>r ) r= r rB � enumerater� r �minr rC )r` �rowsr� Zwidthsr} rz ri r� r r r �_compute_widths| s ( zPrettyTable._compute_widthsc C sD |d d ur|d }n|d }|d d ur4|d }n|d }||fS )Nr0 r/ r1 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} �Z sortindexr r r � r z)PrettyTable._get_rows.<locals>.<listcomp>r' r( )�reverse�keyc S s g | ]}|d d� �qS )r Nr r r r r r � r )r r rA 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� ri r� r r r � r 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 z,PrettyTable._format_rows.<locals>.<listcomp>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 fv r�|�| j� |D ]}|�| � ||�� q�|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 r8 r# r� r r$ r )r� ro r r r �_stringify_hrule�_hruler� �_stringify_headerrO rM �_stringify_rowrY �join)r` ra r� r r �formatted_rowsr} r r r r� � s" zPrettyTable.get_stringc C s� |d s dS | � |�\}}|d ttfv r6|d g}n |d g}| js^|�|d � d�|�S t| j| j�D ]b\}}|d r�||d vr�ql|�|| | |d � |d tkr�|�|d � ql|�|d � ql|d tkr�|�� |�|d � d�|�S )Nr# r� r% r6 r4 r ) r rO rM r= r� r r� rC r� )r` r� r r r� r� r r r r r � s( z PrettyTable._stringify_hrulec C s� g }| � |�\}}|d rj|d ttfv r@|�| j� |�d� |d ttfv r`|�|d � n |�d� | js�|d ttfv r�|�|d � n |�d� t| j| j�D ]�\}}|d r�||d vr�q�| jdkr�|� � }nD| jd kr�|� � }n0| jd k�r|� � }n| jd k�r|� � }n|}|�d| | � ||| j| � d| � |d r�|d tk�rn|�|d � q�|�d� q�|d �r�|d tk�r�|�� |�|d � |d �r�|d tk�r�|�d� |�| j� d �|�S ) Nr# r$ r r% r2 rk r r� r� r� r� r� )r rO rM r� r r= r� rC rK � capitalizer� r� r� rn r> r� r� r )r` r� r� r r r� r r� r r r r sJ * z PrettyTable._stringify_headerc C s� t tdt|��| j|| j�D ]\\}}}}|�d�}g }|D ]&} t| �|krVt�| |�} |� | � q:|}d� |�}|||< q d} |D ] } t | �d } | | kr�| } q�g } | � |�\}}td| �D ]F}| � g � |d r�|d t tfv r�| | � | j� q�| | � d� q�t | j|| j�D �],\}}}| j| }|�d�}| t|� }|�r�|dk�r~dgt|d � | dg|t|d � }n(|d k�r�dg| | }n|dg| }d}|D ]�}|d �r�||d v�rΐq�| | � d| | �||| j| � d| � |d �r4|d t k�r&| | � | j� n| | � d� |d7 }�q��qtd| �D ]<}|d �rN|d tk�rN| | �� | | � |d � �qN|d �r�|d t k�r�| | d � d� | | d � | j� td| �D ]}d� | | �| |< �q�d� | �S )Nr r r r# r% rk r� r� r r� r r2 r$ )r� r� r r= rC r r �textwrap�fillr� r r r rO rM r2 r? rx rn r> r� r )r` r} r� rz r� ri r r Z new_linesr Z row_heightr� �hr� r r �yr� ZdHeightrj 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 falser9 )r� �_get_formatted_html_string�_get_simple_html_string)r` ra r� �stringr r r �get_html_string� s zPrettyTable.get_html_stringc C s^ g }|d rd}nd}g }|� d� |d rT|d D ] }|� d||d | f � q4|� d� |� d�|�� |d r�|� d � | jD ]4}|d r�||d vr�q�|� d t|��d |� � q�|� d� | �|�}| �||�}|D ]b} |� d � t| j| �D ]<\}} |d �r ||d v�r q�|� dt| ��d |� � q�|� d� q�|� d� | �d ��|�S )Nr: �<br/>�<br>�<tabler; � %s="%s"�>r� r � <tr>r z <th>%s</th>r � </tr>z <td>%s</td>�</table>) r� r r= r �replacer r r� rY ) 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<|d D ] }|�d||d | f � �q|�d� |�d�|�� |d �r�|�d� | jD ]B}|d �r�||d v�r��qp|�d||t|��d|�f � �qp|�d� | �|�} | � | |�} g } g } | jD ]>}| �dddd �| j | � | �d d d d �| j | � �q�| D ]z} |�d� t | j| | | �D ]N\}}}}|d �rl||d v�rl�qD|�d!||||t|��d|�f � �qD|�d� �q&|�d"� | � d��|�S )#Nr: r& r' r( r# 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"r; r) r* r� r r+ r zW <th style="padding-left: %dem; padding-right: %dem; text-align: center">%s</th>r r, �left�right�center)rj rl r� �topZmiddleZbottom)r� r� r� zg <td style="padding-left: %dem; padding-right: %dem; text-align: %s; vertical-align: %s">%s</td>r- )r r� rO rM r r= r r. r r r> r? r� rY )r` r� r r r r/ r0 r1 r� r r ZalignsZvalignsr} r2 rm r� r r r r" � sd $ ( z&PrettyTable._get_formatted_html_string)N)r� r� )��__name__� __module__� __qualname__rc rY rn rr r~ �py3kr� r rF r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �propertyrB r� r� rm 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� r1 r� r� r2 r� r� r4 r� r� r6 r� r� r9 r� r� r8 r� r� r; 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 | dv r�dS | dv r�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& .rD c C s t ttttt�d| ����S )Nr� )�sum�itermaprD �ord�_re�sub)r� r r r r / s r c K s� t �� �| �d��}| �d� t �| |�}tf i |��}|rD||_n.tr^dd� t |�D �|_ndd� |� � D �|_|D ]}|� dd� |D �� qv|S )Ni r c S s g | ] }|� � �qS r ��stripr� r r r r B r z from_csv.<locals>.<listcomp>c S s g | ] }|� � �qS r rJ r� r r r r D r c S s g | ] }|� � �qS r rJ r� r r r r G r ) �csvZSnifferZsniff�read�seek�readerr rB r: �nextrw )�fprB ra ZdialectrO �tabler} r r r �from_csv7 s rS c K sF | j rBtf i |��}dd� | j D �|_| �� D ]}|�|� q.|S d S )Nc S s g | ] }|d �qS r r )r �colr r r r O r z"from_db_cursor.<locals>.<listcomp>)� descriptionr rB Zfetchallrw )Zcursorra rR r} r r r �from_db_cursorK s rV 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 rc ra �tables�last_rowr � max_row_width�active� last_content�is_last_row_header)r` ra r r r rc 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� |dv r | j �� }| j�|� |dkrZ| j�| j| jf� t| jt| j��| _g | _d| _|dkr�| � | j�}| j �|� g | _d| _ d | _ d S )N)r^ Ztd�trFrR rk ) r\ rK rY r� r r] r rZ r �generate_tablerX r[ )r` r_ Zstripped_contentrR r r r � handle_endtagf s � zTableHandler.handle_endtagc C s | j |7 _ d S r) )r\ )r` �datar r r � handle_datax s zTableHandler.handle_datac C s� t f i | j��}| jD ]z}t|d �| jk r\| jt|d � }td|�D ]}|d �d� qH|d dkr�| �|d � |d |_q|� |d � q|S )zE Generates from a list of rows a PrettyTable object. r r r5 T) r ra r r rZ r� r� �make_fields_uniquerB rw )r` r rR r} Zappendsr� r r r rc { s zTableHandler.generate_tablec C sP t dt|��D ]<}t |d t|��D ]$}|| || kr$|| d7 < q$qdS )zB iterates over the row and make each field unique r r �'N)r� r )r` r r� �jr r r rg � s z TableHandler.make_fields_uniqueN) r7 r8 r9 rc ra rd rf rc rg r r r r rW T s rW c K s t f i |��}|�| � |jS )z� Generates a list of PrettyTables from a string of HTML code. Each <table> in the HTML becomes one PrettyTable object. )rW ZfeedrX )� html_codera �parserr r r � from_html� s rl c K sH t | fi |��}zt|�dks"J �W n ty> td��Y n0 |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 )rl r r� ry )rj ra rX r r r � from_html_one� s rm c C s� t g d��} d| _d| _d| jd< d| _d| jd< | �g d �� | �g d �� | �g d �� | �g d �� | �g d �� | �g d�� | �g d�� t| � d S )N)� City name�Area� PopulationzAnnual Rainfallrp TZ04dro z6.1frj rn )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. rm rw �printr* r r r �main� s rs �__main__)N)2� __version__r rL r� �re�sysr � itertoolsrA � version_infor: rg rh rf �maprF r� Ziterzip�chrrB Z html.parserr �imapZizipZunichrZhtmlr ZcgirM rO r� r� r� r� r� r� �compilerH r �objectr rD r rS rV rW rl rm rs r7 r r r r �<module> sj E! B