3
\&^ @ s d Z dddgZddlZddlZddlZddlZddlmZ ddlm Z
e
j Z dZ d Z
d
ZdZd Zd
ZdZe dZe dZejdejejB ejB ZejdZejdZej j Z dd Z d ddZ G dd dZ G dd dZ!G dd de"Z#dS ) z+Header encoding and decoding functionality.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.| dfgS g }x| j D ]}tj|}d}xp|r|jd}|rn|j }d}|r|j|ddf |rP|jdj }|jdj }|jd}|j|||f qPW qM s z!decode_header..NTr 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\}}|dk r8t|t r8t|}|j|| qW |S )a Create a Header from a sequence of pairs as returned by decode_header()
decode_header() takes a header value string and returns a sequence of
pairs of the format (decoded_string, charset) where charset is the string
name of the character set.
This function takes one of those sequence of pairs and returns a Header
instance. Optional maxlinelen, header_name, and continuation_ws are as in
the Header constructor.
)
maxlinelen header_namecontinuation_wsN)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| |dkrt }nt|ts t|}|| _|| _g | _|dk rH| j||| |dkrTt}|| _|dkrjd| _ nt
|d | _ dS )aD Create a MIME-compliant header that can contain many character sets.
Optional s is the initial header value. If None, the initial header
value is not set. You can later append to the header with .append()
method calls. s may be a byte string or a Unicode string, but see the
.append() documentation for semantics.
Optional charset serves two purposes: it has the same meaning as the
charset argument to the .append() method. It also sets the default
character set for all subsequent .append() calls that omit the charset
argument. If charset is not provided in the constructor, the us-ascii
charset is used both as s's initial charset and as the default for
subsequent .append() calls.
The maximum line length can be specified explicitly via maxlinelen. For
splitting the first line to a shorter value (to account for the field
header which isn't included in s, e.g. `Subject') pass in the name of
the field in header_name. The default maxlinelen is 78 as recommended
by RFC 2822.
continuation_ws must be RFC 2822 compliant folding whitespace (usually
either a space or a hard tab) which will be prepended to continuation
lines.
errors is passed through to the .append() call.
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|d kr| r|jt d}n|d
kr| r|jt |o| j|d }|}|j| q W t j
|S ) z&Return the string value of the header.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 |dkr| 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 dS )a. Append a string to the MIME header.
Optional charset, if given, should be a Charset instance or the name
of a character set (which will be converted to a Charset instance). A
value of None (the default) means that the charset given in the
constructor is used.
s may be a byte string or a Unicode string. If it is a byte string
(i.e. isinstance(s, str) is false), then charset is the encoding of
that byte string, and a UnicodeError will be raised if the string
cannot be decoded with that charset. If s is a Unicode string, then
charset is a hint specifying the character set of the characters in
the string. In either case, when producing an RFC 2822 compliant
header using RFC 2047 rules, the string will be encoded using the
output codec of the charset. If the string cannot be encoded to the
output codec, a UnicodeError will be raised.
Optional `errors' is passed as the errors argument to the decode
call if s is a byte string.
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 )z=True if string s is not a ctext character of RFC822.
()\)r\ r] r^ )r$ )rE r> r r r rP 4 s zHeader._nonctext;, r c C s | j |dkr| j}|dkr"d}t| j|| j|}d}d }}x4| jD ](\}} |dk 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 dk r,|j | jd| j | n.| j } | dt
| t
| }
|j |
| | qW t
|
dkrJ|j
qJW | jr|j |j|}tj|rtd j||S )a Encode a message header into an RFC-compliant format.
There are many issues involved in converting a given string for use in
an email header. Only certain character sets are readable in most
email clients, and as header strings can only contain a subset of
7-bit ASCII, care must be taken to properly convert and encode (with
Base64 or quoted-printable) header strings. In addition, there is a
75-character length limit on any given encoded header field, so
line-wrapping must be performed, even with double-byte character sets.
Optional maxlinelen specifies the maximum length of each generated
line, exclusive of the linesep string. Individual lines may be longer
than maxlinelen if a folding point cannot be found. The first line
will be shorter by the length of the header name plus ": " if a header
name was specified at Header construction time. The default value for
maxlinelen is determined at header construction time.
Optional splitchars is a string containing characters which should be
given extra weight by the splitting algorithm during normal header
wrapping. This is in very rough support of RFC 2822's `higher level
syntactic breaks': split points preceded by a splitchar are preferred
during line splitting, with the characters preferred in the order in
which they appear in the string. Space and tab may be included in the
string to indicate whether preference should be given to one over the
other as a split point when other split chars do not appear in the line
being split. Splitchars does not affect RFC 2047 encoded lines.
Optional linesep is a string to be used to separate the lines of
the value. The default value is the most useful for typical
Python applications, but it can be set to \r\n to produce RFC-compliant
line separators when needed.
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 )$__doc____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