3
\O � @ sf d dl mZ d dlmZ d dlmZ d dlmZ G dd� d�ZG dd� d�ZG d d
� d
e �Z
d d � Z G d
d� d�Z G dd� de �Z
G dd� d�ZG dd� de�ZG dd� d�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d �ZG d d � d �ZG d!d"� d"e�ZG d#d$� d$e�ZG d%d&� d&�Ze
eeeeeeeeeeeeeeeeeed'�ZG d(d)� d)�Zd*S )+� )�MappingProxyType)�utils)�errors)�_header_value_parserc @ s^ e Zd Zddd�Zedd� �Zedd� �Zed d
� �Zed d � �Zd
d� Z dd� Z
dd� Z dS )�Address� Nc C sl |d k rV|s|rt d��tj|�\}}|r:tdj||���|jrJ|jd �|j}|j}|| _|| _ || _
d S )Nz=addrspec specified when username and/or domain also specifiedz6Invalid addr_spec; only '{}' could be parsed from '{}'r ) � TypeError�parserZ
get_addr_spec�
ValueError�format� all_defects�
local_part�domain�
_display_name� _username�_domain)�self� display_name�usernamer � addr_specZa_s�rest� r �,/usr/lib64/python3.6/email/headerregistry.py�__init__ s
zAddress.__init__c C s | j S )N)r )r r r r r 7 s zAddress.display_namec C s | j S )N)r )r r r r r ; s zAddress.usernamec C s | j S )N)r )r r r r r ? s zAddress.domainc C sT t | j�}t|�t|tj �kr.tj| j�}n| j}| jrH|d | j S |sPdS |S )N�@z<>)�setr �lenr Z
DOT_ATOM_ENDS� quote_stringr )r �namesetZlpr r r r C s
zAddress.addr_specc C s dj | jj| j| j| j�S )Nz1{}(display_name={!r}, username={!r}, domain={!r}))r � __class__�__name__r r r )r r r r �__repr__S s zAddress.__repr__c C s^ t | j�}t|�t|tj �kr.tj| j�}n| j}|rX| jdkrFdn| j}dj||�S | jS )Nz<>r z{} <{}>)r r r r �SPECIALSr r r )r r �dispr r r r �__str__X s
zAddress.__str__c C s8 t |�t | �krdS | j|jko6| j|jko6| j|jkS )NF)�typer r r )r �otherr r r �__eq__c s
zAddress.__eq__)r r r N) r �
__module__� __qualname__r �propertyr r r r r! r$ r' r r r r r s
% r c @ sF e Zd Zddd�Zedd� �Zedd� �Zdd � Zd
d � Zd d
� Z dS )�GroupNc C s || _ |rt|�nt� | _d S )N)r �tuple�
_addresses)r r � addressesr r r r m s zGroup.__init__c C s | j S )N)r )r r r r r s zGroup.display_namec C s | j S )N)r- )r r r r r. � s zGroup.addressesc C s dj | jj| j| j�S )Nz${}(display_name={!r}, addresses={!r})r r r r r. )r r r r r! � s zGroup.__repr__c C s� | j d kr&t| j�dkr&t| jd �S | j }|d k r\t|�}t|�t|tj �kr\tj|�}djdd� | jD ��}|r~d| n|}dj ||�S )N� r z, c s s | ]}t |�V qd S )N)�str)�.0�xr r r � � s z Group.__str__..� z{}:{};)
r r r. r0 r r r" r �joinr )r r# r Zadrstrr r r r$ � s
z
Group.__str__c C s, t |�t | �krdS | j|jko*| j|jkS )NF)r% r r. )r r& r r r r' � s z Group.__eq__)NN)
r r( r) r r* r r. r! r$ r' r r r r r+ k s
r+ c @ sP e Zd Zdd� Zdd� Zedd� �Zedd� �Zd d
� Ze d d � �Z
d
d� Z dS )�
BaseHeaderc C s\ dg i}| j ||� tj|d �r4tj|d �|d< tj| |d �}|d= |j|f|� |S )N�defects�decoded)�parser Z_has_surrogates� _sanitizer0 �__new__�init)�cls�name�value�kwdsr r r r r; � s zBaseHeader.__new__c C s || _ || _|| _d S )N)�_name� _parse_tree�_defects)r r> �
parse_treer7 r r r r< � s zBaseHeader.initc C s | j S )N)rA )r r r r r> � s zBaseHeader.namec C s
t | j�S )N)r, rC )r r r r r7 � s zBaseHeader.defectsc C s t | jj| jjt| �f| jfS )N)�_reconstruct_headerr r � __bases__r0 �__dict__)r r r r �
__reduce__� s
zBaseHeader.__reduce__c C s t j| |�S )N)r0 r; )r= r? r r r � _reconstruct� s zBaseHeader._reconstructc C s` t jt jt j| jd�t jdd�g�g�}| jrH|jt jt jdd�g�� |j| j� |j |d�S )Nz header-name�:z
header-sepr4 Zfws)�policy)
r ZHeaderZ HeaderLabelZ
ValueTerminalr> rB �appendZCFWSListZWhiteSpaceTerminal�fold)r rK �headerr r r rM � s zBaseHeader.foldN) r r( r) r; r< r* r> r7 rH � classmethodrI rM r r r r r6 � s "
r6 c C s t | |i �j|�S )N)r% rI )Zcls_name�basesr? r r r rE s rE c @ s& e Zd ZdZeej�Zedd� �Z dS )�UnstructuredHeaderNc C s" | j |�|d< t|d �|d< d S )NrD r8 )� value_parserr0 )r= r? r@ r r r r9 s zUnstructuredHeader.parse)
r r( r) � max_count� staticmethodr �get_unstructuredrR rO r9 r r r r rQ s
rQ c @ s e Zd ZdZdS )�UniqueUnstructuredHeaderr/ N)r r( r) rS r r r r rV s rV c sB e Zd ZdZeej�Zedd� �Z � fdd�Z
e dd� �Z � Z
S )�
DateHeaderNc C sz |s6|d j tj� � d |d< d|d< tj� |d< d S t|t�rJtj|�}||d< tj |d �|d< | j
|d �|d< d S )Nr7 �datetimer r8 rD ) rL r ZHeaderMissingRequiredValuer Z TokenList�
isinstancer0 r Zparsedate_to_datetimeZformat_datetimerR )r= r? r@ r r r r9 ) s
zDateHeader.parsec s |j d�| _t� j||� d S )NrX )�pop� _datetime�superr< )r �args�kw)r r r r< 7 s zDateHeader.initc C s | j S )N)r[ )r r r r rX ; s zDateHeader.datetime)r r( r) rS rT r rU rR rO r9 r< r* rX �
__classcell__r r )r r rW s
rW c @ s e Zd ZdZdS )�UniqueDateHeaderr/ N)r r( r) rS r r r r r` @ s r` c sP e Zd ZdZedd� �Zedd� �Z� fdd�Ze dd � �Z
e d
d � �Z � Z S ) �
AddressHeaderNc C s t j| �\}} |S )N)r Zget_address_list)r? � address_listr r r rR I s zAddressHeader.value_parserc C s� t |t�rZ| j|� |d< }g }x,|jD ]"}|jt|jdd� |jD ��� q(W t|j �}n"t
|d�sj|g}dd� |D �}g }||d< ||d< dj d d� |D ��|d
< d|kr�| j|d
�|d< d S ) NrD c S s* g | ]"}t |jpd |jpd |jp"d ��qS )r )r r r
r )r1 Zmbr r r �
X s z'AddressHeader.parse..�__iter__c S s& g | ] }t |d �s td|g�n|�qS )r. N)�hasattrr+ )r1 �itemr r r rc a s �groupsr7 z, c S s g | ] }t |��qS r )r0 )r1 rf r r r rc g s r8 ) rY r0 rR r. rL r+ r Z
all_mailboxes�listr re r5 )r= r? r@ rb rg Zaddrr7 r r r r9 O s$
zAddressHeader.parsec s( t |jd��| _d | _t� j||� d S )Nrg )r, rZ �_groupsr- r\ r< )r r] r^ )r r r r< k s zAddressHeader.initc C s | j S )N)ri )r r r r rg p s zAddressHeader.groupsc C s& | j d kr tdd� | jD ��| _ | j S )Nc S s g | ]}|j D ]}|�qqS r )r. )r1 �group�addressr r r rc w s z+AddressHeader.addresses..)r- r, ri )r r r r r. t s
zAddressHeader.addresses)
r r( r) rS rT rR rO r9 r< r* rg r. r_ r r )r r ra E s ra c @ s e Zd ZdZdS )�UniqueAddressHeaderr/ N)r r( r) rS r r r r rl | s rl c @ s e Zd Zedd� �ZdS )�SingleAddressHeaderc C s( t | j�dkr tdj| j���| jd S )Nr/ z9value of single address header {} is not a single addressr )r r. r
r r> )r r r r rk � s
zSingleAddressHeader.addressN)r r( r) r* rk r r r r rm � s rm c @ s e Zd ZdZdS )�UniqueSingleAddressHeaderr/ N)r r( r) rS r r r r rn � s rn c sZ e Zd ZdZeej�Zedd� �Z � fdd�Z
e dd� �Z e dd � �Z
e d
d � �Z� ZS ) �MIMEVersionHeaderr/ c C s� | j |� |d< }t|�|d< |d j|j� |jd kr r? r r r �
� s z1ParameterizedMIMEHeader.parse..)rR r0 rs r rx )r= r? r@ rD r r r r9 � s
z ParameterizedMIMEHeader.parsec s |j d�| _t� j||� d S )Nrx )rZ �_paramsr\ r< )r r] r^ )r r r r< � s z ParameterizedMIMEHeader.initc C s
t | j�S )N)r r{ )r r r r rx � s z ParameterizedMIMEHeader.params)
r r( r) rS rO r9 r< r* rx r_ r r )r r rw � s
rw c sJ e Zd Zeej�Z� fdd�Zedd� �Z edd� �Z
edd� �Z � Z S ) �ContentTypeHeaderc s2 t � j||� tj| jj�| _tj| jj�| _d S )N) r\ r< r r: rB �maintype� _maintype�subtype�_subtype)r r] r^ )r r r r< � s zContentTypeHeader.initc C s | j S )N)r~ )r r r r r} � s zContentTypeHeader.maintypec C s | j S )N)r� )r r r r r � s zContentTypeHeader.subtypec C s | j d | j S )N�/)r} r )r r r r � content_type� s z ContentTypeHeader.content_type)
r r( r) rT r Zparse_content_type_headerrR r< r* r} r r� r_ r r )r r r| � s
r| c s2 e Zd Zeej�Z� fdd�Zedd� �Z � Z
S )�ContentDispositionHeaderc s2 t � j||� | jj}|d kr"|ntj|�| _d S )N)r\ r< rB �content_dispositionr r: �_content_disposition)r r] r^ Zcd)r r r r< � s z ContentDispositionHeader.initc C s | j S )N)r� )r r r r r� � s z,ContentDispositionHeader.content_disposition) r r( r) rT r Z parse_content_disposition_headerrR r< r* r� r_ r r )r r r� � s
r� c sB e Zd ZdZeej�Zedd� �Z � fdd�Z
e dd� �Z � Z
S )� ContentTransferEncodingHeaderr/ c C s2 | j |� |d< }t|�|d< |d j|j� d S )NrD r8 r7 )rR r0 rs r )r= r? r@ rD r r r r9 � s z#ContentTransferEncodingHeader.parsec s" t � j||� tj| jj�| _d S )N)r\ r< r r: rB �cte�_cte)r r] r^ )r r r r< s z"ContentTransferEncodingHeader.initc C s | j S )N)r� )r r r r r� s z!ContentTransferEncodingHeader.cte)r r( r) rS rT r Z&parse_content_transfer_encoding_headerrR rO r9 r< r* r� r_ r r )r r r� � s
r� )ZsubjectZdatez resent-datez orig-dateZsenderz
resent-senderZtoz resent-toZccz resent-ccZbccz
resent-bcc�fromz resent-fromzreply-toz mime-versionz content-typezcontent-dispositionzcontent-transfer-encodingc @ s4 e Zd Zeedfdd�Zdd� Zdd� Zdd � Zd
S ) �HeaderRegistryTc C s& i | _ || _|| _|r"| j jt� d S )N)�registry�
base_class�
default_class�update�_default_header_map)r r� r� Zuse_default_mapr r r r ' s
zHeaderRegistry.__init__c C s || j |j� < d S )N)r� ry )r r> r= r r r � map_to_type9 s zHeaderRegistry.map_to_typec C s, | j j|j� | j�}td|j || jfi �S )N�_)r� �getry r� r% r r� )r r> r= r r r � __getitem__? s zHeaderRegistry.__getitem__c C s | | ||�S )Nr )r r> r? r r r �__call__C s
zHeaderRegistry.__call__N) r r( r) r6 rQ r r� r� r� r r r r r� # s
r� N)�typesr Zemailr r r r r r+ r0 r6 rE rQ rV rW r` ra rl rm rn ro rw r| r� r� r� r� r r r r �
sP [6d '7
%