B
�/g^�) c @ s: d dl Z d dlZd dlZd dlZd dlmZ G dd� d�Ze� Zd%dd�Ze� de� d d
� Z
x d � � D ]Z e� e e
� qhW d d
� Z
x d� � D ]Ze� de e
� q�W dd� Ze� de� dd� Zdd� Zdd� Zdd� Zd&d d �Ze�ee� d'd d!�Ze�ejje� d(d#d$�Zx ee e fD ]Z e�e e� �q W dS ))� N)�
quoprimimec @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�ContentManagerc C s i | _ i | _d S �N)� get_handlers� set_handlers)�self� r �,/usr/lib64/python3.7/email/contentmanager.py�__init__ s zContentManager.__init__c C � || j |< d S r )r )r �key�handlerr r r �add_get_handler
� z ContentManager.add_get_handlerc O s| |� � }|| jkr(| j| |f|�|�S |�� }|| jkrP| j| |f|�|�S d| jkrp| jd |f|�|�S t|��d S )N� )Zget_content_typer �get_content_maintype�KeyError)r �msg�args�kwZ content_type�maintyper r r � get_content s
zContentManager.get_contentc C r
r )r )r Ztypekeyr r r r �add_set_handler r z ContentManager.add_set_handlerc O s> |� � dkrtd��| �||�}|�� |||f|�|� d S )NZ multipartz"set_content not valid on multipart)r Z TypeError�_find_set_handlerZ
clear_content)r r �objr r r r r r � set_content s
zContentManager.set_contentc C s� d }x�t |�jD ]�}|| jkr(| j| S |j}t|dd�}|rLd�||f�n|}|d kr\|}|| jkrp| j| S || jkr�| j| S |j}|| jkr| j| S qW d | jkr�| jd S t|��d S )N�
__module__r Z.)ZtypeZ__mro__r � __qualname__Zgetattr�join�__name__r ) r r r Zfull_path_for_error�typZqnameZmodnameZ full_path�namer r r r ' s&
z ContentManager._find_set_handlerN) r r r r r
r r r r r r r r r s r �replacec C s& | j dd�}| �dd�}|j||d�S )NT��decode�charsetZASCII)�errors)� get_payloadZ get_paramr# )r r% Zcontentr$ r r r �get_text_content@ s r' �textc C s | j dd�S )NTr" �r& �r r r r �get_non_text_contentG r r+ z audio image video applicationc C s
| � d�S �Nr r) r* r r r �get_message_contentM r r- zrfc822 external-bodyzmessage/c C s t | �d��S r, )�bytesr& r* r r r �%get_and_fixup_unknown_message_contentS s r/ �messagec
s� d� ||f�| d< |r�t|d d�s<| j� � fdd�|D �}y,x&|D ] }|jrX|jd �|| |j< qDW W n@ tjjk
r� } z td� |j
| jd���|�W d d }~X Y nX d S ) Nz/z Content-Typer r c s g | ]}� j � �|g�� �qS r )Zheader_factoryZheader_source_parse)�.0�header�Zmpr r z
<listcomp>d s z _prepare_set.<locals>.<listcomp>zInvalid header: {})�policy) r Zhasattrr4 Zdefectsr �emailr% Z HeaderDefect�
ValueError�formatZfold)r r �subtype�headersr2 Zexcr r3 r � _prepare_set_ s
r: c C s| |d kr|d k rd}|d k r$|| d< |d k r>| j d|ddd� |d k rN|| d< |d k rxx |�� D ]\}}| � ||� q`W d S )NZ
attachmentzContent-Disposition�filenameT)r2 r! z
Content-ID)� set_paramZitems)r � dispositionr; �cid�paramsr Zvaluer r r �
_finalize_setp s r@ c C sZ g }|d d }x>t dt| �|�D ]*}| ||| � }|�t�|��d�� q"W d�|�S )Ni i r �asciir )Zrange�lenZappend�binascii�
b2a_base64r# r )�data�max_line_lengthZ
encoded_linesZunencoded_bytes_per_lineZiZthisliner r r �_encode_base64� s rG c s| | � |��� }|j� d�� � fdd�}dd� }|d kr�yd||��d�fS tk
r\ Y nX |jdkr�tdd � |D ��|jkr�d||��dd
�fS ||d d � �}t� |�d �|j�}t
� |�} t |�t | �kr�d
}nd}t |�d kr�||fS |dk�r
||��d�}
nj|dk�r&||��dd
�}
nN|dk�rJt� ||��d �|j�}
n*|d
k�rft
||�|j�}
ntd�|���||
fS )NrA c s � � | �� S r �r ��lines��linesepr r �
embedded_body� � z#_encode_text.<locals>.embedded_bodyc S s d� | �d S )Ns
rH rI r r r � normal_body� rN z!_encode_text.<locals>.normal_body�7bit�8bitc s s | ]}t |�V qd S r )rB )r1 Zxr r r z <genexpr>� s z _encode_text.<locals>.<genexpr>�surrogateescapei
zlatin-1�base64�quoted-printablez$Unknown content transfer encoding {})�encodeZ
splitlinesrL r# ZUnicodeDecodeErrorZcte_typeZmaxrF r Z body_encoderC rD rB rG r6 r7 ) �stringr$ �cter4 rJ rM rO ZsniffZsniff_qpZ sniff_base64rE r rK r � _encode_text� s@
rX �plain�utf-8c
C sd t | d|| � t|||| j�\}}
| �|
� | jdtjj�||�dd� || d< t | ||||� d S )Nr( r$ T�r! �Content-Transfer-Encoding)
r: rX r4 � set_payloadr< r5 r$ ZALIASESZgetr@ ) r rV r8 r$ rW r= r; r> r? r9 Zpayloadr r r �set_text_content� s
r^ �rfc822c C s� |dkrt d��|dkr@|dkr.t d�|���|d kr:dn|}n0|dkrd|dkr^t d �|���d
}n |d krpd
}t| d ||� | �|g� || d < t| ||||� d S )
NZpartialz4message/partial is not supported for Message objectsr_ )NrP rQ �binaryz*message/rfc822 parts do not support cte={}rQ z
external-body)NrP z1message/external-body parts do not support cte={}rP r0 r\ )r6 r7 r: r] r@ ) r r0 r8 rW r= r; r> r? r9 r r r �set_message_content� s$ ra rS c
C s� t | ||| � |dkr(t|| jjd�}nN|dkrNtj|dddd�}|�d�}n(|dkrb|�d� n|d krv|�dd
�}| �|� || d < t | ||||� d S ) NrS )rF rT FT)Zistextr2 Z quotetabsrA rP )rQ r` rR r\ )
r: rG r4 rF rC Zb2a_qpr# rU r] r@ )
r rE r r8 rW r= r; r> r? r9 r r r �set_bytes_content� s
rb r[ )rY rZ NNNNNN)r_ NNNNNN)rS NNNNN) rC Z
email.charsetr5 Z
email.messageZ email.errorsr r Zraw_data_managerr' r
r+ Zsplitr r- r8 r/ r: r@ rG rX r^ r Zstrra r0 ZMessagerb r. Z bytearrayZ
memoryviewr r r r r Z<module> sD 6
'