3
�Qg�P � @ s� d ddgZ ddlZddlZddlZddlZddlmZ ddlmZm Z ddl
m Z ddl m
Z
d Zd
Zejd �Zejd ej�Zejd
�ZG dd � 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 kr|d krdn|j }|| _|| _|| _|| _d S )NT)� mangle_from_�_fp�
_mangle_from_� maxheaderlenr )�self�outfpr
r r � r �'/usr/lib64/python3.6/email/generator.py�__init__& s zGenerator.__init__c C s | j j|� d S )N)r �write)r �sr r r r F s zGenerator.writeFc C s� | j d kr|j n| j }|d k r*|j|d�}| jd k rB|j| jd�}|j| _| j| j�| _d| _| j| j�| _| j }|j }zL|| _ ||_ |r�|j � }|s�dt
j t
j
� � }| j || j � | j
|� W d || _ ||_ X d S )N)�linesep)�max_line_length� z From nobody )r �cloner r �_NL�_encode� _encoded_NLZ_EMPTY�_encoded_EMPTYZ get_unixfrom�timeZctimer �_write)r �msg�unixfromr r Zold_gen_policyZold_msg_policyZufromr r r �flattenJ s,
zGenerator.flattenc C s | j || jd | jd�S )N)r )� __class__r r )r �fpr r r r { s zGenerator.clonec C s t � S )N)r )r r r r � _new_buffer� s zGenerator._new_bufferc C s |S )Nr )r r r r r r � s zGenerator._encodec C sX |sd S t j|�}x*|d d� D ]}| j|� | j| j� q W |d rT| j|d � d S )N� ���r) r) )�NLCRE�splitr r )r �lines�liner r r � _write_lines� s
zGenerator._write_linesc C s� | j }z"d | _| j� | _ }| j|� W d || _ | j}| `X |rht|�}|jd|d � |jd|d � t|dd �}|d kr�| j|� n|| � | j j|j � � d S )Nzcontent-transfer-encodingr z content-typer( �_write_headers)
r �
_munge_cter' � _dispatchr Zreplace_header�getattrr/ r �getvalue)r r" ZoldfpZsfpZ munge_cte�methr r r r! � s" zGenerator._writec C st |j � }|j� }tj||f�jdd�}t| d| d �}|d krh|jdd�}t| d| d �}|d krh| j}||� d S )N�-r
Z_handle_)�get_content_maintype�get_content_subtype�
UNDERSCORE�join�replacer2 �
_writeBody)r r" �main�subZspecificr4 Zgenericr r r r1 � s zGenerator._dispatchc C s� x�|j � D ]�\}}| jj||�}| jjr�| jj}|j| jj�sRtd|�d|����|}|j|�rr|d t|� � }tj |�r�td|����| j
|� q
W | j
| j � d S )Nz folded header does not end with z: z folded header contains newline: ) � raw_itemsr ZfoldZverify_generated_headersr �endswithr �len�NEWLINE_WITHOUT_FWSP�searchr r )r r" �h�vZfoldedr Zfolded_no_linesepr r r r/ � s
zGenerator._write_headersc C s� |j � }|d krd S t|t�s.tdt|� ��t|j�r~|jd�}|d k r~t|�}|d= |j ||� |j � }|d |d f| _
| j r�t j
d|�}| j|� d S )Nzstring payload expected: %s�charsetzcontent-transfer-encodingz content-typez>From )� get_payload�
isinstance�str� TypeError�typer �_payloadZ get_paramr Z set_payloadr0 r �fcrer= r. )r r" �payloadrE r r r � _handle_text� s"
zGenerator._handle_textc C s� g }|j � }|d krg }n(t|t�r2| j|� d S t|t�sB|g}x>|D ]6}| j� }| j|�}|j|d| jd� |j |j
� � qHW |j � }|s�| j j
|�}| j|�}|j|� |jd k r�| jr�tjd|j�} n|j} | j| � | j| j� | jd| | j � |�r| jj|jd�� x4|D ],}
| j| jd | | j � | jj|
� �q W | j| jd | d | j � |jd k �r�| j�r�tjd|j�} n|j} | j| � d S )NF)r# r z>From z--r )rF rG rH r �listr' r r$ r �appendr3 Z get_boundaryr r9 �_make_boundaryZ set_boundary�preambler rL r= r. r �pop�epilogue) r r" ZmsgtextsZsubparts�partr �g�boundaryZalltextrR Z body_partrT r r r �_handle_multipart sJ
zGenerator._handle_multipartc C s0 | j }|jdd�| _ z| j|� W d || _ X d S )Nr )r )r r rX )r r" �pr r r �_handle_multipart_signedF s
z"Generator._handle_multipart_signedc C s� g }x�|j � D ]t}| j� }| j|�}|j|d| jd� |j� }|j| j�}|rx|d | jkrx|j | jj
|d d� �� q|j |� qW | j j | jj
|�� d S )NF)r# r r( r) r) )
rF r' r r$ r r3 r+ r r rP r9 r r )r r" ZblocksrU r rV �textr, r r r � _handle_message_delivery_statusQ s
z)Generator._handle_message_delivery_statusc C s^ | j � }| j|�}|j}t|t�rD|j|jd�d| jd� |j� }n
| j |�}| j
j |� d S )Nr F)r# r ) r' r rK rG rO r$ rF r r3 r r r )r r" r rV rM r r r �_handle_messagef s
zGenerator._handle_messagec C sz t jtj�}d t| d }|d kr(|S |}d}xD| jdtj|� d tj�}|j |�sZP |d t
|� }|d7 }q2W |S )
N�=� z==r z^--z(--)?$�.r( z===============) �randomZ randrange�sys�maxsize�_fmt� _compile_re�re�escape� MULTILINErB rH )�clsr[ �tokenrW �bZcounterZcrer r r rQ s
zGenerator._make_boundaryc C s t j||�S )N)rf �compile)ri r �flagsr r r re � s zGenerator._compile_re)NN)FN)N)�__name__�
__module__� __qualname__r r r$ r r' r r. r! r1 r/ rN r; rX rZ r\ r] � classmethodrQ re r r r r r s(
1#: c sL e Zd Zdd� Zdd� Zdd� Zdd� Z� fd d
�ZeZe d d � �Z
� Z S )
r c C s | j j|jdd�� d S )N�ascii�surrogateescape)r r �encode)r r r r r r � s zBytesGenerator.writec C s t � S )N)r )r r r r r' � s zBytesGenerator._new_bufferc C s
|j d�S )Nrr )rt )r r r r r r � s zBytesGenerator._encodec C s< x*|j � D ] \}}| jj| jj||�� q
W | j| j� d S )N)r> r r r Z fold_binaryr )r r" rC rD r r r r/ � s z BytesGenerator._write_headersc s^ |j d krd S t|j �rJ| jjdk rJ| jr<tjd|j �|_ | j|j � ntt | �j
|� d S )NZ7bitz>From ) rK r r Zcte_typer rL r= r. �superr rN )r r" )r% r r rN � s
zBytesGenerator._handle_textc C s t j|jd�|�S )Nrr )rf rl rt )ri r rm r r r re � s zBytesGenerator._compile_re) rn ro rp r r' r r/ rN r; rq re �
__classcell__r r )r% r r � s
zD[Non-text (%(type)s) part of message omitted, filename %(filename)s]c @ s$ e Zd Zddd�dd�Zdd� ZdS )r N)r c C s. t j| ||||d� |d kr$t| _n|| _d S )N)r )r r �_FMTrd )r r r
r Zfmtr r r r r � s
zDecodedGenerator.__init__c C s� x�|j � D ]v}|j� }|dkr4t|jdd�| d� q
|dkr>q
t| j|j� |j� |j� |jd�|jdd�|jd d
�d � | d� q
W d S ) Nr[ F)�decode)�fileZ multipartz
[no filename]zContent-Descriptionz[no description]zContent-Transfer-Encodingz
[no encoding])rJ �maintype�subtype�filenameZ description�encoding) �walkr6 �printrF rd Zget_content_typer7 � get_filename�get)r r" rU rz r r r r1 � s
zDecodedGenerator._dispatch)NNN)rn ro rp r r1 r r r r r � s r( z%%0%dd) �__all__rf rb r ra �copyr �ior r Z email.utilsr Z email.errorsr r8 �NLrl r* rh rL rA r r rw r r@ �reprrc Z_widthrd rQ r r r r �<module> s,
|3;