a
���e�N � @ s� g d �Z ddlZddlZddlZddlZddlmZ ddlmZm Z ddl
m Z dZ dZ
e�d�Ze�d ej�ZG d
d � d �ZG d d
� d
e�ZdZG dd� de�Zeeejd ��Zde ZejZdS ))� Generator�DecodedGenerator�BytesGenerator� N)�deepcopy)�StringIO�BytesIO)�_has_surrogates�_�
z
\r\n|\r|\nz^From c @ s� e Zd Zd&dd�dd�Zdd� Zd'dd �Zd
d � Zd d
� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
e
Zdd� Zd d � Zd d � Zd d!� Zed(d"d#��Zed$d%� �ZdS ))r N��policyc C s6 |d u r|d u rdn|j }|| _|| _|| _|| _d S )NT)� mangle_from_�_fp�
_mangle_from_� maxheaderlenr )�self�outfpr
r r � r �4/opt/alt/python39/lib64/python3.9/email/generator.py�__init__$ s zGenerator.__init__c C s | j �|� d S �N)r �write�r �sr r r r D s zGenerator.writeFc C s� | j d u r|j n| j }|d ur*|j|d�}| jd urB|j| jd�}|j| _| �| j�| _d| _| �| j�| _| j }|j }zX|| _ ||_ |r�|� � }|s�dt
� t
�
� � }| � || j � | �
|� W || _ ||_ n|| _ ||_ 0 d S )N)�linesep�Zmax_line_length� z From nobody )r �cloner r �_NL�_encode� _encoded_NLZ_EMPTY�_encoded_EMPTYZ get_unixfrom�time�ctimer �_write)r �msg�unixfromr r Zold_gen_policyZold_msg_policyZufromr r r �flattenH s0
�zGenerator.flattenc C s | j || jd | jd�S �Nr )� __class__r r )r �fpr r r r y s
�zGenerator.clonec C s t � S r )r �r r r r � _new_buffer� s zGenerator._new_bufferc C s |S r r r r r r r � s zGenerator._encodec C sT |sd S t �|�}|d d� D ]}| �|� | �| j� q |d rP| �|d � d S )N���)�NLCRE�splitr r )r �lines�liner r r � _write_lines� s
zGenerator._write_linesc C s� | j }z2d | _| �� | _ }| �|� W || _ | j}| `n|| _ | j}| `0 |r�t|�}|�d�d u rt|d |d< n|�d|d � |�d|d � t|dd �}|d u r�| �|� n|| � | j � |�
� � d S )N�content-transfer-encodingr �Content-Transfer-Encoding� content-type� �_write_headers) r �
_munge_cter, � _dispatchr �getZreplace_header�getattrr7 r �getvalue)r r% ZoldfpZsfpZ munge_cte�methr r r r$ � s, � zGenerator._writec C st |� � }|�� }t�||f��dd�}t| d| d �}|d u rh|�dd�}t| d| d �}|d u rh| j}||� d S )N�-r Z_handle_)�get_content_maintype�get_content_subtype�
UNDERSCORE�join�replacer; �
_writeBody)r r% �main�subZspecificr= Zgenericr r r r9 � s zGenerator._dispatchc C s6 |� � D ] \}}| �| j�||�� q| �| j� d S r )� raw_itemsr r Zfoldr �r r% �h�vr r r r7 � s zGenerator._write_headersc C s� |� � }|d u rd S t|t�s.tdt|� ��t|j�r~|�d�}|d ur~t|�}|d= |� ||� |� � }|d |d f| _
| j r�t �
d|�}| �|� d S )Nzstring payload expected: %s�charsetr3 r5 �>From )� get_payload�
isinstance�str� TypeError�typer �_payloadZ get_paramr Z set_payloadr8 r �fcrerF r2 )r r% �payloadrK r r r � _handle_text� s$
� zGenerator._handle_textc C s� g }|� � }|d u rg }n(t|t�r2| �|� d S t|t�sB|g}|D ]6}| �� }| �|�}|j|d| jd� |� |�
� � qF|� � }|s�| j �
|�}| �|�}|�|� |jd ur�| jr�t�d|j�} n|j} | �| � | �| j� | �d| | j � |�r| j�|�d�� |D ],}
| �| jd | | j � | j�|
� �q| �| jd | d | j � |jd u�r�| j�r�t�d|j�} n|j} | �| � d S )NF�r& r rL z--r )rM rN rO r �listr, r r' r �appendr< Z get_boundaryr rB �_make_boundaryZ set_boundary�preambler rS rF r2 r �pop�epilogue) r r% ZmsgtextsZsubparts�partr �g�boundaryZalltextrZ Z body_partr\ r r r �_handle_multipart sJ
zGenerator._handle_multipartc C s6 | j }|jdd�| _ z| �|� W || _ n|| _ 0 d S )Nr r )r r r` )r r% �pr r r �_handle_multipart_signed< s
z"Generator._handle_multipart_signedc C s� g }|� � D ]t}| �� }| �|�}|j|d| jd� |�� }|�| j�}|rv|d | jkrv|� | j�
|d d� �� q |� |� q | j � | j�
|�� d S )NFrV r- )
rM r, r r' r r< r/ r r! rX rB r r )r r% Zblocksr] r r^ �textr0 r r r � _handle_message_delivery_statusG s
z)Generator._handle_message_delivery_statusc C s^ | � � }| �|�}|j}t|t�rD|j|�d�d| jd� |�� }n
| � |�}| j
� |� d S )Nr FrV ) r, r rR rN rW r' rM r r< r r r )r r% r r^ rT r r r �_handle_message\ s
zGenerator._handle_messagec C sv t �tj�}dt| d }|d u r(|S |}d}| �dt�|� d tj�}|� |�sXqr|d t
|� }|d7 }q0|S )Nz===============z==r z^--z(--)?$�.r6 ) �randomZ randrange�sys�maxsize�_fmt� _compile_re�re�escape� MULTILINE�searchrO )�clsrc �tokenr_ �bZcounterZcrer r r rY u s
zGenerator._make_boundaryc C s t �||�S r )rl �compile�rp r �flagsr r r rk � s zGenerator._compile_re)NN)FN)N)�__name__�
__module__� __qualname__r r r' r r, r r2 r$ r9 r7 rU rD r` rb rd re � classmethodrY rk r r r r r s,
�
1'
: r c sL e Zd Zdd� Zdd� Zdd� Zdd� Z� fd d
�ZeZe d d � �Z
� Z S )
r c C s | j �|�dd�� d S )N�ascii�surrogateescape)r r �encoder r r r r � s zBytesGenerator.writec C s t � S r )r r+ r r r r, � s zBytesGenerator._new_bufferc C s
|� d�S �Nrz )r| r r r r r � s zBytesGenerator._encodec C s8 |� � D ] \}}| j�| j�||�� q| �| j� d S r )rG r r r Z fold_binaryr rH r r r r7 � s z BytesGenerator._write_headersc s\ |j d u rd S t|j �rH| jjdksH| jr:t�d|j �|_ | �|j � ntt | ��
|� d S )NZ7bitrL ) rR r r Zcte_typer rS rF r2 �superr rU )r r% �r) r r rU � s
zBytesGenerator._handle_textc C s t �|�d�|�S r} )rl rs r| rt r r r rk � s zBytesGenerator._compile_re) rv rw rx r r, r r7 rU rD ry rk �
__classcell__r r r r r � s
r zD[Non-text (%(type)s) part of message omitted, filename %(filename)s]c @ s$ e Zd Zddd�dd�Zdd� ZdS )r Nr c C s. t j| ||||d� |d u r$t| _n|| _d S r( )r r �_FMTrj )r r r
r Zfmtr r r r r � s �zDecodedGenerator.__init__c C s� |� � D ]v}|�� }|dkr2t|jdd�| d� q|dkr<qt| j|�� |�� |�� |�d�|�dd�|�d d
�d � | d� qd S ) Nrc F)�decode)�fileZ multipartz
[no filename]zContent-Descriptionz[no description]r4 z
[no encoding])rQ �maintypeZsubtype�filenameZ description�encoding) �walkr? �printrM rj Zget_content_typer@ � get_filenamer: )r r% r] r� r r r r9 � s( ��� �zDecodedGenerator._dispatch)NNN)rv rw rx r r9 r r r r r � s � r r6 z%%0%dd) �__all__rl rh r" rg �copyr �ior r Z email.utilsr rA �NLrs r. rn rS r r r� r �len�reprri Z_widthrj rY r r r r �<module> s(
t3; |