File: //lib64/python3.9/email/__pycache__/generator.cpython-39.pyc
a
�DOgPQ � @ s� d Z g d�ZddlZddlZddlZddlZddlmZ ddlm Z m
Z
ddl m Z ddl
mZ dZd Ze�d
�Ze�d ej�Ze�d �ZG d
d� d�ZG dd� de�ZdZG dd� de�Zeeej d ��Z de Z ej Z dS )z:Classes to generate plain text from a message object tree.)� 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 ZdZd'dd�dd�Zdd� 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 z�Generates output from a Message object tree.
This basic generator writes the message to the given file object as plain
text.
N��policyc C s6 |du r|du rdn|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
)�self�outfpr r r
� r �'/usr/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 du r|j n| j }|dur*|j|d�}| jdurB|j| jd�}|j| _| �| j�| _d| _| �| j�| _| j }|j }zX|| _ ||_ |r�|� � }|s�dt
� t
�
� � }| � || j � | �
|� W || _ ||_ n|| _ ||_ 0 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�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 )z1Clone this generator with the exact same options.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�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 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 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& �payloadrN 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 rO z--r )rP rQ rR r �listr, r r( r �appendr< Z get_boundaryr! rB �_make_boundaryZ set_boundary�preambler rV rF r2 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 rc )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 )NFrY r- )
rP r, r r( r r< r/ r! r" r[ rB r r )r r& Zblocksr` r ra �textr0 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 FrY ) r, r rU rQ rZ r( rP r r< r r r )r r& r ra rW 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(--)?$�.r6 ) �random� randrange�sys�maxsize�_fmt� _compile_re�re�escape� MULTILINErJ rR )�clsrf �tokenrb �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__�__doc__r r r( r r, r r2 r% r9 r7 rX rD rc re rg rh � classmethodr\ ro r r r r r s. �
1': r 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 �|�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 )rG r r r
Z fold_binaryr )r r& rL rM 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 )NZ7bitrO ) rU r r
Zcte_typer rV rF r2 �superr rX )r r& �r) r r rX � 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, r r7 rX rD 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 ZdZddd�dd�Zdd� ZdS ) r z�Generates 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.
Nr c C s. t j| ||||d� |du r$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 �_FMTrn )r r r r �fmtr
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 ) Nrf F)�decode)�fileZ multipartz
[no filename]zContent-Descriptionz[no description]r4 z
[no encoding])rT �maintypeZsubtype�filename� description�encoding) �walkr? �printrP rn Zget_content_typer@ � get_filenamer: )r r& r` r� r r r r9 � s( ��� �zDecodedGenerator._dispatch)NNN)ry rz r{ r| r r9 r r r r r � s
� r r6 z%%0%dd) r| �__all__rp rl r# rj �copyr �ior r Z email.utilsr Z email.errorsr rA �NLrv r. rr rV rI r r r� r �len�reprrm �_widthrn r\ r r r r �<module> s.
}3;