File: //usr/lib/python3/dist-packages/docutils/parsers/rst/directives/__pycache__/tables.cpython-310.pyc
o
�?�] Y � @ 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.
�reStructuredText� N)�io�nodes� statemachine�utils)�
SafeString)�SystemMessagePropagation)� Directive)�
directivesc C s t �| d�S )N)�left�center�right)r
�choice)�argument� r �H/usr/lib/python3/dist-packages/docutils/parsers/rst/directives/tables.py�align s r c @ sx e Zd ZdZdZdZejeje ej
e� dej �d�Z
dZdd� Zdd � Zd
d � Zd d
� Zedd� �Zdd� Zdd� ZdS )�Tablez#
Generic table base class.
� T)�auto�grid)�class�namer �width�widthsc C sf | j r+| j d }| j�|| j�\}}tj|dg|�R � }| j�| j�\|_|_ ||fS d }g }||fS )Nr � )
� arguments�state� inline_text�linenor �title�
state_machine�get_source_and_line�source�line)�self�
title_text�
text_nodes�messagesr r r r �
make_title+ s
� ��zTable.make_titlec C sX | j �| jd �}g }d}d| jv r(| �| jd �d�| �� |�\}}|�|� ||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_option8 s
�
zTable.process_header_optionc C sB t |�|k r#| jjjd|t |�| jf t�| j| j�| jd�}t |��t |�| kr/dkrMn n | jjjdt |�| jf t�| j| j�| jd�}t |��|D ]O}t |�|k rt| jjjd|t |�| jf t�| j| j�| jd�}t |��t |�| kr�dkr�n qO| jjjdt |�| jf t�| j| j�| jd�}t |��qOd 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% r4 � header_rows� stub_columnsr9 �rowr r r �check_table_dimensionsC s^ ��� ��� ��� �����z Table.check_table_dimensionsc C s" d| j v r| j �d�|d< d S d S )Nr �r- �get)r% �
table_noder r r �set_table_widthb s
�zTable.set_table_widthc C s | j �dd�S )Nr r r@ �r% r r r r f s z Table.widthsc C s� t | jt�r,t| j�|kr'| jjjd| j|f t� | j
| j
�| j d�}t |��| j}|S |r7d| g| }|S | jjjdt� | j
| j
�| j d�}t |��)Nz="%s" widths do not match the number of columns in table (%s).r6 �d z#No table data detected in CSV file.)
�
isinstancer �listr7 r! r8 r9 r r r: r; r r )r% �max_colsr9 �
col_widthsr r r �get_column_widthsj s0 �������zTable.get_column_widthsc C sD |D ] }|D ]}t |�|k r |�dddg fg|t |� � qqd S )Nr )r7 r1 )r% �columns�parts�partr> r r r �"extend_short_rows_with_empty_cells| s ���z(Table.extend_short_rows_with_empty_cellsN)�__name__�
__module__� __qualname__�__doc__�optional_arguments�final_argument_whitespacer
� class_option� unchangedr � length_or_percentage_or_unitless�value_or�positive_int_list� option_spec� has_contentr) r5 r? rC �propertyr rJ rN r r r r r s* ��
r c @ s e Zd Zdd� ZdS )�RSTTablec C s� | j s| jjjd| j t�| j| j�| jd�}|gS | � � \}}t�
� }| j � | j | j
|� t|�dks<t|d tj�sS| jjjd| j t�| j| j�| jd�}|gS |d }|d | j�dg �7 < | �|� d| jv rv| j�d�|d< |d }t| jt�r�d d
� |jD �}t|| j�D ]\} }
|
| d < q�| jd kr�|d d
g7 < n | jr�|d dg7 < | �|� |r�|�d|� |g| S )Nz:Content block expected for the "%s" directive; none found.r6 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)�tagname)�.0�childr r r �
<listcomp>� s
�z RSTTable.run.<locals>.<listcomp>�colwidthr �colwidths-auto�colwidths-given)�contentr! r8 �warningr r r: r; r r) �Elementr � nested_parse�content_offsetr7 rF �tabler9 r- rA rC r rG �children�zip�add_name�insert) r% rh r r( �noder9 rB �tgroup�colspecsr_ � col_widthr r r �run� sV ��� ���
z RSTTable.runN)rO rP rQ ru r r r r r] � s r] c @ s� e Zd Zejejejeje�dej�ej ej
ej ej eje
ejejejejd�ZG dd� dej�ZG dd� dej�Zdd� Zd d
� Zd d � Zejd
k rTdd� Zdd� Z ndd� Zdd� Z e e�Ze e �Z dd� Z dS )�CSVTable�r )� header-rows� stub-columnsr* r 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|v r t �|d �| _d|v rd| _d|v r t �|d �| _d|v r.d| _t �|d �| _tj� | � d S )Nr} r~ Fr r� )
rv �encode_for_csv� delimiter�skipinitialspace� quotechar� doublequote�
escapechar�csv�Dialect�__init__)r% r- r r r r� � s z!CSVTable.DocutilsDialect.__init__N)rO rP rQ rR r� r� r� r� �strict�lineterminatorr� �
QUOTE_MINIMAL�quotingr� r r r r �DocutilsDialect� s r� c @ s2 e Zd ZdZdZdZdZdZdZdZ dZ
e j Z
dS ) zCSVTable.HeaderDialectz0CSV dialect to use for the "header" option data.r� r� �\FTr+ N)rO rP rQ rR r� r� r� r� r� r� r� r� r� r� r r r r r0 � s
r0 c C s d S �Nr rD r r r �check_requirements� � zCSVTable.check_requirementsc
C s8 z�| j jjjs)d| jv sd| jv r)| jjjd| j t �
| j | j �| j d�}|gW S | �
� | �� \}}| �� \}}| �� \}}| �|| �| j�|�\}} t| |�} | j�dd�}
| j�dd�} | �||
| � |�|d |
� � ||
d � } | �| �}
| �| || f� W nT ty� } z |jd gW Y d }~S d }~w tj y� } z0t |�}t j dk r�d |v r�|d
7 }| jjj d | j|f t �
| j | j �| j d�}|gW Y d }~S d }~ww |
|| f}| j j!|| j"| | j#d �}|d
| j�dg �7 < d| jv �r| j�d�|d< | �$|� | �%|� |�r|�&d|� |g| S )Nrz r{ z9File and URL access deactivated; ignoring "%s" directive.r6 rx r ry �� r z1-character stringz1
with Python 2.x this must be an ASCII character.z)Error with CSV data in "%s" directive:
%s)r r^ r r )'r �document�settings�file_insertion_enabledr- r! r8 rh r r r: r; r r� r) � get_csv_datar5 r. r� �maxrA r? r1 rJ rN r �argsr� �Error�str�sys� version_infor9 � build_tablerk r rC ro rp )r% rh r r( �csv_datar# r2 r3 r4 rH r< r= �
table_bodyrI �detail�messager9 rl rB r r r ru � s|
��� �
�������
�
z CSVTable.runc C s� | j �d| jjjj�}| jjjj}| jrCd| j v s d| j v r6| jj j
d| j t �
| j| j�| jd�}t|��| j�d�}| j}||fS d| j v r�d| j v re| jj j
d| j t �
| j| j�| jd�}t|��tj�tj�| jjj��}tj�tj�|| j d ��}t�d|�}z | jjjj� |� t j |||d �}|� � � � }W ||fS t!y� } z | jj j"d
| j t#|�f t �
| j| j�| jd�}t|��d}~ww d| j v �rAt$j%d kr�dd l&m'} dd
l(m)}
nddl*m'} m)}
| j d }z| |�� � } W n1 |
t!t+t,f�y) } z | jj j"d| j | j d t#|�f t �
| j| j�| jd�}t|��d}~ww t j-| ||| jjjjd�}|� � � � }||fS | jj j.d| j t �
| j| j�| jd�}t|��)zr
Get CSV data from the directive content, from an external
file, or from a URL reference.
r| rz r{ zF"%s" directive may not both specify an external file and have content.r6 r zXThe "file" and "url" options may not be simultaneously specified for the "%s" directive.N)� source_pathr| �
error_handlerz&Problems with "%s" directive path:
%s.r� )�urlopen)�URLError)r� r� z*Problems with "%s" directive URL "%s":
%s.)r# r� r| r� z3The "%s" directive requires content; none supplied.)/r- rA r r� r� �input_encoding� input_encoding_error_handlerrg r! r8 r9 r r r: r; r r r# �os�path�dirname�abspath�current_source�normpath�joinr �
relative_path�record_dependencies�addr � FileInput�read�
splitlines�IOError�severer r� r� �urllib.requestr� � urllib.errorr� �urllib2�OSError�
ValueError� StringInputrh ) r% r| r� r9 r# r� �
source_dir�csv_filer� r� r� �csv_textr r r r� s� � ��� <
�
���� �(� ����
����� ����zCSVTable.get_csv_datar� c C �
| � d�S �Nzutf-8)�decode��sr r r �decode_from_csvn �
zCSVTable.decode_from_csvc C r� r� )�encoder� r r r r� p r� zCSVTable.encode_for_csvc C � | S r� r r� r r r r� s r� c C r� r� r r� r r r r� u r� c s� t j� fdd�|D �|d�}g }d}|D ],}g }|D ]} � �| �}
dddtj|
�� |d�f} |�| � q|�|� t|t|��}q||fS )Nc s g | ] }� � |d ��qS )r+ )r� )ra r$ rD r r rc | s �z5CSVTable.parse_csv_data_into_rows.<locals>.<listcomp>)�dialectr )r# ) r� �readerr� r �
StringListr� �appendr� r7 ) r% r� r� r# �
csv_readerr4 rH r> �row_data�cell� cell_text� cell_datar rD r r. z s$ ��
�
z!CSVTable.parse_csv_data_into_rowsN) rO rP rQ r
�nonnegative_intrV rW rX rY r� �urir| rU r �$single_char_or_whitespace_or_unicode�flag�single_char_or_unicoderZ r� r� r� r0 r� ru r� r� r� r� r� � staticmethodr. r r r r rv � s@ ��
0
O
rv c @ sP e Zd ZdZejejeje�dej�ej ej
e d�Z dd� Z
dd� Zdd � 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
rw )rx ry r r r r r c
C sN | j s| jjjd| j t�| j| j�| jd�}|gS | � � \}}t�
� }| j � | j | j
|� z'| �|�\}}dd� |d D �}| j�dd�}| j�dd�} | �||| � W n tym }
z |
jd gW Y d }
~
S d }
~
ww | �|||| �} d| jv r�| j�d�| d< | d | j�d
g �7 < | �| � | �| � |r�| �d|� | g| S ) Nz.The "%s" directive is empty; content required.r6 c S s g | ] }d d� |d D ��qS )c S s g | ]}|j �qS r )rm )ra �itemr r r rc � s z,ListTable.run.<locals>.<listcomp>.<listcomp>r r )ra �row_listr r r rc � s �z!ListTable.run.<locals>.<listcomp>r rx ry r r^ r )rg r! r8 r9 r r r: r; r r) ri r rj rk �check_list_contentr- rA r? r r� �build_table_from_listrC ro rp ) r% r9 r r( rq �num_colsrI �
table_datar< r= r� rB r r r ru � sB � ����
z
ListTable.runc C s> t |�dkst|d tj�s&| jjjd| j t�| j | j �| j
d�}t |��|d }d}t t |��D ]c}|| }t |�dksFt|d tj�sb| jjjd| j|d f t�| j | j �| j
d�}t |��|r�t |d �|kr�| jjjd| j|d t |d �|f t�| j | j �| j
d�}t |��q2t |d �}q2| �
|�}||fS )Nr r zUError parsing content block for the "%s" directive: exactly one bullet list expected.r6 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).)r7 rF r � bullet_listr! r8 r9 r r: r; r r �rangerJ )r% rq r9 � list_noder� �
item_indexr� rI r r r r� � sL �� ������
z ListTable.check_list_contentc C s& t �� }| jdkr|d dg7 < n | jr |d dg7 < t jt|�d�}||7 }|D ] }t �� }|d ur<||jd< |rGd|jd< |d8 }||7 }q-g } |D ] }
t �� } |
D ]} t �� }
|
| 7 }
| |
7 } qX| � | � qP|r�t �
� }|� | d |� � ||7 }t � � }|� | |d � � ||7 }|S ) Nr r^ re rf )�colsrd r �stub)
r rl r rr r7 r_ �
attributesr> �entryr� �theadr1 �tbody)r% r� rI r< r= rl rr rt r_ r4 r> �row_noder� r� r� r� r r r r� � s>
z ListTable.build_table_from_listN)rO rP rQ rR r
r� rW rX rY rU rV r rZ ru r� r� r r r r r� � s �� %r� )rR �
__docformat__r� �os.pathr� r� �docutilsr r r r � docutils.utils.error_reportingr �docutils.utilsr �docutils.parsers.rstr r
r r r] rv r� r r r r �<module> s h* b