a
XC?hPQ � @ s� g d �Z ddlZddlZddlZddlZddlmZ ddlmZm Z ddl
m Z ddl m
Z
dZdZe�d �Ze�d
ej�Ze�d �ZG d d
� d
�ZG dd� de�ZdZG dd� de�Zeeejd ��Z de Z ej Z dS ))� Generator�DecodedGenerator�BytesGenerator� N)�deepcopy)�StringIO�BytesIO)�_has_surrogates)�HeaderWriteError�_�
z
\r\n|\r|\nz^From z \r\n[^ \t]|\r[^ \n\t]|\n[^ \t]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 F 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 �flattenJ s0
�zGenerator.flattenc C s | j || jd | jd�S �Nr )� __class__r r
)r �fpr r r r { 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�getattrr8 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 r: � s zGenerator._dispatchc C s� |� � D ]p\}}| j�||�}| jjrn| jj}|�| jj�sPtd|�d|����t�|� |��rntd|����| �
|� q| �
| j � d S )Nz folded header does not end with z: z folded header contains newline: ) � raw_itemsr
ZfoldZverify_generated_headersr �endswithr �NEWLINE_WITHOUT_FWSP�search� removesuffixr r )r r& �h�vZfoldedr r r r r8 � 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�charsetr4 r6 �>From )� get_payload�
isinstance�str� TypeError�typer �_payloadZ get_paramr Z set_payloadr9 r �fcrerG r3 )r r& �payloadrO 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 rP z--r )rQ rR rS r �listr- r r( r �appendr= Z get_boundaryr! rC �_make_boundaryZ set_boundary�preambler rW rG r3 r �pop�epilogue) r r& ZmsgtextsZsubparts�partr �g�boundaryZalltextr^ 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 rd )r r& �pr r r �_handle_multipart_signedG 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 )NFrZ r. )
rQ r- r r( r r= r0 r! r" r\ rC r r )r r& Zblocksra r rb �textr1 r r r � _handle_message_delivery_statusR s
z)Generator._handle_message_delivery_statusc C s^ | � � }| �|�}|j}t|t�rD|j|�d�d| jd� |�� }n
| � |�}| j
� |� d S )Nr FrZ ) r- r rV rR r[ r( rQ r r= r r r )r r& r rb rX r r r �_handle_messageg 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(--)?$�.r7 ) �randomZ randrange�sys�maxsize�_fmt� _compile_re�re�escape� MULTILINErK rS )�clsrg �tokenrc �bZcounterZcrer r r r] � s
zGenerator._make_boundaryc C s t �||�S r )rp �compile�rs r �flagsr r r ro � s zGenerator._compile_re)NN)FN)N)�__name__�
__module__� __qualname__r r r( r r- r r3 r% r: r8 rY rE rd rf rh ri � classmethodr] ro 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 �Nr} )r r r r r r � s zBytesGenerator._encodec C s8 |� � D ] \}}| j�| j�||�� q| �| j� d S r )rH r r r
Z fold_binaryr )r r& rM rN r r r r8 � s z BytesGenerator._write_headersc s\ |j d u rd S t|j �rH| jjdksH| jr:t�d|j �|_ | �|j � ntt | ��
|� d S )NZ7bitrP ) rV r r
Zcte_typer rW rG r3 �superr rY )r r& �r* r r rY � s
zBytesGenerator._handle_textc C s t �|�d�|�S r� )rp rv r rw r r r ro � s zBytesGenerator._compile_re) ry rz r{ r r- r r8 rY rE r| ro �
__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 �_FMTrn )r r r r Zfmtr
r r r r � s �zDecodedGenerator.__init__c C s� |� � D ]v}|�� }|dkr2t|jdd�| d� q|dkr s,
}3;