File: //lib/python3.6/site-packages/docutils/parsers/rst/directives/__pycache__/tables.cpython-36.pyc
3
v2�eXW � @ s� d Z dZddlZddlZddlZddlmZmZm Z m
Z
ddl m Z ddl
mZ ddlmZ ddlmZ d d
� ZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�ZdS )z
Directives for table elements.
ZreStructuredText� N)�io�nodes� statemachine�utils)�
SafeString)�SystemMessagePropagation)� Directive)�
directivesc C s t j| d�S )N�left�center�right)r
r r )r Zchoice)Zargument� r
� /usr/lib/python3.6/tables.py�align s r c @ sl e Zd ZdZdZdZejeje ej
dej �d�Z dZ
dd� Zd d
� Zd d � Zed
d� �Zdd� Zdd� ZdS )�Tablez#
Generic table base class.
� T�auto�grid)�class�namer �widthsc C s^ | j rN| j d }| jj|| j�\}}tj|df|�� }| jj| j�\|_|_ nd }g }||fS )Nr � )
Z arguments�stateZ inline_text�linenor �title�
state_machineZget_source_and_line�source�line)�selfZ
title_textZ
text_nodes�messagesr r
r
r �
make_title* s
zTable.make_titlec C sX | j j| jd �}g }d}d| jkrP| j| jd jd�| j� |�\}}|j|� ||fS )Nr r �header�
)r �
get_sourcer �options�parse_csv_data_into_rows�split�
HeaderDialect�extend)r r �
table_head�max_header_cols�rowsr
r
r �process_header_option7 s
zTable.process_header_optionc C sH t |�|k rF| jjjd|t |�| jf tj| j| j�| jd�}t |��t |�| ko\dkn r�| jjjdt |�| jf tj| j| j�| jd�}t |��x�|D ]�}t |�|k r�| jjjd|t |�| jf tj| j| j�| jd�}t |��t |�| k�odkn r�| jjjdt |�| jf tj| j| j�| jd�}t |��q�W d S )NzP%s header row(s) specified but only %s row(s) of data supplied ("%s" directive).)r r zeInsufficient data supplied (%s row(s)); no data remaining for table body, required by "%s" directive.zU%s stub column(s) specified but only %s columns(s) of data supplied ("%s" directive).ziInsufficient data supplied (%s columns(s)); no data remaining for table body, required by "%s" directive.)
�lenr �reporter�errorr r �
literal_block�
block_textr r )r r+ � header_rows� stub_columnsr/ �rowr
r
r �check_table_dimensionsB s2
z Table.check_table_dimensionsc C s | j jdd�S )Nr r )r$ �get)r r
r
r r a s z Table.widthsc C s� t | j�tkrXt| j�|krP| jjjd| j|f tj | j
| j
�| j d�}t |��| j}n>|rld| g| }n*| jjjdtj | j
| j
�| j d�}t |��|S )Nz="%s" widths do not match the number of columns in table (%s).)r �d z#No table data detected in CSV file.)
�typer �listr- r r. r/ r r r0 r1 r r )r �max_colsr/ �
col_widthsr
r
r �get_column_widthse s zTable.get_column_widthsc C sL xF|D ]>}x8|D ]0}t |�|k r|jdddg fg|t |� � qW qW d S )Nr )r- r( )r �columns�parts�partr4 r
r
r �"extend_short_rows_with_empty_cellsw s
z(Table.extend_short_rows_with_empty_cellsN)r r )�__name__�
__module__� __qualname__�__doc__Zoptional_argumentsZfinal_argument_whitespacer � class_option� unchangedr �value_or�positive_int_list� option_specZ has_contentr r, r5 �propertyr r< r@ r
r
r
r r s
r c @ s e Zd Zdd� ZdS )�RSTTablec C s� | j s4| jjjd| j tj| j| j�| jd�}|gS | j � \}}tj
� }| j j | j | j
|� t|�dkszt|d tj� r�| jjjd| j tj| j| j�| jd�}|gS |d }|d | jjdg �7 < d| jkr�| jjd�|d< |d }t| j�tk�r0d d
� |jD �}x"t|| j�D ]\} }
|
| d < �qW | jd k�rP|d d
g7 < n| j�rj|d dg7 < | j|� |�r�|jd|� |g| S )Nz:Content block expected for the "%s" directive; none found.)r r r zOError parsing content block for the "%s" directive: exactly one table expected.�classesr r c S s g | ]}|j d kr|�qS )�colspec)Ztagname)�.0Zchildr
r
r �
<listcomp>� s z RSTTable.run.<locals>.<listcomp>�colwidthr zcolwidths-autozcolwidths-given)�contentr r. �warningr r r0 r1 r r �Elementr � nested_parse�content_offsetr- �
isinstance�tabler/ r$ r6 r8 r r9 �children�zip�add_name�insert) r rR r r �noder/ �
table_node�tgroupZcolspecsrM � col_widthr
r
r �run� s@
z RSTTable.runN)rA rB rC r` r
r
r
r rK ~ s rK c @ s� e Zd Zejejejejdej�ejej ej
ej eje ej
ejejejd�ZG dd� dej�ZG dd� dej�Zdd� Zd d
� Zd d � Zejdk r�dd� Zdd� Zndd� Zdd� Ze e�Ze e�Zdd� Z dS )�CSVTabler )z header-rowsz stub-columnsr! r �file�url�encodingr r r �delim� keepspace�quote�escapec @ s6 e Zd ZdZdZdZdZdZdZdZ e
j Z dd� Z
dS ) zCSVTable.DocutilsDialectz&CSV dialect for `csv_table` directive.�,�"Tr" c C sl d|krt j|d �| _d|kr&d| _d|kr>t j|d �| _d|kr\d| _t j|d �| _tjj | � d S )Nre rf Frg rh )
ra �encode_for_csv� delimiter�skipinitialspace� quotechar� doublequote�
escapechar�csv�Dialect�__init__)r r$ r
r
r rs � s z!CSVTable.DocutilsDialect.__init__N)rA rB rC rD rl rn ro rm �strict�lineterminatorrq �
QUOTE_MINIMAL�quotingrs r
r
r
r �DocutilsDialect� s rx c @ s2 e Zd ZdZdZdZdZdZdZdZ dZ
e j Z
dS ) zCSVTable.HeaderDialectz0CSV dialect to use for the "header" option data.ri rj �\FTr" N)rA rB rC rD rl rn rp ro rm rt ru rq rv rw r
r
r
r r' � s r' c C s d S )Nr
)r r
r
r �check_requirements� s zCSVTable.check_requirementsc C s( �y| j jjj rTd| jks&d| jkrT| jjjd| j t j
| j | j �| j d�}|gS | j
� | j� \}}| j� \}}| j� \}}| j|| j| j�|�\}} t| |�} | jjdd�}
| jjdd�} | j||
| � |j|d |
� � ||
d � } | j| �}
| j| || f� W n� tk
�r2 } z |jd gS d }~X nx tj k
�r� } zXt |�}t j dk �rld |k�rl|d
7 }| jjj d | j|f t j
| j | j �| j d�}|gS d }~X nX |
|| f}| j j!|| j"| | j#d �}|d
| jjdg �7 < d| jk�r| jjd�|d< | j$|� |�r |j%d|� |g| S )Nrb rc z9File and URL access deactivated; ignoring "%s" directive.)r z header-rowsr z stub-columns� z1-character stringz1
with Python 2.x this must be an ASCII character.z)Error with CSV data in "%s" directive:
%s)r rL r r )r{ )&r �document�settingsZfile_insertion_enabledr$ r r. rR r r r0 r1 r rz r � get_csv_datar, r% rx �maxr6 r5 r( r< r@ r �argsrq �Error�str�sys� version_infor/ Z build_tablerU r rZ r[ )r rR r r �csv_datar r) r* r+ r: r2 r3 Z
table_bodyr; �detail�messager/ rW r] r
r
r r` � sZ
z CSVTable.runc C s� | j jd| jjjj�}| jjjj}| jr�d| j ks<d| j krl| jj j
d| j t j
| j| j�| jd�}t|��| jjd�}| j}�n(d| j k�r�d| j kr�| jj j
d| j t j
| j| j�| jd�}t|��tjjtjj| jjj��}tjjtjj|| j d ��}tjd|�}y2| jjjjj |� t j |||d �}|j � j � }W n\ t!k
�r� } z>| jj j"d
| j t#|�f t j
| j| j�| jd�}t|��W Y dd}~X nX �n d| j k�rzddl$} ddl%} ddl&} | j d }y| j'j(|�j � }
W np | j
j)t!t*t+fk
�rP } zF| jj j"d | j | j d t#|�f t j
| j| j�| jd�}t|��W Y dd}~X nX t j,|
||| jjjjd �}|j � j � }n0| jj j-d
| j t j
| j| j�| jd�}t|��||fS )zr
Get CSV data from the directive content, from an external
file, or from a URL reference.
rd rb rc zF"%s" directive may not both specify an external file and have content.)r r zXThe "file" and "url" options may not be simultaneously specified for the "%s" directive.N)� source_pathrd �
error_handlerz&Problems with "%s" directive path:
%s.z*Problems with "%s" directive URL "%s":
%s.)r r� rd r� z3The "%s" directive requires content; none supplied.).r$ r6 r r| r} Zinput_encodingZ input_encoding_error_handlerrQ r r. r/ r r r0 r1 r r r �os�path�dirname�abspathZcurrent_source�normpath�joinr Z
relative_pathZrecord_dependencies�addr Z FileInput�read�
splitlines�IOError�severer Zurllib.requestZ urllib.errorZ urllib.parseZrequestZurlopenZURLError�OSError�
ValueErrorZ StringInputrR ) r rd r� r/ r r� Z
source_dirZcsv_filer� ZurllibZcsv_textr
r
r r~ sx
zCSVTable.get_csv_datar{ c C s
| j d�S )Nzutf-8)�decode)�sr
r
r �decode_from_csva s zCSVTable.decode_from_csvc C s
| j d�S )Nzutf-8)�encode)r� r
r
r rk c s zCSVTable.encode_for_csvc C s | S )Nr
)r� r
r
r r� f s c C s | S )Nr
)r� r
r
r rk h s c s� t j� fdd�|D �|d�}g }d}xd|D ]\}g }x:|D ]2} � j| �}
dddtj|
j� |d�f} |j| � q8W |j|� t|t|��}q*W ||fS )Nc s g | ]}� j |d ��qS )r" )rk )rN r )r r
r rO o s z5CSVTable.parse_csv_data_into_rows.<locals>.<listcomp>)�dialectr )r ) rq �readerr� r Z
StringListr� �appendr r- ) r r� r� r Z
csv_readerr+ r: r4 Zrow_data�cellZ cell_textZ cell_datar
)r r r% m s
z!CSVTable.parse_csv_data_into_rowsN)r )r{ ) rA rB rC r �nonnegative_intrF rG rH r� Zurird rE r Z$single_char_or_whitespace_or_unicode�flagZsingle_char_or_unicoderI rq rr rx r' rz r` r~ r� r� r� rk � staticmethodr% r
r
r
r ra � s8
/J
ra c @ sL e Zd ZdZejejejd ej�ejej e
d�Z dd� Z dd� Z
dd � Zd
S ) � ListTablez�
Implement tables whose data is encoded as a uniform two-level bullet list.
For further ideas, see
http://docutils.sf.net/docs/dev/rst/alternatives.html#list-driven-tables
r )z header-rowsz stub-columnsr r r r c C s@ | j s4| jjjd| j tj| j| j�| jd�}|gS | j � \}}tj
� }| j j | j | j
|� yN| j|�\}}dd� |d D �}| jjdd�}| jjdd�} | j||| � W n( tk
r� }
z |
jd gS d }
~
X nX | j|||| �} d| jk�r | jjd�| d< | d | jjd
g �7 < | j| � |�r6| jd|� | g| S ) Nz.The "%s" directive is empty; content required.)r c S s g | ]}d d� |d D ��qS )c S s g | ]
}|j �qS r
)rX )rN �itemr
r
r rO � s z,ListTable.run.<locals>.<listcomp>.<listcomp>r r
)rN Zrow_listr
r
r rO � s z!ListTable.run.<locals>.<listcomp>r z header-rowsz stub-columnsr rL r )rQ r r. r/ r r r0 r1 r r rS r rT rU �check_list_contentr$ r6 r5 r r� �build_table_from_listrZ r[ ) r r/ r r r\ �num_colsr; �
table_datar2 r3 r� r] r
r
r r` � s6
z
ListTable.runc C sH t |�dks t|d tj� rN| jjjd| j tj| j | j �| j
d�}t |��|d }x�t t |��D ]�}|| }t |�dks�t|d tj� r�| jjjd| j|d f tj| j | j �| j
d�}t |��qd|�r&t |d �|k�r2| jjjd| j|d t |d �|f tj| j | j �| j
d�}t |��qdt |d �}qdW | j
|�}||fS )Nr r zUError parsing content block for the "%s" directive: exactly one bullet list expected.)r z�Error parsing content block for the "%s" directive: two-level bullet list expected, but row %s does not contain a second-level bullet list.z�Error parsing content block for the "%s" directive: uniform two-level bullet list expected, but row %s does not contain the same number of items as row 1 (%s vs %s).)r- rV r Z bullet_listr r. r/ r r0 r1 r r �ranger< )r r\ r/ Z list_nodeZ
item_indexr� r� r; r
r
r r� � s6
z ListTable.check_list_contentc C s4 t j� }| jdkr&|d dg7 < n| jr>|d dg7 < t jt|�d�}||7 }xD|D ]<}t j� }|d k rz||jd< |r�d|jd< |d8 }||7 }q\W g } xD|D ]<}
t j� } x$|
D ] } t j� }
|
| 7 }
| |
7 } q�W | j | � q�W |�rt j
� }|j | d |� � ||7 }t j � }|j | |d � � ||7 }|S ) Nr rL zcolwidths-autozcolwidths-given)ZcolsrP r Zstub)
r rW r r^ r- rM Z
attributesr4 �entryr� �theadr( �tbody)r r� r; r2 r3 rW r^ r_ rM r+ r4 Zrow_noder� r� r� r� r
r
r r� � s>
z ListTable.build_table_from_listN)r )rA rB rC rD r r� rG rH rE rF r rI r` r� r� r
r
r
r r� � s 'r� )rD Z
__docformat__r� Zos.pathr� rq Zdocutilsr r r r Z docutils.utils.error_reportingr Zdocutils.utilsr Zdocutils.parsers.rstr r r r rK ra r� r
r
r
r �<module> s c) [