3
\&^ @ s d ddgZ ddlZddlZddlZddlZddlmZ ddlmZ e j
Z
dZ dZ d Z
dZd Zd Zd
Ze
dZe
dZejdejejB ejB ZejdZejdZejj Z dd Z dddZ G dd d Z G dd dZ G dd de!Z"dS ) Header
decode_header make_header N)HeaderParseError)charset
N z zus-asciizutf-8ai
=\? # literal =?
(?P[^?]*?) # non-greedy up to the next ? is the charset
\? # literal ?
(?P[qb]) # either a "q" or a "b", case insensitive
\? # literal ?
(?P.*?) # non-greedy up to the next ?= is the encoded string
\?= # literal ?=
z[\041-\176]+:$z
\n[^ \t]+:c C s t | drdd | jD S tj| s.| d fgS g }x| j D ]}tj|}d}xp|r|jd}|rn|j }d}|r|j|d d f |rP|jdj }|jdj }|jd}|j|||f qPW qM s z!decode_header..Tr F qb z===zBase64 decoding errorzUnexpected encoding: zraw-unicode-escape)hasattrr
ecresearch
splitlinessplitpoplstripappendlower enumerateisspacereversedemail
quoprimimeZ
header_decodelenZ
base64mimedecodebinasciiErrorr AssertionError
isinstancer bytesBSPACE)headerZwordslinepartsfirstZ unencodedr encodingZencodedZdroplistnwdZ
decoded_wordsZencoded_stringZwordZpaderrZ collapsedZ last_word last_charsetr r r r = sz
4
c C sL t |||d}x8| D ]0\}}|d k r8t|t r8t|}|j|| qW |S )N)
maxlinelen header_namecontinuation_ws)r r- Charsetr! )Z decoded_seqr9 r: r; hsr r r r r s c @ sJ e Zd ZdddZdd Zdd Zdd
d Zd d
ZdddZdd Z dS )r Nr strictc C s| |d krt }nt|ts t|}|| _|| _g | _|d k rH| j||| |d krTt}|| _|d krjd| _ nt
|d | _ d S )Nr r ) USASCIIr- r< r _continuation_wsr
r!
MAXLINELEN _maxlinelen
_headerlenr( )selfr> r r9 r: r; errorsr r r __init__ s
zHeader.__init__c C s | j g }d }d }x| jD ]\}}|}|tjkrJ|jdd}|jdd}|r|o^| j|d }|dkr|dkr| r|jt d }n|d kr| r|jt |o| j|d
}|}|j| q W t j
|S ) Nasciisurrogateescapereplacer us-asciir )NrK )NrK )NrK )
_normalizer
r UNKNOWN8BITencoder) _nonctextr! SPACE EMPTYSTRINGjoin) rE Zuchunkslastcs lastspacer r ZnextcsZoriginal_byteshasspacer r r __str__ s*
zHeader.__str__c C s |t | kS )N)r )rE otherr r r __eq__ s z
Header.__eq__c C s |d kr| j }nt|ts"t|}t|tsZ|jp4d}|t jkrN|jdd}n |j||}|jpbd}|t jkry|j|| W n" t k
r |dkr t
}Y nX | j j ||f d S )Nzus-asciirI )
r r- r< r Z input_codecrN r) Z output_codecrO UnicodeEncodeErrorUTF8r
r! )rE r> r rF Z
input_charsetZoutput_charsetr r r r! s$
z
Header.appendc C s |j p|dkS )N()\)r\ r] r^ )r$ )rE r> r r r rP 4 s zHeader._nonctext;, r c C s | j |d kr| j}|dkr"d}t| j|| j|}d }d }}x4| jD ](\}} |d k r|ol| j|d }|d kr| s| d
kr|j n| d kr| r|j |o| j|d }| }d}|j }
|
r|j d|
d | n|j dd| xr|
dd D ]b} |j
| j d k r,|j | jd| j | n.| j } | d t
| t
| }
|j |
| | qW t
|
dkrJ|j
qJW | jr|j |j|}tj|rtdj||S )
Nr i@B us-asciir Fr r z8header value appears to contain an embedded header: {!r})Nr` )Nr` )Nr` rL )rM rC _ValueFormatterrD rA r
rP add_transitionr feednewlineheader_encodingr r( _str_embedded_headerr r format)rE
splitcharsr9 linesepZ formatterrT rV rU r r linesr1 Zslinefwsvaluer r r rO 9 sR !
z
Header.encodec C s| g }d }g }xL| j D ]B\}}||kr0|j| q|d k rL|jtj||f |g}|}qW |rr|jtj||f || _ d S )N)r
r! rQ rS )rE Zchunksr8 Z
last_chunkr r r r r rM s zHeader._normalize)NNNNr r? )Nr? )r_ Nr )
__name__
__module__ __qualname__rG rW rY r! rP rO rM r r r r r s
-
+
Pc @ sT e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dS )ra c C s0 || _ || _t|| _|| _g | _t|| _d S )N)_maxlenrA r( _continuation_ws_len _splitchars_lines _Accumulator
_current_line)rE Z headerlenmaxlenr; ri r r r rG s
z_ValueFormatter.__init__c C s | j |j| jS )N)rd rS rt )rE rj r r r rf s z_ValueFormatter._strc C s
| j tS )N)rf NL)rE r r r rW s z_ValueFormatter.__str__c C sp | j j }|dkr | j j| t| j dkrb| j j rP| jd t| j 7 < n| jjt| j | j j d S )Nr r r r )r r rL ) rv r pushr( is_onlywsrt r r! reset)rE Z end_of_liner r r rd s
z_ValueFormatter.newlinec C s | j jdd d S )Nr r )rv ry )rE r r r rb s z _ValueFormatter.add_transitionc C s |j d kr | j||| j d S |j|| j }y|jd}W n tk
rP d S X |d k rf| j|| y |j }W n tk
r d S X | j | j j
| j | x |D ]}| j j
| j | qW d S )Nr )re _ascii_splitrs Zheader_encode_lines _maxlengthsr
IndexError
_append_chunkrd rv ry rA rt r! )rE rl r r Z
encoded_linesZ
first_line last_liner1 r r r rc s$
z_ValueFormatter.feedc c s* | j t| j V x| j | j V qW d S )N)rq r( rv rr )rE r r r r} s z_ValueFormatter._maxlengthsc C sj t jdt d || }|d r0dg|d d< n
|jd x*tt|gd D ]\}}| j|| qNW d S )Nz([z]+)r r r )rer FWSr zipiterr )rE rl r ri r2 partr r r r| s
z _ValueFormatter._ascii_splitc C s
| j j|| t| j | jkrx| jD ]t}xlt| j j d ddD ]P}|j rp| j | d }|rp|d |krpP | j |d d }|rD|d |krDP qDW q(P q(W | j j \}}| j j dkr| j
|sd}| j j|| d S | j j |}| j j
t| j | j j| d S )Nr r r rL rL )rv ry r( rq rs range
part_countr$ r
_initial_sizerd pop_fromrt r! r r{ )rE rl r ZchiZprevpartr Z remainderr r r r s. z _ValueFormatter._append_chunkN) rn ro rp rG rf rW rd rb rc r} r| r r r r r ra s %ra c sj e Zd Zd fdd Zdd ZdddZ fdd Zd
d Zd d
ZdddZ dd Z
fddZ Z S )ru r c s || _ t j d S )N)r superrG )rE Z initial_size) __class__r r rG s z_Accumulator.__init__c C s | j ||f d S )N)r! )rE rl r r r r ry # s z_Accumulator.pushc C s | |d }g | |d < |S )Nr )rE r Zpoppedr r r r & s z_Accumulator.pop_fromc s | j dkrdS t j S )Nr r )r r )r r r )rE )r r r r + s z_Accumulator.popc C s t dd | D | jS )Nc s s" | ]\}}t |t | V qd S )N)r( )r rl r r r r 1 s z'_Accumulator.__len__..)sumr )rE r r r __len__0 s z_Accumulator.__len__c C s t jdd | D S )Nc s s | ]\}}t j||fV qd S )N)rR rS )r rl r r r r r 5 s z'_Accumulator.__str__..)rR rS )rE r r r rW 4 s
z_Accumulator.__str__Nc C s" |d kr g }|| d d < d| _ d S )Nr )r )rE Zstartvalr r r r{ 8 s z_Accumulator.resetc C s | j dko| pt| j S )Nr )r r r$ )rE r r r rz > s z_Accumulator.is_onlywsc s
t j S )N)r r )rE )r r r r A s z_Accumulator.part_count)r )r )N)
rn ro rp rG ry r r r rW r{ rz r
__classcell__r r )r r ru s
ru z )NNr )#__all__r r* Zemail.quoprimimer& Zemail.base64mimeZ email.errorsr r r r< rx rQ r/ ZSPACE8rR rB r r@ r[ compileVERBOSE
IGNORECASE MULTILINEr Zfcrerg r' Z _max_appendr r r ra listru r r r r s>
_
k