3
_Vj#~ @ s^ d ddddddddd d
d d d
gZ ddlZddlZddlZddlZddlZddlmZ ddlZddl Z ddl
Z
ddl Z ddl m
Z
ddlmZ ejddedZe jej G dd deZG dd deZejdd ZG dd deZd d Z d d Z d d Z d d Z G d d dZ d!Z!e
j"d"Z#dOd$d%Z$d&d' Z%e
j"d(e
j&Z'd)d* Z(d+d, Z)d-d. Z*G d/d0 d0Z+G d1d2 d2Z,G d3d4 d4e,Z-d5d6 Z.G d7d de/Z0d8d9d:d;d<Z1e2 Z3G d=d> d>Z4d?d@ Z5e6e7e8eje9fZ:G dAdB dBe2Z;dCdD Zee )r r> r r r r' s
z
Data.__init__c C s | t |S )N)_decode_base64)r4 r> r r r
fromBase64 s zData.fromBase64L c C s t | j|S )N)_encode_base64r> )r
maxlinelengthr r r asBase64 s z
Data.asBase64c C s4 t || jr| j|jkS t |tr,| j|kS tS d S )N)r+ r) r> rA NotImplemented)r otherr r r __eq__ s
z Data.__eq__c C s d| j jt| jf S )Nz%s(%s))r) r" reprr> )r r r r __repr__ s z
Data.__repr__N)rE ) r" r# r$ r' r8 rD rH rK rM r r r r r s
s
zv[\x00\x01\x02\x03\x04\x05\x06\x07\x08\x0b\x0c\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f]rE c C sT |d d }g }x8t dt| |D ]$}| ||| }|jtj| q"W dj|S )N r )rangelenappendbinasciiZ
b2a_base64join)srG Z
maxbinsizepiecesichunkr r r rF s rF c C s( t | trtj| jdS tj| S d S )Nzutf-8)r+ r, rT Z
a2b_base64encode)rV r r r rC
s
rC z{(?P\d\d\d\d)(?:-(?P\d\d)(?:-(?P\d\d)(?:T(?P\d\d)(?::(?P\d\d)(?::(?P\d\d))?)?)?)?)?Zc C sN d}t j| j }g }x,|D ]$}|| }|d kr2P |jt| q W tj| S )Nyearmonthdayhourminutesecond)r[ r\ r] r^ r_ r` ) _dateParsermatch groupdictrS intdatetime)rV orderZgdZlstkeyvalr r r _date_from_string s
ri c C s d| j | j| j| j| j| jf S )Nz %04d-%02d-%02dT%02d:%02d:%02dZ)r[ r\ r] r^ r_ r` )dr r r _date_to_string& s rk c C sZ t j| }|d k rtd| jdd} | jdd} | jdd} | jdd} | jd d
} | S ) Nzz>)_controlCharPatsearch
ValueErrorreplace)textmr r r _escape, s
rw c @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'S )( _PlistParserc C s" g | _ d | _d | _|| _|| _d S )N)stack current_keyroot_use_builtin_types
_dict_type)r r: r; r r r r' 9 s
z_PlistParser.__init__c C s8 t | _| j| j_| j| j_| j| j_| jj| | j S )N)
r parserhandle_begin_elementZStartElementHandlerhandle_end_elementZEndElementHandler handle_dataZCharacterDataHandlerZ ParseFiler{ )r Zfileobjr r r parse@ s
z_PlistParser.parsec C s* g | _ t| d| d }|d k r&|| d S )NZbegin_)r> getattr)r elementattrshandlerr r r r H s z!_PlistParser.handle_begin_elementc C s" t | d| d }|d k r | d S )NZend_)r )r r r r r r r N s z _PlistParser.handle_end_elementc C s | j j| d S )N)r> rS )r r> r r r r S s z_PlistParser.handle_datac C s | j d k rFt| jd ti s.td| jj || jd | j < d | _ nB| jsT|| _n4t| jd tg sxtd| jj | jd j| d S )N z unexpected element at line %dr r r ) rz r+ ry typers r~ CurrentLineNumberr{ rS )r r r r r
add_objectV s
z_PlistParser.add_objectc C s dj | j}g | _|S )N )rU r> )r r> r r r get_dataf s z_PlistParser.get_datac C s" | j }| j| | jj| d S )N)r} r ry rS )r r rj r r r
begin_dictm s
z_PlistParser.begin_dictc C s* | j r td| j | jjf | jj d S )Nz%missing value for key '%s' at line %d)rz rs r~ r ry pop)r r r r end_dictr s z_PlistParser.end_dictc C s: | j s t| jd ti r,td| jj | j | _ d S )Nr zunexpected key at line %dr )rz r+ ry r rs r~ r r )r r r r end_keyx s z_PlistParser.end_keyc C s g }| j | | jj| d S )N)r ry rS )r r ar r r begin_array~ s
z_PlistParser.begin_arrayc C s | j j d S )N)ry r )r r r r end_array s z_PlistParser.end_arrayc C s | j d d S )NT)r )r r r r end_true s z_PlistParser.end_truec C s | j d d S )NF)r )r r r r end_false s z_PlistParser.end_falsec C s | j t| j d S )N)r rd r )r r r r end_integer s z_PlistParser.end_integerc C s | j t| j d S )N)r floatr )r r r r end_real s z_PlistParser.end_realc C s | j | j d S )N)r r )r r r r
end_string s z_PlistParser.end_stringc C s2 | j r| jt| j n| jtj| j d S )N)r| r rC r r rD )r r r r end_data s z_PlistParser.end_datac C s | j t| j d S )N)r ri r )r r r r end_date s z_PlistParser.end_dateN)r" r# r$ r' r r r r r r r r r r r r r r r r r r r r r r rx 8 s& rx c @ s8 e Zd ZdddZdd Zdd Zdd
d Zd d
Zd S )_DumbXMLWriterr c C s || _ g | _|| _|| _d S )N)filery
_indent_levelindent)r r indent_levelr r r r r' s z_DumbXMLWriter.__init__c C s, | j j| | jd| | jd7 _d S )Nz<%s>r )ry rS writelnr )r r r r r
begin_element s z _DumbXMLWriter.begin_elementc C s | j d8 _ | jd| d S )Nr z)r r )r r r r r end_element s z_DumbXMLWriter.end_elementNc C s8 |d k r&t |}| jd|||f n| jd| d S )Nz <%s>%sz<%s/>)rw r )r r r r r r simple_element s z _DumbXMLWriter.simple_elementc C sH |r8t |tr|jd}| jj| j| j | jj| | jjd d S )Nzutf-8
)r+ r, rZ r r7 r r )r liner r r r s
z_DumbXMLWriter.writeln)r r )N)r" r# r$ r' r r r r r r r r r s
r c @ sF e Zd ZdddZdd Zd
d Zd d
Zdd Zdd Zdd Z dS ) _PlistWriterr r TFc C s. |r|j t tj| ||| || _|| _d S )N)r7 PLISTHEADERr r'
_sort_keys _skipkeys)r r r r Z writeHeaderr< r= r r r r' s
z_PlistWriter.__init__c C s" | j d | j| | j d d S )Nzz)r write_value)r r r r r r7 s
z_PlistWriter.writec C s4 t |tr| jd| n|dkr0| jd n |dkrD| jd nt |trd| ko`dk n rx| jd d
| nt|nt |tr| jd t| nt |tr| j| n|t |t r| j
| nft |t t fr| j
| nLt |tjr| jd t| n,t |ttfr | j| ntd
t| d S )NstringTtrueFZfalser ? @ Zintegerz%drealZdatezunsupported type: %sr l l )r+ r, r rd
OverflowErrorr rL dict
write_dictr
write_datarA bytearray write_bytesre rk tuplelist write_arrayrB r )r r r r r r s.
z_PlistWriter.write_valuec C s | j |j d S )N)r r> )r r> r r r r s z_PlistWriter.write_datac C s~ | j d | jd8 _tddt| jjdd | j }x&t||jdD ]}|rL| j| qLW | jd7 _| j d d S )
Nr> r rE r r s )
r r maxrR r rt rF splitr r )r r> rG r r r r r s
z_PlistWriter.write_bytesc C s |rx| j d | jr"t|j }n|j }x@|D ]8\}}t|tsR| jrJq0td| jd| | j | q0W | j
d n
| jd d S )Nr zkeys must be stringsrg ) r r sorteditemsr+ r, r rB r r r )r rj r rg r r r r r s
z_PlistWriter.write_dictc C s@ |r2| j d x|D ]}| j| qW | jd n
| jd d S )Narray)r r r r )r r r r r r r s
z_PlistWriter.write_arrayN)r r r TF)
r" r# r$ r' r7 r r r r r r r r r r s
%
r c C s d }x|D ]}| j |r
dS q
W xntjdftjdftjdffD ]N\}}| j |sRq>x8|D ]0}||jdj| }| d t| |krXdS qXW q>W dS )
N 6xBBQQQi)_fpseekosSEEK_ENDreadrR r structunpack _ref_size
_read_ints_object_offsets
_undefined_objects _read_objectOSError
IndexErrorerrorr rs )r r0 trailer offset_size num_objects
top_objectoffset_table_offsetr r r r _ s
z_BinaryPlistParser.parsec C sL |dkrH| j jdd d@ }d|> }dt| }tj|| j j|d S |S )N r r rO rp )r r _BINARY_FORMATr r )r tokenLrv rV r@ r r r _get_sizex s z _BinaryPlistParser._get_sizec sv | j j| tkr2tjd| t S sHt | krNt t fddtd| D S d S )Nrp c 3 s& | ] }t j || d V qdS )bigN)rd
from_bytes).0rX )r> sizer r s z0_BinaryPlistParser._read_ints..r ) r r r r r rR r r rQ )r nr r )r> r r r s z _BinaryPlistParser._read_intsc C s | j || jS )N)r r )r r r r r
_read_refs s z _BinaryPlistParser._read_refsc s j | }|tk r|S j| } jj| jjdd }|d@ |d@ }}|dkr^d }n|dkrnd}n||dkr~d}nl|dkrd }n\|d
krtj jjd|> d |d kd
}n0|dkrtj d jjdd }n |dkrtj d jjdd }n|dkrDtj d jjdd }t
j
dddt
j |d }n|dkr j |} jj|}t
||krxt jst|}n^|dkrΈ j |} jj|} t
| |krt | jd}n |dkr j |d } jj|} t
| |krt | jd}n|d krX j |} j|}
g }| j |< |j fd d |
D n|d kr j |} j|} j|}
j }| j |< y2x,t| |
D ] \} }
j|
| j| < qW W n tk
r t Y nX nt | j |< |S ) Nr r r r F TrP r r rO )signed" z>frN # z>d3 i )Zsecondsr P r ` r zutf-16be c 3 s | ]} j |V qd S )N)r )r x)r r r r s z2_BinaryPlistParser._read_object.. )r r r r r r rd r r r re Z timedeltar rR r r| r r r extendr} zipr rB )r refresultoffsettokenZtokenHr r@ rV r> Zobj_refsZkey_refskor )r r r s
z _BinaryPlistParser._read_objectN) r" r# r$ r' r r r r r r r r r r S s
r c C s0 | dk r dS | dk rdS | d> d> r(dS dS d S ) Nr r r r r rN i r )countr r r _count_to_size s r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )_BinaryPlistWriterc C s || _ || _|| _d S )N)r r r )r r0 r< r= r r r r' s z_BinaryPlistWriter.__init__c
C s g | _ i | _i | _| j| t| j }dg| | _t|| _t| j | _ | j
j d x| j D ]}| j | q\W | j
|}| j
j }t|}dt| | }| j
j tj|f| j d}||| j|||f} | j
j tjd| d S )Nr s bplist00rp >5xBBBQQQ)r )_objlist _objtable _objidtable_flattenrR r r r r _ref_formatr r7
_write_object
_getrefnumtellr pack)
r r r objr r r Z
offset_formatZ sort_versionr r r r r7
s(
z_BinaryPlistWriter.writec C s~ t |tr"t||f| jkrZd S n8t |trHt|j|jf| jkrZd S nt|| jkrZd S t| j }| j j
| t |tr|| jt||f< n0t |tr|| jt|j|jf< n|| jt|< t |t rPg }g }|j }| j
rt|}xB|D ]:\}}t |ts| jr qtd|j
| |j
| qW xLtj||D ]}| j| q:W n*t |ttfrzx|D ]}| j| qfW d S )Nzkeys must be strings)r+ _scalarsr r r r> idr rR r rS r r r r r, r rB itertoolschainr r r ) r r Zrefnumkeysvaluesr r
vr r r r r 7 sB
z_BinaryPlistWriter._flattenc C sN t |tr | jt||f S t |tr<| jt|j|jf S | jt| S d S )N)r+ r r r r r> r r )r r r r r r g s
z _BinaryPlistWriter._getrefnumc C s |dk r"| j jtjd||B n|dk rH| j jtjd|dB d| nh|dk rn| j jtjd|dB d| nB|dk r| j jtjd
|dB d | n | j jtjd |dB d
| d S )Nr z>Br r z>BBBr z>BBH r z>BBL z>BBQ r i l )r r7 r r )r r r r r r _write_sizeo s z _BinaryPlistWriter._write_sizec s j |} jj j|< |d kr2 jjd n|dkrJ jjd n|dkrb jjd nt|trl|dk ry jjtjdd| W n tj k
r t
|d Y nX n|d'k rԈ jjtjd d | n|d(k r jjtjd
d| nt|d)k r jjtjdd| nR|d*k r8 jjtjdd| n0|d+k r` jjd|j d ddd nt
|nxt|t r jjtjdd| nRt|t
j
r|t
j
dd d j } jjtjdd| nt|tr jdt|j jj|j nt|ttfr0 jdt| jj| nt|try |jd } jd t| W n4 tk
r |jd } jd t|d Y nX jj| nDt|ttfr fd!d"|D }t|} jd#| jjtjd$ j| f| nt|trg g }} j r&t |j } n|j } xR| D ]J\}
} t|
ts\ j rTq4t d%|j! j |
|j! j | q4W t|} jd&| jjtjd$ j| f| jjtjd$ j| f| nt |d S ),N F Tr r z>Bqr' r r z>BBr z>BHr% r z>BLr& r z>BQr r )r z>Bdr i r r r zutf-16ber r c s g | ]} j |qS r )r )r r )r r r
s z4_BinaryPlistWriter._write_object..r rp zkeys must be stringsr r i l l l )"r r r r r7 r+ rd r r r r to_bytesr re Z
total_secondsr r( rR r> rA r r, rZ UnicodeEncodeErrorr r r r r r r r rB rS ) r r r r@ tZrefsrV ZkeyRefsZvalRefsZ rootItemsr
r$ r )r r r s
$
"$z _BinaryPlistWriter._write_objectN) r" r# r$ r' r7 r r r( r r r r r r s -0r c C s | d d dkS )Nr s bplist00r )r r r r _is_fmt_binary s r0 )detectr~ writerT)r9 r: r; c C sn |d krL| j d}| jd x:tj D ]}|d |r&|d }P q&W t n t| d }|||d}|j| S )Nr r r1 r~ )r: r; )r r _FORMATSr# r r )r0 r9 r: r; r infoPpr r r r s
c C s t | }t||||dS )N)r9 r: r; )r r )r r9 r: r; r0 r r r r
s F)r9 r< r= c C s: |t krtd|f t | d |||d}|j| d S )NzUnsupported format: %rr2 )r< r= )r3 rs r7 )r r0 r9 r< r= r2 r r r r s )r9 r= r< c C s t }t| ||||d |j S )N)r9 r= r< )r r r? )r r9 r= r< r0 r r r r s )rE )D__all__rT r
contextlibre enumior r r rer warningsr Zxml.parsers.expatr Enumr" r globalsr3 __members__r r r contextmanagerr1 r r r r r r r compilerq rF rC ASCIIra ri rk rw rx r r r rs r r objectr r r r, rd r rA r r r0 r r
r3 r r
r r r r r r 1 sx
0
h&d! #
S