a
R�f? � @ s d dl Z d dlmZmZmZmZmZmZmZm Z m
Z
m Z m Z m
Z
mZmZ d dlmZ d dlmZ g d�ZG dd� d�ZG dd � d e�Zed e� G d
d � d e�Zed e� G d
d� de�Zede� G dd� d�ZG dd� d�Zze W n ey� e ZY n0 G dd� d�Z dS )� N)�Error� __version__�writer�reader�register_dialect�unregister_dialect� get_dialect�
list_dialects�field_size_limit�
QUOTE_MINIMAL� QUOTE_ALL�QUOTE_NONNUMERIC�
QUOTE_NONE�__doc__)�Dialect)�StringIO)r r r
r r r r �excel� excel_tabr
r r r r r �Snifferr r �
DictReader�
DictWriter� unix_dialectc @ s@ e Zd ZdZdZdZdZdZdZdZ dZ
dZ dd� Z dd� Z
dS )r � FNc C s | j tkrd| _| �� d S )NT)� __class__r �_valid� _validate��self� r �(/opt/alt/python39/lib64/python3.9/csv.py�__init__* s
zDialect.__init__c
C sB z t | � W n0 ty< } ztt|���W Y d }~n
d }~0 0 d S �N)�_Dialect� TypeErrorr �str)r �er r r r / s zDialect._validate)�__name__�
__module__� __qualname__�_namer � delimiter� quotecharZ
escapechar� doublequote�skipinitialspace�lineterminator�quotingr r r r r r r s r c @ s$ e Zd ZdZdZdZdZdZeZ dS )r �,�"TF�
N)
r& r' r( r* r+ r, r- r. r r/ r r r r r 6 s r c @ s e Zd ZdZdS )r � N)r& r' r( r* r r r r r @ s r z excel-tabc @ s$ e Zd ZdZdZdZdZdZeZ dS )r r0 r1 TF�
N)
r& r' r( r* r+ r, r- r. r r/ r r r r r E s r Zunixc @ s@ e Zd Zd dd�Zdd� Zedd� �Zejd d� �Zd
d � ZdS )
r Nr c O s< || _ || _|| _t||g|�R i |��| _|| _d| _d S �Nr )� _fieldnames�restkey�restvalr �dialect�line_num)r �f�
fieldnamesr7 r8 r9 �args�kwdsr r r r Q s zDictReader.__init__c C s | S r! r r r r r �__iter__Z s zDictReader.__iter__c C s> | j d u r.zt| j�| _ W n ty, Y n0 | jj| _| j S r! )r6 �nextr �
StopIterationr: r r r r r< ] s
zDictReader.fieldnamesc C s
|| _ d S r! )r6 )r �valuer r r r< g s c C s� | j dkr| j t| j�}| jj | _ |g kr8t| j�}q$tt| j|��}t| j�}t|�}||k rv||d � || j< n&||kr�| j|d � D ]}| j||< q�|S r5 ) r: r< r@ r �dict�zip�lenr7 r8 )r �row�dZlfZlr�keyr r r �__next__k s
zDictReader.__next__)NNNr ) r& r' r( r r? �propertyr< �setterrI r r r r r P s �
r c @ s6 e Zd Zddd�Zdd� Zdd � Zd
d � Zd d
� ZdS )r r �raiser c O sH || _ || _|�� dvr$td| ��|| _t||g|�R i |��| _d S )N)rL �ignorez-extrasaction (%s) must be 'raise' or 'ignore')r< r8 �lower�
ValueError� extrasactionr )r r; r< r8 rP r9 r= r> r r r r � s �zDictWriter.__init__c C s t t| j| j��}| �|�S r! )rC rD r< �writerow)r �headerr r r � writeheader� s zDictWriter.writeheaderc sN �j dkr8� �� �j }|r8tdd�dd� |D �� ��� �fdd��jD �S )NrL z(dict contains fields not in fieldnames: z, c S s g | ] }t |��qS r )�repr)�.0�xr r r �
<listcomp>� � z,DictWriter._dict_to_list.<locals>.<listcomp>c 3 s | ]}� � |�j�V qd S r! )�getr8 )rU rH ��rowdictr r r � <genexpr>� rX z+DictWriter._dict_to_list.<locals>.<genexpr>)rP �keysr< rO �join)r r[ Z wrong_fieldsr rZ r �
_dict_to_list� s
�zDictWriter._dict_to_listc C s | j �| �|��S r! )r rQ r_ )r r[ r r r rQ � s zDictWriter.writerowc C s | j �t| j|��S r! )r � writerows�mapr_ )r Zrowdictsr r r r` � s zDictWriter.writerowsN)r rL r )r& r' r( r rS r_ rQ r` r r r r r � s �
r c @ s6 e Zd Zdd� Zd dd�Zdd� Zdd � Zd
d � ZdS )
r c C s g d�| _ d S )N)r0 r3 �;� �:)� preferredr r r r r � s zSniffer.__init__Nc C sd | � ||�\}}}}|s(| �||�\}}|s4td��G dd� dt�}||_||_|pVd|_||_|S )Nz Could not determine delimiterc @ s e Zd ZdZdZeZdS )z Sniffer.sniff.<locals>.dialectZsniffedr2 N)r& r' r( r) r. r r/ r r r r r9 � s r9 r1 )�_guess_quote_and_delimiter�_guess_delimiterr r r, r* r+ r- )r �sample�
delimitersr+ r, r* r- r9 r r r �sniff� s
�
�
z
Sniffer.sniffc C s� g }dD ]*}t �|t jt jB �}|�|�}|r q4q|s<dS i }i }d}|j} |D ]�}
| d d } |
| } | r�|�| d�d || < z| d d } |
| } W n ty� Y qRY n0 | r�|d u s�| |v r�|�| d�d || < z| d d } W n t�y Y qRY n0 |
| rR|d7 }qRt||jd�}
|�rTt||jd�}|| |k}|d k�r\d
}nd
}d}t �d t � |�|
d � t j�}|�
|��r�d
}nd}|
|||fS )N)zI(?P<delim>[^\w\n"\'])(?P<space> ?)(?P<quote>["\']).*?(?P=quote)(?P=delim)zG(?:^|\n)(?P<quote>["\']).*?(?P=quote)(?P<delim>[^\w\n"\'])(?P<space> ?)zG(?P<delim>[^\w\n"\'])(?P<space> ?)(?P<quote>["\']).*?(?P=quote)(?:$|\n)z-(?:^|\n)(?P<quote>["\']).*?(?P=quote)(?:$|\n))r FNr r �quote� �delimZspace�rH r4 r z]((%(delim)s)|^)\W*%(quote)s[^%(delim)s\n]*%(quote)s[^%(delim)s\n]*%(quote)s\W*((%(delim)s)|$))rm rk TF) �re�compile�DOTALL� MULTILINE�findall�
groupindexrY �KeyError�max�escape�search)r �datari ZmatchesZrestrZregexpZquotes�delims�spacesrt �m�nrH r+ rm r- Z dq_regexpr, r r r rf � s`
�� z"Sniffer._guess_quote_and_delimiterc C s t td |�d���}dd� td�D �}tdt|��}d}i }i }i }d| } }
| t|�k �rR|d7 }|| |
� D ]@} |D ]6} |�| i �}
| �| �}|
�|d�d |
|< |
|| < qxqp|�� D ]�} t || � � �}t|�dkr�|d d dkr�q�t|�dk�rLt
|dd � d
�|| < |� || � || d || d t d d � |D �� f|| < q�|d || < q�|� � }t
t|| t|���}d
}d}t|�dk�r�||k�r�|D ]T\}}|d dk�r�|d dk�r�|d | |k�r�|d u �s�||v �r�|||< �q�|d8 }�q�t|�dk�rDt |�� �d }|d �|�|d �d| �k}||fS |
} |
|7 }
qN|�s\dS t|�dk�r�| jD ]@}||�� v �rp|d �|�|d �d| �k}||f S �qpdd� |� � D �}|�� |d d }|d �|�|d �d| �k}||fS )Nr4 c S s g | ] }t |��qS r )�chr)rU �cr r r rW - rX z,Sniffer._guess_delimiter.<locals>.<listcomp>� �
r rl c S s | d S )Nrl r )rV r r r �<lambda>G rX z*Sniffer._guess_delimiter.<locals>.<lambda>rn c s s | ]}|d V qdS )rl Nr )rU �itemr r r r\ L rX z+Sniffer._guess_delimiter.<locals>.<genexpr>g �?g�������?g{�G�z�?z%c )r r c S s g | ]\}}||f�qS r r )rU �k�vr r r rW v rX ���)�list�filter�split�range�minrE rY �countr] �itemsrv �remove�sum�floatre �sort)r ry ri �asciiZ chunkLengthZ iterationZ
charFrequencyZmodesrz �start�end�line�charZ
metaFrequencyZfreqr� ZmodeList�totalZ consistencyZ thresholdr� r� rm r- rG r r r rg s�
� �� �
� �zSniffer._guess_delimiterc
C s� t t|�| �|��}t|�}t|�}i }t|�D ] }d ||< q0d}|D ]�}|dkrV q�|d7 }t|�|krlqFt|�� �D ]v} tt t
fD ]2}
z|
|| � W q�W q� t t fy� Y q�0 q�t|| �}
|
|| krx|| d u r�|
|| < qx|| = qxqFd} |�
� D ]z\} } t| �td�k�r>t|| �| k�r4| d7 } n| d8 } q�z| || � W n t tf�yn | d7 } Y q�0 | d8 } q�| dkS )Nr � rl )r r rj r@ rE r� r� r] �intr� �complexrO �
OverflowErrorr� �typer# )
r rh ZrdrrR �columnsZ columnTypes�i�checkedrF �colZthisTypeZ hasHeaderZcolTyper r r �
has_header sH
zSniffer.has_header)N)r& r' r( r rj rf rg r� r r r r r � s
Lgr ) ro Z_csvr r r r r r r r r
r r r
r r r r" �ior �__all__r r r r r r� � NameErrorr� r r r r r �<module> s$ @
2