3
fN @ s d Z dddgZddlZddlZddlZddlZddlmZ ddlm Z m
Z
ddl m Z d Z
d
Zejd Zejd ejZG d
d dZG dd deZdZG dd deZeeejd Zde Zej Z dS )z:Classes to generate plain text from a message object tree. GeneratorDecodedGeneratorBytesGenerator N)deepcopy)StringIOBytesIO)_has_surrogates_
z
\r\n|\r|\nz^From c @ s e Zd ZdZd'ddddZdd Zd(d d
Zd d Zd
d Zdd Z dd Z
dd Z dd Z dd Z
dd ZeZdd Zd d Zd d Zd!d" Zed)d#d$Zed%d& ZdS )*r zGenerates output from a Message object tree.
This basic generator writes the message to the given file object as plain
text.
N)policyc C s6 |dkr|dkrdn|j }|| _|| _|| _|| _dS )a Create the generator for message flattening.
outfp is the output file-like object for writing the message to. It
must have a write() method.
Optional mangle_from_ is a flag that, when True (the default if policy
is not set), escapes From_ lines in the body of the message by putting
a `>' in front of them.
Optional maxheaderlen specifies the longest length for a non-continued
header. When a header line is longer (in characters, with tabs
expanded to 8 spaces) than maxheaderlen, the header will split as
defined in the Header class. Set maxheaderlen to zero to disable
header wrapping. The default is 78, as recommended (but not required)
by RFC 2822.
The policy keyword specifies a policy object that controls a number of
aspects of the generator's operation. If no policy is specified,
the policy associated with the Message object passed to the
flatten method is used.
NT) mangle_from__fp
_mangle_from_ maxheaderlenr )selfoutfpr r r r 4/opt/alt/python36/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 D s zGenerator.writeFc C s | j dkr|j n| j }|dk r*|j|d}| jdk rB|j| jd}|j| _| j| j| _d| _| j| j| _| j }|j }zL|| _ ||_ |r|j }|sdt
j t
j
}| j || j | j
| W d|| _ ||_ X dS )a Print the message object tree rooted at msg to the output file
specified when the Generator instance was created.
unixfrom is a flag that forces the printing of a Unix From_ delimiter
before the first object in the message tree. If the original message
has no From_ delimiter, a `standard' one is crafted. By default, this
is False to inhibit the printing of any From_ delimiter.
Note that for subobjects, no From_ line is printed.
linesep specifies the characters used to indicate a new line in
the output. The default value is determined by the policy specified
when the Generator instance was created or, if none was specified,
from the policy associated with the msg.
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 flattenH s,
zGenerator.flattenc C s | j || jd| jdS )z1Clone this generator with the exact same options.N)r ) __class__r r )r fpr r r r y 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
UNDERSCOREjoinreplacer1
_writeBody)r r! mainsubZspecificr3 Zgenericr r r r0 s zGenerator._dispatchc C s: x(|j D ] \}}| j| jj|| q
W | j| j d S )N) raw_itemsr r Zfoldr )r r! hvr 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_payloadr/ r fcrer< r- )r r! payloadr@ 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 )rA rB rC r listr& r r# r appendr2 Z get_boundaryr r8 _make_boundaryZ set_boundarypreambler rG r< r- r
popepilogue) r r! ZmsgtextsZsubpartspartr gboundaryZalltextrM Z body_partrO r r r _handle_multipart sJ
zGenerator._handle_multipartc C s0 | j }|jdd| _ z| j| W d || _ X d S )Nr )r )r r rS )r r! pr r r _handle_multipart_signed8 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( )
rA r& r r# r r2 r* r r rK r8 r
r )r r! ZblocksrP r rQ textr+ r r r _handle_message_delivery_statusC 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 rF rB rJ r# rA r r2 r r
r )r r! r rQ rH r r r _handle_messageX 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 MULTILINEsearchrC )clsrV tokenrR bZcounterZcrer r r rL q s
zGenerator._make_boundaryc C s t j||S )N)ra compile)re r flagsr r r r` s zGenerator._compile_re)NN)FN)N)__name__
__module__ __qualname____doc__r r r# r r& r r- r r0 r. rI r: rS rU rW rX classmethodrL r` r r r r r s*
1#
: c sP e Zd ZdZdd Zdd Zdd Zdd Z fd
d ZeZ e
d d
Z Z S )r a Generates a bytes version of a Message object tree.
Functionally identical to the base Generator except that the output is
bytes and not string. When surrogates were used in the input to encode
bytes, these are decoded back to bytes for output. If the policy has
cte_type set to 7bit, then the message is transformed such that the
non-ASCII bytes are properly content transfer encoded, using the charset
unknown-8bit.
The outfp object must accept bytes in its write method.
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 )Nro )rq )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! r> r? r r r r. s z BytesGenerator._write_headersc s^ |j d krd S t|j rJ| jjdk rJ| jrFrom ) rF r r Zcte_typer rG r< r- superr rI )r r! )r$ r r rI s
zBytesGenerator._handle_textc C s t j|jd|S )Nro )ra rh rq )re r ri r r r r` s zBytesGenerator._compile_re)
rj rk rl rm r r& r r. rI r: rn r`
__classcell__r r )r$ r r s
zD[Non-text (%(type)s) part of message omitted, filename %(filename)s]c @ s( e Zd ZdZdddddZdd ZdS ) r zGenerates a text representation of a message.
Like the Generator base class, except that non-text parts are substituted
with a format string representing the part.
N)r c C s. t j| ||||d |dkr$t| _n|| _dS )a Like Generator.__init__() except that an additional optional
argument is allowed.
Walks through all subparts of a message. If the subpart is of main
type `text', then it prints the decoded payload of the subpart.
Otherwise, fmt is a format string that is used instead of the message
payload. fmt is expanded with the following keywords (in
%(keyword)s format):
type : Full MIME type of the non-text part
maintype : Main MIME type of the non-text part
subtype : Sub-MIME type of the non-text part
filename : Filename of the non-text part
description: Description associated with the non-text part
encoding : Content transfer encoding of the non-text part
The default value for fmt is None, meaning
[Non-text (%(type)s) part of message omitted, filename %(filename)s]
)r N)r r _FMTr_ )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 ) NrV F)decode)fileZ multipartz
[no filename]zContent-Descriptionz[no description]zContent-Transfer-Encodingz
[no encoding])rE maintypesubtypefilenameZ descriptionencoding) walkr5 printrA r_ Zget_content_typer6 get_filenameget)r r! rP rw r r r r0 s
zDecodedGenerator._dispatch)NNN)rj rk rl rm r r0 r r r r r s r' z%%0%dd) rm __all__ra r] r r\ copyr ior r Z email.utilsr r7 NLrh r) rc rG r r rt r lenreprr^ Z_widthr_ rL r r r r s*
p3;