3
LjS @ s d ddgZ ddlZddlZddlZddlZddlmZ ddlmZm Z ddl
m Z ddl m
Z
d Zd
Zejd Zejd ejZejd
ZejdZG dd d ZG dd deZdZG dd deZeeej d Z de Z ej Z dS ) GeneratorDecodedGeneratorBytesGenerator N)deepcopy)StringIOBytesIO)_has_surrogates)HeaderWriteError_
z
\r\n|\r|\nz^From z \r\n[^ \t]|\r[^ \n\t]|\n[^ \t]s \r\n[^ \t]|\r[^ \n\t]|\n[^ \t]c @ s e Zd Zd&ddddZdd 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 )selfoutfpr
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 G 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 }|sdt
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_unixfromtimeZctimer _write)r msgunixfromr r Zold_gen_policyZold_msg_policyZufromr r r flattenK s,
zGenerator.flattenc C s | j || jd | jdS )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) )NLCREsplitr r )r linesliner 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_headergetattrr/ r getvalue)r r" ZoldfpZsfpZ munge_ctemethr r r r! s" zGenerator._writec C st |j }|j }tj||fjdd}t| d| d }|d krh|jdd}t| d| d }|d krh| j}|| d S )N-r
Z_handle_)get_content_maintypeget_content_subtype
UNDERSCOREjoinreplacer2
_writeBody)r r" mainsubZspecificr4 Zgenericr r r r1 s zGenerator._dispatchc C s x|j D ]\}}| jj||}| jjr| jj}|j| jjsRtd|d||}|j|rr|d t| }tj |rtd|| j
| q
W | j
| j d S )Nz folded header does not end with z: z folded header contains newline: ) raw_itemsr Zfoldverify_generated_headersr endswithr lenNEWLINE_WITHOUT_FWSPsearchr r )r r" hvfoldedr folded_no_linesepr r r r/ s
zGenerator._write_headersc C s |j }|d krd S t|ts.tdt| t|jr~|jd}|d k r~t|}|d= |j || |j }|d |d f| _
| j rt j
d|}| j| d S )Nzstring payload expected: %scharsetzcontent-transfer-encodingz content-typez>From ) get_payload
isinstancestr TypeErrortyper _payloadZ get_paramr Z set_payloadr0 r fcrer= r. )r r" payloadrH r r r _handle_text s"
zGenerator._handle_textc C s g }|j }|d krg }n(t|tr2| j| d S t|tsB|g}x>|D ]6}| j }| j|}|j|d| jd |j |j
qHW |j }|s| j j
|}| j|}|j| |jd k r| jrtjd|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| jrtjd|j} n|j} | j| d S )NF)r# r z>From z--r )rI rJ rK r listr' r r$ r appendr3 Z get_boundaryr r9 _make_boundaryZ set_boundarypreambler rO r= r. r popepilogue) r r" ZmsgtextsZsubpartspartr gboundaryZalltextrU Z body_partrW r r r _handle_multipart
sJ
zGenerator._handle_multipartc C s0 | j }|jdd| _ z| j| W d || _ X d S )Nr )r )r r r[ )r r" pr r r _handle_multipart_signedG 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) )
rI r' r r$ r r3 r+ r r rS r9 r r )r r" ZblocksrX r rY textr, r r r _handle_message_delivery_statusR s
z)Generator._handle_message_delivery_statusc C s^ | j }| j|}|j}t|trD|j|jdd| jd |j }n
| j |}| j
j | d S )Nr F)r# r ) r' r rN rJ rR r$ rI r r3 r r r )r r" r rY rP r r r _handle_messageg 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 randrangesysmaxsize_fmt _compile_rereescape MULTILINErC rK )clsr^ tokenrZ bZcounterZcrer r r rT s
zGenerator._make_boundaryc C s t j||S )N)ri compile)rl r flagsr r r rh s zGenerator._compile_re)NN)FN)N)__name__
__module__ __qualname__r r r$ r r' r r. r! r1 r/ rQ r; r[ r] r_ r` classmethodrT rh 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 )Nasciisurrogateescape)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 dS )Nru )rw )r r r r r r s zBytesGenerator._encodec C s x|j D ]\}}| jj||}| jjr| jjj }|j|sRtd|d||}|j|rr|d t| }t j
|rtd|| j j | q
W | j | j
d S )Nz folded header does not end with z: z folded header contains newline: )r> r Z fold_binaryr? r rw r@ r rA NEWLINE_WITHOUT_FWSP_BYTESrC r r r )r r" rD rE rF r rG r r r r/ s
z BytesGenerator._write_headersc s^ |j d krd S t|j rJ| jjdk rJ| jrFrom ) rN r r Zcte_typer rO r= r. superr rQ )r r" )r% r r rQ s
zBytesGenerator._handle_textc C s t j|jd|S )Nru )ri ro rw )rl r rp r r r rh s zBytesGenerator._compile_re) rq rr rs r r' r r/ rQ r; rt rh
__classcell__r r )r% r r s
zD[Non-text (%(type)s) part of message omitted, filename %(filename)s]c @ s$ e Zd ZdddddZdd ZdS )r N)r c C s. t j| ||||d |d kr$t| _n|| _d S )N)r )r r _FMTrg )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])rM maintypesubtypefilenameZ descriptionencoding) walkr6 printrI rg Zget_content_typer7 get_filenameget)r r" rX r~ r r r r1 s
zDecodedGenerator._dispatch)NNN)rq rr rs r r1 r r r r r s r( z%%0%dd) __all__ri re r rd copyr ior r Z email.utilsr Z email.errorsr r8 NLro r* rk rO rB rx r r r{ r rA reprrf Z_widthrg rT r r r r s.
|?;