B
Uý0<8Vü^ã @ sæ d ddgZ ddlZddlZddlZddlZddlmZ ddlmZ e j
Z
dZ dZ d Z
d
Zd Zd Zd
Ze
dƒZe
dƒZe dejejB ¡Ze d¡Ze d¡ZejjZdd„ Z ddd„Z G dd „ d ƒZ G dd„ dƒZ G dd„ de ƒZ!dS )ÚHeaderÚ
decode_headerÚ make_headeré N)ÚHeaderParseError)ÚcharsetÚ
ú ó z Ú éN z zus-asciizutf-8ai
=\? # literal =?
(?P[^?]*?) # non-greedy up to the next ? is the charset
\? # literal ?
(?P[qQbB]) # 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 | dƒrdd„ | jD ƒS t | ¡s.| d fgS g }x| ¡ D ]„}t |¡}d}xp|r¾| d¡}|rn| ¡ }d}|r‚| |d d f¡ |rP| d¡ ¡ }| d¡ ¡ }| d¡}| |||f¡ qPW qM s z!decode_header..Tr Fé é ÚqÚbé z===zBase64 decoding errorzUnexpected encoding: zraw-unicode-escape)Úhasattrr ÚecreÚsearchÚ
splitlinesÚsplitÚpopÚlstripÚappendÚlowerÚ enumerateÚisspaceÚreversedÚemailÚ
quoprimimeZ
header_decodeÚlenZ
base64mimeÚdecodeÚbinasciiÚErrorr ÚAssertionErrorÚ
isinstancer ÚbytesÚBSPACE)ÚheaderZwordsÚlineÚpartsÚfirstZ unencodedr ÚencodingZencodedZdroplistÚnÚwÚdZ
decoded_wordsZencoded_stringZwordZpaderrZ collapsedZ last_wordÚ last_charsetr
r
r r = sz
4
c C sJ t |||d}x6| D ].\}}|d k r6t|tƒs6t|ƒ}| ||¡ qW |S )N)Ú
maxlinelenÚ header_nameÚcontinuation_ws)r r, ÚCharsetr )Z decoded_seqr8 r9 r: ÚhÚsr r
r
r r œ s c @ sJ e Zd Zddd„Zdd„ Zdd „ Zdd
d „Zd d
„ Zddd„Zdd„ Z dS )r Nr Ústrictc C s| |d krt }nt|tƒs t|ƒ}|| _|| _g | _|d k rH| |||¡ |d krTt}|| _|d krjd| _ nt
|ƒd | _ d S )Nr r ) ÚUSASCIIr, r; r Ú_continuation_wsr r Ú
MAXLINELENÚ _maxlinelenÚ
_headerlenr' )Úselfr= r r8 r9 r: Úerrorsr
r
r Ú__init__´ s
zHeader.__init__c C sÈ | ¡ g }d }d }x¨| jD ]ž\}}|}|tjkrJ| dd¡}| dd¡}|rš|o^| |d ¡}|dkr„|dkrš|sš| t¡ d }n|dkrš|sš| t¡ |oª| |d ¡}|}| |¡ q W t
|¡S )NÚasciiÚsurrogateescapeÚreplacer )Nzus-asciiéÿÿÿÿ) Ú
_normalizer r Ú UNKNOWN8BITÚencoder( Ú _nonctextr ÚSPACEÚ EMPTYSTRINGÚjoin) rD ZuchunksÚlastcsÚ lastspacer r ZnextcsZoriginal_bytesÚhasspacer
r
r Ú__str__ã s*
zHeader.__str__c C s |t | ƒkS )N)r )rD Úotherr
r
r Ú__eq__ s z
Header.__eq__c C s¶ |d kr| j }nt|tƒs"t|ƒ}t|tƒsZ|jp4d}|t jkrN| dd¡}n | ||¡}|jpbd}|t jkr¢y| ||¡ W n" t k
r |dkr˜‚ t
}Y nX | j ||f¡ d S )Nzus-asciirH )
r r, r; r Z input_codecrL r( Z output_codecrM ÚUnicodeEncodeErrorÚUTF8r r )rD r= r rE Z
input_charsetÚoutput_charsetr
r
r r s$
z
Header.appendc C s | ¡ p|dkS )N)ú(ú)ú\)r# )rD r= r
r
r rN 4 s zHeader._nonctextú;, r c C sª | ¡ |d kr| j}|dkr"d}t| j|| j|ƒ}d }d }}x.| jD ]"\}} |d k r |ol| |d ¡}|dkrŒ|r‚| dkr | ¡ n| dkr |s | ¡ |o°| |d ¡}| }d}| ¡ }
|
rÚ| d|
d | ¡ n| dd| ¡ xp|
dd … D ]`} |
¡ | j d k r(| | jd| ¡ | ¡ qö| ¡ } | d t
| ƒt
| ƒ … }
| |
| | ¡ qöW t
|
ƒdkrJ|
¡ qJW | jr‚| ¡ | |¡}t |¡r¦td |¡ƒ‚|S )
Nr i@B )Nzus-asciirJ Fr
r r z8header value appears to contain an embedded header: {!r})rK rB Ú_ValueFormatterrC r@ r rN Úadd_transitionr ÚfeedÚnewlineÚheader_encodingr r' Ú_strÚ_embedded_headerr r Úformat)rD Ú
splitcharsr8 ÚlinesepÚ formatterrR rT rS r r Úlinesr0 ZslineÚfwsÚvaluer
r
r rM 9 sR !
z
Header.encodec C s| g }d }g }xL| j D ]B\}}||kr0| |¡ q|d k rL| t |¡|f¡ |g}|}qW |rr| t |¡|f¡ || _ d S )N)r r rO rQ )rD Zchunksr7 Z
last_chunkr r r
r
r rK ‰ s zHeader._normalize)NNNNr r> )Nr> )r^ Nr )
Ú__name__Ú
__module__Ú __qualname__rF rU rW r rN rM rK 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 )r_ c C s0 || _ || _t|ƒ| _|| _g | _t|ƒ| _d S )N)Ú_maxlenr@ r' Ú_continuation_ws_lenÚ _splitcharsÚ_linesÚ _AccumulatorÚ
_current_line)rD Z headerlenÚmaxlenr: rg r
r
r rF ž s
z_ValueFormatter.__init__c C s | ¡ | | j¡S )N)rb rQ rs )rD rh r
r
r rd ¦ s z_ValueFormatter._strc C s
| t¡S )N)rd ÚNL)rD r
r
r rU ª s z_ValueFormatter.__str__c C sv | j ¡ }|dkr | j j|Ž t| j ƒdkrh| j ¡ rV| jrV| jd t| j ƒ7 < n| j t| j ƒ¡ | j ¡ d S )N)r r
r rJ ) ru r Úpushr' Ú is_onlywsrs r r Úreset)rD Z end_of_liner
r
r rb s
z_ValueFormatter.newlinec C s | j dd¡ d S )Nr r
)ru rx )rD r
r
r r` ¸ s z _ValueFormatter.add_transitionc C sÄ |j d kr | ||| j¡ d S | || ¡ ¡}y| d¡}W n tk
rP d S X |d k rf| ||¡ y | ¡ }W n tk
r† d S X | ¡ | j
| j |¡ x |D ]}| j
| j | ¡ q¦W d S )Nr )rc Ú _ascii_splitrr Zheader_encode_linesÚ _maxlengthsr Ú
IndexErrorÚ
_append_chunkrb ru rx r@ rs r )rD rk r r Z
encoded_linesZ
first_lineÚ last_liner0 r
r
r ra » s$
z_ValueFormatter.feedc c s* | j t| jƒ V x| j | j V qW d S )N)rp r' ru rq )rD r
r
r r| à s z_ValueFormatter._maxlengthsc C sj t dt d || ¡}|d r0dg|d d…< n
| d¡ x*tt|ƒgd Ž D ]\}}| ||¡ qNW d S )Nz([z]+)r r
r )Úrer ÚFWSr ÚzipÚiterr~ )rD rk r rg r1 Úpartr
r
r r{ æ s
z _ValueFormatter._ascii_splitc C s
| j ||¡ t| j ƒ| jkrxº| jD ]t}xlt| j ¡ d ddƒD ]P}| ¡ rp| j | d }|rp|d |krpP | j |d d }|rD|d |krDP qDW q(P q(W | j ¡ \}}| j j dkrÊ|
¡ |sÊd}| j ||¡ d S | j |¡}| j
t| j ƒ¡ | j |¡ d S )Nr r rJ r )ru rx r' rp rr ÚrangeÚ
part_countr# r Ú
_initial_sizerb Úpop_fromrs r r rz )rD rk r ZchÚiZprevpartr„ Z remainderr
r
r r~ ü s. z _ValueFormatter._append_chunkN) rm rn ro rF rd rU rb r` ra r| r{ r~ r
r
r
r r_ s %r_ c sj e Zd Zd‡ fdd„ Zdd„ Zddd„Z‡ fdd „Zd
d „ Zd d
„ Zddd„Z dd„ Z
‡ fdd„Z ‡ Z S )rt r c s || _ tƒ ¡ d S )N)r‡ ÚsuperrF )rD Z initial_size)Ú __class__r
r rF s z_Accumulator.__init__c C s | ||f¡ d S )N)r )rD rk r r
r
r rx # s z_Accumulator.pushc C s | |d … }g | |d …< |S )Nr
)rD r‰ Zpoppedr
r
r rˆ & s z_Accumulator.pop_fromc s | ¡ dkrdS tƒ ¡ S )Nr )r
r
)r† rŠ r )rD )r‹ r
r r + s z_Accumulator.popc C s t dd„ | D ƒ| jƒS )Nc s s" | ]\}}t |ƒt |ƒ V qd S )N)r' )r rk r„ r
r
r ú 1 s z'_Accumulator.__len__..)Úsumr‡ )rD r
r
r Ú__len__0 s z_Accumulator.__len__c C s t dd„ | D ƒ¡S )Nc s s | ]\}}t ||f¡V qd S )N)rP rQ )r rk r„ r
r
r rΠ5 s z'_Accumulator.__str__..)rP rQ )rD r
r
r rU 4 s
z_Accumulator.__str__Nc C s" |d kr g }|| d d …< d| _ d S )Nr )r‡ )rD Zstartvalr
r
r rz 8 s z_Accumulator.resetc C s | j dko| pt| ƒ ¡ S )Nr )r‡ r r# )rD r
r
r ry > s z_Accumulator.is_onlywsc s
t ƒ ¡ S )N)rŠ rŽ )rD )r‹ r
r r† A s z_Accumulator.part_count)r )r )N)
rm rn ro rF rx rˆ r rŽ rU rz ry r† Ú
__classcell__r
r
)r‹ r rt s
rt )NNr )"Ú__all__r€ r) Zemail.quoprimimer% Zemail.base64mimeZ email.errorsr r r r; rw rO r. ZSPACE8rP rA r r? rY ÚcompileÚVERBOSEÚ MULTILINEr Zfcrere r& Z _max_appendr r r r_ Úlistrt r
r
r
r Ú s>
_
k