B
kjg @ s* d ddddddddd d
d g Z d d
lZd d
lZd d
lZd d
lZd d
lZd dlmZ d d
lZd d
l Z d d
l
Z
d d
l Z d dl m
Z
d dlmZ ejddedZe ej ejdd Zdd Zdd Zdd Zdd ZG dd dZ dZ e
d Z dId d Z d d! Z!e
d"e
j"Z#d#d$ Z$d%d& Z%d'd( Z&G d)d* d*Z'G d+d, d,Z(G d-d. d.e(Z)d/d0 Z*G d1d de+Z,d2d3d4d5d6Z-e. Z/G d7d8 d8Z0d9d: Z1e2e3e4eje5fZ6G d;d< d Z8e9e:e*e'e)d?e;e:e8e0e7d?iZe9d@dDdEdFd Z?e9dDd@dGdHd Z@d
S )J readPlist
writePlistreadPlistFromByteswritePlistToBytesDataInvalidFileExceptionFMT_XML
FMT_BINARYloaddumploadsdumps N)BytesIO)warn) ParserCreate PlistFormatzFMT_XML FMT_BINARY)modulec c s2 t | tr(t| | }|V W d Q R X n| V d S )N)
isinstancestropen)
pathOrFilemodefp r -/opt/alt/python37/lib64/python3.7/plistlib.py _maybe_openO s
r c C s0 t dtd t| d}t|d ddS Q R X d S )Nz8The readPlist function is deprecated, use load() instead rbF)fmtuse_builtin_types)r DeprecationWarningr r )r r r r r r Y s c C s8 t dtd t|d}t| |tddd W d Q R X d S )Nz9The writePlist function is deprecated, use dump() insteadr wbTF)r sort_keysskipkeys)r r r r
r )valuer r r r r r f s c C s t dtd tt| d ddS )NzBThe readPlistFromBytes function is deprecated, use loads() insteadr F)r r )r r r r )datar r r r s s c C s, t dtd t }t| |tddd | S )NzAThe writePlistToBytes function is deprecated, use dumps() insteadr TF)r r" r# )r r r r
r getvalue)r$ fr r r r ~ s
c @ s: e Zd Zdd Zedd Zd
ddZdd Zd
d Zd S )r c C s t |tstd|| _d S )Nzdata must be as bytes)r bytes TypeErrorr% )selfr% r r r __init__ s
z
Data.__init__c C s | t |S )N)_decode_base64)clsr% 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 __class__r% r( NotImplemented)r* otherr r r __eq__ s
z Data.__eq__c C s d| j jt| jf S )Nz%s(%s))r3 __name__reprr% )r* r r r __repr__ s z
Data.__repr__N)r/ ) r7
__module__ __qualname__r+ classmethodr. r2 r6 r9 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]r/ c C sT |d d }g }x8t dt| |D ]$}| ||| }|t| q"W d|S )N r
)rangelenappendbinasciiZ
b2a_base64join)sr1 Z
maxbinsizepiecesichunkr r r r0 s r0 c C s( t | trt| dS t| S d S )Nzutf-8)r r rC Z
a2b_base64encode)rE r r r r, s
r, 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 | }g }x,|D ]$}|| }|d kr2P |t| q W tj| S )N)yearmonthdayhourminutesecond) _dateParsermatch groupdictrB intdatetime)rE ordergdZlstkeyvalr r r _date_from_string s
rY c C s d| j | j| j| j| j| jf S )Nz %04d-%02d-%02dT%02d:%02d:%02dZ)rJ rK rL rM rN rO )dr r r _date_to_string s r[ c C sZ t | }|d k rtd| dd} | dd} | dd} | dd} | d d
} | S ) Nzz>)_controlCharPatsearch
ValueErrorreplace)textmr r r _escape s
rg 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'd( Zd)S )* _PlistParserc C s" g | _ d | _d | _|| _|| _d S )N)stack current_keyroot_use_builtin_types
_dict_type)r* r dict_typer r r r+ s
z_PlistParser.__init__c C sB t | _| j| j_| j| j_| j| j_| j| j_ | j
| | j S )N) r parserhandle_begin_elementZStartElementHandlerhandle_end_elementZEndElementHandler handle_dataZCharacterDataHandlerhandle_entity_declZEntityDeclHandlerZ ParseFilerk )r* Zfileobjr r r parse s
z_PlistParser.parsec C s t dd S )Nz8XML entity declarations are not supported in plist files)r )r* Z entity_nameZis_parameter_entityr$ baseZ system_idZ public_idZ
notation_namer r r rs s z _PlistParser.handle_entity_declc C s* g | _ t| d| d }|d k r&|| d S )NZbegin_)r% getattr)r* elementattrshandlerr r r rp s z!_PlistParser.handle_begin_elementc C s" t | d| d }|d k r | d S )NZend_)rv )r* rw ry r r r rq s z _PlistParser.handle_end_elementc C s | j | d S )N)r% rB )r* r% r r r rr 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 | d S )Nz unexpected element at line %d) rj r ri typerc ro CurrentLineNumberrk rB )r* r$ r r r
add_object s
z_PlistParser.add_objectc C s d | j}g | _|S )N )rD r% )r* r% r r r get_data, s z_PlistParser.get_datac C s" | }| | | j| d S )N)rm r} ri rB )r* rx rZ r r r
begin_dict3 s
z_PlistParser.begin_dictc C s* | j r td| j | jjf | j d S )Nz%missing value for key '%s' at line %d)rj rc ro r| ri pop)r* r r r end_dict8 s z_PlistParser.end_dictc C s8 | j st| jd ti s*td| jj | | _ d S )Nrz zunexpected key at line %d)rj r ri r{ rc ro r| r )r* r r r end_key> s z_PlistParser.end_keyc C s g }| | | j| d S )N)r} ri rB )r* rx ar r r begin_arrayD s
z_PlistParser.begin_arrayc C s | j d S )N)ri r )r* r r r end_arrayI s z_PlistParser.end_arrayc C s | d d S )NT)r} )r* r r r end_trueL s z_PlistParser.end_truec C s | d d S )NF)r} )r* r r r end_falseO s z_PlistParser.end_falsec C s | t| d S )N)r} rS r )r* r r r end_integerR s z_PlistParser.end_integerc C s | t| d S )N)r} floatr )r* r r r end_realU s z_PlistParser.end_realc C s | | d S )N)r} r )r* r r r
end_stringX s z_PlistParser.end_stringc C s2 | j r| t| n| t| d S )N)rl r} r, r r r. )r* r r r end_data[ s z_PlistParser.end_datac C s | t| d S )N)r} rY r )r* r r r end_dateb s z_PlistParser.end_dateN)r7 r: r; r+ rt rs rp rq rr r} r r r r r r r r r r r r r r r r r rh s( rh c @ s8 e Zd ZdddZdd Zdd Zdd
d Zd d
Zd S )_DumbXMLWriterr
c C s || _ g | _|| _|| _d S )N)fileri
_indent_levelindent)r* r indent_levelr r r r r+ g s z_DumbXMLWriter.__init__c C s, | j | | d| | jd7 _d S )Nz<%s> )ri rB writelnr )r* rw r r r
begin_elementm s z _DumbXMLWriter.begin_elementc C s | j d8 _ | d| d S )Nr z)r r )r* rw r r r end_elementr s z_DumbXMLWriter.end_elementNc C s8 |d k r&t |}| d|||f n| d| d S )Nz <%s>%sz<%s/>)rg r )r* rw r$ r r r simple_elementx s z _DumbXMLWriter.simple_elementc C sH |r8t |tr|d}| j| j| j | j| | jd d S )Nzutf-8
)r r rI r writer r )r* liner r r r s
z_DumbXMLWriter.writeln)r
r )N)r7 r: r; r+ r r r r r r r r r f s
r c @ sF e Zd ZdddZdd Zd
d Zd d
Zdd Zdd Zdd Z dS ) _PlistWriterr
r TFc C s. |r| t t| ||| || _|| _d S )N)r PLISTHEADERr r+
_sort_keys _skipkeys)r* r r r Z writeHeaderr" r# r r r r+ s
z_PlistWriter.__init__c C s" | d | | | d d S )Nzz)r write_value)r* r$ r r r r s
z_PlistWriter.writec C s4 t |tr| d| n|dkr0| d n |dkrD| d nt |trd| krbdk rxn n| dd | nt|nt |tr| d
t| nt |tr| | n|t |t r|
| nft |t t fr|
| nLt |tjr| d t| n,t |ttfr | | ntd t| d S )
NstringTtrueFZfalsel l Zintegerz%drealZdatezunsupported type: %s)r r r rS
OverflowErrorr r8 dict
write_dictr
write_datar( bytearray write_bytesrT r[ tuplelist write_arrayr) r{ )r* r$ r r r r s.
z_PlistWriter.write_valuec C s | |j d S )N)r r% )r* r% r r r r s z_PlistWriter.write_datac C s~ | d | jd8 _tddt| jdd| j }x&t||dD ]}|rL| | qLW | jd7 _| d d S )Nr% r r/ r s r )
r r maxrA r rd r0 splitr r )r* r% r1 r r r r r s
z_PlistWriter.write_bytesc C s |rx| d | jr"t| }n| }x@|D ]8\}}t|tsR| jrJq0td| d| | | q0W |
d n
| d d S )Nr zkeys must be stringsrW ) r r sorteditemsr r r r) r r r )r* rZ r rW r$ r r r r s
z_PlistWriter.write_dictc C s@ |r2| d x|D ]}| | qW | d n
| d d S )Narray)r r r r )r* r r$ r r r r s
z_PlistWriter.write_arrayN)r
r r TF)
r7 r: r; r+ r r r r r r r r r r r s
%
r c C s d}x|D ]}| |r
dS q
W xntjdftjdftjdffD ]N\}}| |sRq>x8|D ]0}||d| }| d t| |krXdS qXW q>W dS )N)s 6xBBQQQ)_fpseekosSEEK_ENDreadrA r structunpack _ref_size
_read_ints_object_offsets
_undefined_objects _read_objectOSError
IndexErrorerrorr rc )r* r trailer offset_size num_objects
top_objectoffset_table_offsetr r r rt % s
z_BinaryPlistParser.parsec C sL |dkrH| j dd d@ }d|> }dt| }t|| j |d S |S )N r r
r> r` )r r _BINARY_FORMATr r )r* tokenLrf rE r' r r r _get_size> s z _BinaryPlistParser._get_sizec st | j | tkr2td| t S rFt | krLt t fddtd| D S d S )Nr` c 3 s& | ] }t || d V qdS )bigN)rS
from_bytes).0rG )r% sizer r O s z0_BinaryPlistParser._read_ints..r
) r r r r r rA r r r@ )r* nr r )r% r r r H s z _BinaryPlistParser._read_intsc C s | || jS )N)r r )r* r r r r
_read_refsR s z _BinaryPlistParser._read_refsc s j | }|tk r|S j| } j| jdd }|d@ |d@ }}|dkr^d }n|dkrnd}n||dkr~d}nl|dkrd }n\|d
krtj jd|> d |d kd
}n0|dkrt d jdd }n |dkrt d jdd }n|dkrDt d jdd }t
dddt
j |d }n|dkr |} j|}t
||krxt jst|}n^|dkrΈ |} j|} t
| |krt | d}n |dkr |d } j|} t
| |krt | d}n|d krX |} |}
g }| j |< | fd d |
D n|d kr |} |} |}
}| j |< y2x,t| |
D ] \} }
|
| | < qW W n tk
r t Y nX nt | j |< |S ) Nr r
r r F Tr? r r r> )signed" z>fr= # z>d3 i )Zseconds@ P r ` r zutf-16be c 3 s | ]} |V qd S )N)r )r x)r* r r r s z2_BinaryPlistParser._read_object.. )r r r r r r rS r r r rT Z timedeltar rA r rl r r r extendrm zipr r) )r* refresultoffsettokenZtokenHr r' rE r% Zobj_refsZkey_refskor )r* r r U s
z _BinaryPlistParser._read_objectN) r7 r: r; r+ rt r r r r r r r r r s
r c C s0 | dk r dS | dk rdS | d> d> r(dS dS d S )N r i r r r= r 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* r r" r# r r r r+ s z_BinaryPlistWriter.__init__c
C s g | _ i | _i | _| | t| j }dg| | _t|| _t| j | _ | j
d x| j D ]}| | q\W |
|}| j
}t|}dt| | }| j
tj|f| j d}||| j|||f} | j
tjd| d S )Nr
s bplist00r` >5xBBBQQQ)r )_objlist _objtable _objidtable_flattenrA r r r r _ref_formatr r
_write_object
_getrefnumtellr pack)
r* r$ r objr r r Z
offset_formatZ sort_versionr r r r r s(
z_BinaryPlistWriter.writec C s~ t |tr"t||f| jkrZd S n8t |trHt|j|jf| jkrZd S nt|| jkrZd S t| j }| j
| t |tr|| jt||f< n0t |tr|| jt|j|jf< n|| jt|< t |t rPg }g }| }| j
rt|}xB|D ]:\}}t |ts| jr qtd|
| |
| qW xLt||D ]}| | q:W n*t |ttfrzx|D ]}| | qfW d S )Nzkeys must be strings)r _scalarsr{ r r r% idr rA r rB r r r r r r r) itertoolschainr r r ) r* r$ Zrefnumkeysvaluesr r vr r r r r 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
- s
z _BinaryPlistWriter._getrefnumc C s |dk r"| j td||B n|dk rH| j td|dB d| nh|dk rn| j td|dB d| nB|d k r| j td
|dB d | n | j td |dB d
| d S )Nr z>Br z>BBBr i z>BBH l z>BBL z>BBQ )r r r r )r* r r r r r _write_size5 s z _BinaryPlistWriter._write_sizec s |} j j|< |d kr2 jd n|dkrJ jd n|dkrb jd nt|trl|dk ry jtdd| W n tj k
r t
|d Y nX n|d k rԈ jtd
d | n|d k r jtd
d| nt|dk r jtdd| nR|dk r8 jtdd| n0|dk r` jd|j d ddd nt
|nxt|t r jtdd| nRt|t
j
r|t
ddd } jtdd | nt|tr d t|j j|j nt|ttfr0 d t| j| nt|try |d } d t| W n4 tk
r |d } d!t|d" Y nX j| nDt|ttfr fd#d$|D }t|} d%| jtjd& j| f| nt|trg g }} j r&t | } n| } xR| D ]J\}
} t|
ts\ j rTq4t d'|! |
|! | q4W t|} d(| jtjd& j| f| jtjd& j| f| nt |d S ))N F Tr r
z>Bqr r z>BBr i z>BHr l z>BLr l z>BQl r )r z>Bdr i r r r r r zutf-16ber r c s g | ]} |qS r )r
)r r )r* r r
~ s z4_BinaryPlistWriter._write_object..r r` zkeys must be stringsr )"r
r r r r r rS r r r r to_bytesr rT Z
total_secondsr r rA r% r( r r rI UnicodeEncodeErrorr r r r r r r r r) rB ) r* r$ r r' tZrefsrE ZkeyRefsZvalRefsZ rootItemsr r r )r* r r E s
$
"$z _BinaryPlistWriter._write_objectN) r7 r: r; r+ r 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 r )detectro writerT)r r rn c C sn |d krL| d}| d x:t D ]}|d |r&|d }P q&W t n t| d }|||d}|| S )Nr r
r! ro )r rn )r r _FORMATSr r rt )r r r rn r infoPpr r r r s
c C s t | }t||||dS )N)r r rn )r r )r$ r r rn r r r r r s F)r r" r# c C s: |t krtd|f t | d |||d}|| d S )NzUnsupported format: %rr" )r" r# )r# rc r )r$ r r r" r# r" r r r r
s )r r# r" c C s t }t| ||||d | S )N)r r# r" )r r
r& )r$ r r# r" r r r r r s )r/ )A__all__rC r
contextlibrT enumior r r rer warningsr Zxml.parsers.expatr Enumr7 r globalsupdate __members__contextmanagerr r r r r r r compilera r0 r, ASCIIrP rY r[ rg rh r r r rc r r objectr r r r rS r r( r r r r r r r# r r r
r r r r r 1 sr
0
o&d! #
S