B
� f�) � @ 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 �9/opt/alt/python37/lib64/python3.7/email/contentmanager.py�__init__ s zContentManager.__init__c C s || j |< d S )N)r )r �key�handlerr r r �add_get_handler
s 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 s || j |< d S )N)r )r Ztypekeyr r r r �add_set_handler s z ContentManager.add_set_handlerc O s> |� � dkrtd��| �||�}|�� |||f|�|� d S )NZ multipartz"set_content not valid on multipart)r � 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
�.)�type�__mro__r � __qualname__�getattr�join�__name__r ) r r r Zfull_path_for_error�typZqname�modname� 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�charset�ASCII)�errors)� get_payloadZ get_paramr' )r r* Zcontentr( r r r �get_text_content@ s r, �textc C s | j dd�S )NT)r' )r+ )r r r r �get_non_text_contentG s r. z audio image video applicationc C s
| � d�S )Nr )r+ )r r r r �get_message_contentM s r/ zrfc822 external-bodyzmessage/c C s t | �d��S )Nr )�bytesr+ )r r r r �%get_and_fixup_unknown_message_contentS s r1 �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 ) N�/z Content-Typer r% c s g | ]}� j � �|g�� �qS r )Zheader_factoryZheader_source_parse)�.0�header)�mpr r �
<listcomp>d s z _prepare_set.<locals>.<listcomp>zInvalid header: {})�policy) r �hasattrr8 Zdefectsr% �emailr* Z HeaderDefect�
ValueError�formatZfold)r r �subtype�headersr5 �excr )r6 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)r5 r& z
Content-ID)� set_param�items)r � dispositionrA �cid�paramsr
�valuer r r �
_finalize_setp s rH c C sZ g }|d d }x>t dt| �|�D ]*}| ||| � }|�t�|��d�� q"W d�|�S )N� � r �asciir
)�range�len�append�binascii�
b2a_base64r' r )�data�max_line_lengthZ
encoded_linesZunencoded_bytes_per_line�iZthisliner r r �_encode_base64� s rT 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 )NrK c s � � | �� S )N)r )�lines)�linesepr r �
embedded_body� � z#_encode_text.<locals>.embedded_bodyc S s d� | �d S )N�
)r )rU r r r � normal_body� rX z!_encode_text.<locals>.normal_body�7bit�8bitc s s | ]}t |�V qd S )N)rM )r4 �xr r r � <genexpr>� s z _encode_text.<locals>.<genexpr>�surrogateescape�
zlatin-1�base64zquoted-printablez$Unknown content transfer encoding {})�encode�
splitlinesrV r' �UnicodeDecodeErrorZcte_type�maxrR r Z body_encoderO rP rM rT r; r< ) �stringr( �cter8 rU rW rZ ZsniffZsniff_qpZ sniff_base64rQ r )rV r � _encode_text� s@
rh �plain�utf-8c
C sd t | d|| � t|||| j�\}}
| �|
� | jdtjj�||�dd� || d< t | ||||� d S )Nr- r( T)r& zContent-Transfer-Encoding)
r@ rh r8 � set_payloadrB r: r( ZALIASES�getrH ) r rf r= r( rg rD rA rE rF r> Zpayloadr r r �set_text_content� s
rm �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 )
N�partialz4message/partial is not supported for Message objectsrn )Nr[ r\ �binaryz*message/rfc822 parts do not support cte={}r\ z
external-body)Nr[ z1message/external-body parts do not support cte={}r[ r2 zContent-Transfer-Encoding)r; r< r@ rk rH ) r r2 r= rg rD rA rE rF r> r r r �set_message_content� s$ rq ra c
C s� t | ||| � |dkr(t|| jjd�}nN|dkrNtj|dddd�}|�d�}n(|dkrb|�d� n|d krv|�dd
�}| �|� || d < t | ||||� d S ) Nra )rR zquoted-printableFT)�istextr5 Z quotetabsrK r[ )r\ rp r_ zContent-Transfer-Encoding)
r@ rT r8 rR rO Zb2a_qpr' rb rk rH )
r rQ r r= rg rD rA rE rF r> r r r �set_bytes_content� s
rs )r& )ri rj NNNNNN)rn NNNNNN)ra NNNNN) rO Z
email.charsetr: Z
email.messageZ email.errorsr r Zraw_data_managerr, r r. �splitr r/ r= r1 r@ rH rT rh rm r �strrq r2 ZMessagers r0 � bytearray�
memoryviewr" r r r r �<module> sD 6
'