3
�Qg#~ � @ s^ d ddddddddd d
d d d
gZ ddlZddlZddlZddlZddlZddlmZ ddlZddl Z ddl
Z
ddl Z ddl m
Z
ddlmZ ejdded�Ze� jej� G dd� de�ZG dd� de�Zejdd� �ZG dd� de�Zd d � Z d d� Z d d� Z d d� Z G d d� d�Z 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� d0�Z+G d1d2� d2�Z,G d3d4� d4e,�Z-d5d6� Z.G d7d� de/�Z0d8d9d:d;d<�Z1e2� Z3G d=d>� d>�Z4d?d@� Z5e6e7e8eje9fZ:G dAdB� dBe2�Z;dCdD� Z<e=ee.e+e-dE�e>ee<e4e;dE�iZ?ddFedG�dHd
�Z@ddFedG�dId �ZAe=dFdJdK�dLd �ZBe=dJdFdM�dNd
�ZCdS )P� readPlist�
writePlist�readPlistFromBytes�writePlistToBytes�Plist�Data�Dict�InvalidFileException�FMT_XML�
FMT_BINARY�load�dump�loads�dumps� N)�BytesIO)�warn)� ParserCreate� PlistFormatzFMT_XML FMT_BINARY)�modulec @ s( e Zd Zf Zdd� Zdd� Zdd� ZdS )�
_InternalDictc C s: y | | }W n t k
r( t|��Y nX tdtd� |S )NzLAttribute access from plist dicts is deprecated, use d[key] notation instead� )�KeyError�AttributeErrorr �DeprecationWarning)�self�attr�value� r � /usr/lib64/python3.6/plistlib.py� __getattr__V s z_InternalDict.__getattr__c C s t dtd� || |< d S )NzLAttribute access from plist dicts is deprecated, use d[key] notation insteadr )r r )r r r r r r � __setattr___ s z_InternalDict.__setattr__c C s8 y
| |= W n t k
r& t|��Y nX tdtd� d S )NzLAttribute access from plist dicts is deprecated, use d[key] notation insteadr )r r r r )r r r r r � __delattr__d s
z_InternalDict.__delattr__N)�__name__�
__module__� __qualname__� __slots__r r r! r r r r r O s r c s e Zd Z� fdd�Z� ZS )r c s t dtd� t� jf |� d S )Nz?The plistlib.Dict class is deprecated, use builtin dict insteadr )r r �super�__init__)r �kwargs)� __class__r r r' o s z
Dict.__init__)r" r# r$ r' �
__classcell__r r )r) r r m s c c s2 t | t�r(t| |�� }|V W d Q R X n| V d S )N)�
isinstance�str�open)�
pathOrFile�mode�fpr r r � _maybe_openu s
r1 c s0 e Zd Z� fdd�Zedd� �Zdd� Z� ZS )r c s t dtd� t� jf |� d S )NzJThe Plist class is deprecated, use the load() and dump() functions insteadr )r r r& r' )r r( )r) r r r' � s zPlist.__init__c C s2 t |d��}t|�}W d Q R X | � }|j|� |S )N�rb)r1 r �update)�clsr. r0 r Zplistr r r �fromFile� s
zPlist.fromFilec
C s$ t |d��}t| |� W d Q R X d S )N�wb)r1 r )r r. r0 r r r �write� s z Plist.write)r" r# r$ r' � classmethodr5 r7 r* r r )r) r r s c
C s2 t dtd� t| d��}t|d dtd�S Q R X d S )Nz8The readPlist function is deprecated, use load() insteadr r2 F)�fmt�use_builtin_types� dict_type)r r r1 r r )r. r0 r r r r � 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 r6 TF)r9 � sort_keys�skipkeys)r r r1 r r )r r. r0 r r r r � s c C s t dtd� tt| �d dtd�S )NzBThe readPlistFromBytes function is deprecated, use loads() insteadr F)r9 r: r; )r r r r r )�datar r r r � s c C s, t dtd� t� }t| |tddd� |j� S )NzAThe writePlistToBytes function is deprecated, use dumps() insteadr TF)r9 r< r= )r r r r r �getvalue)r �fr r r r � s
c @ s: e Zd Zdd� Zedd� �Zd
dd�Zdd � Zd
d � Zd S )r c C s t |t�std��|| _d S )Nzdata must be as bytes)r+ �bytes� TypeErrorr> )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.fromBase64�L c C s t | j|�S )N)�_encode_base64r> )r �
maxlinelengthr r r �asBase64� s z
Data.asBase64c C s4 t || j�r| j|jkS t |t�r,| 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| j�f 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� <?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
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 � )�range�len�append�binasciiZ
b2a_base64�join)�srG Z
maxbinsize�pieces�i�chunkr r r rF s rF c C s( t | t�rtj| jd��S tj| �S d S )Nzutf-8)r+ r, rT Z
a2b_base64�encode)rV r r r rC
s
rC z{(?P<year>\d\d\d\d)(?:-(?P<month>\d\d)(?:-(?P<day>\d\d)(?:T(?P<hour>\d\d)(?::(?P<minute>\d\d)(?::(?P<second>\d\d))?)?)?)?)?Zc C sN d}t j| �j� }g }x,|D ]$}|| }|d kr2P |jt|�� q W tj|� S )N�year�month�day�hour�minute�second)r[ r\ r] r^ r_ r` )� _dateParser�match� groupdictrS �int�datetime)rV �orderZgdZlst�key�valr 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 ) Nz<strings can't contains control characters; use bytes insteadz
�
�
�&z&�<z<�>z>)�_controlCharPat�search�
ValueError�replace)�text�mr 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_key�root�_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 �parser�handle_begin_elementZStartElementHandler�handle_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 �element�attrs�handlerr 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 %d���r� 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 Zddd�Zdd� Zdd� Zdd
d �Zd d
� Zd S )�_DumbXMLWriterr � c C s || _ g | _|| _|| _d S )N)�filery �
_indent_level�indent)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</%s>)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>%s</%s>z<%s/>)rw r� )r r� r r r r �simple_element� s z _DumbXMLWriter.simple_elementc C sH |r8t |t�r|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 Zddd�Zdd � 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 )Nz<plist version="1.0">z</plist>)r� � write_value)r r r r r r7 � s
z_PlistWriter.writec C s4 t |t�r| jd|� �n|dkr0| jd� �n |dkrD| jd� n�t |t�r�d| ko`dk n rx| jd d
| � nt|��n�t |t�r�| jd t|�� n�t |t�r�| j|� n|t |t �r�| j
|� nft |t t f�r�| j
|� nLt |tj��r| jd t|�� n,t |ttf��r | j|� ntd
t|� ��d S )N�stringT�trueFZfalser� �? �@ Zintegerz%d�realZdatezunsupported type: %sr� l���� l )r+ r, r� rd �
OverflowErrorr� rL �dict�
write_dictr �
write_datarA � bytearray� write_bytesre rk �tuple�list� 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||�jd�D ]}|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|t�sR| jrJq0td��| jd|� | j |� q0W | j
d� n
| jd� d S )Nr� zkeys must be stringsrg ) r� r� �sorted�itemsr+ 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 )N�array)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}||jd�j|� }| d t|�� |krXdS qXW q>W dS )
N� <?xml� <plistTzutf-8z utf-16-bez utf-16-le�asciiF)r� r� )�
startswith�codecs�BOM_UTF8� BOM_UTF16_BE� BOM_UTF16_LE�decoderZ rR )�header�prefixesZpfxZbom�encoding�start�prefixr r r � _is_fmt_xml* s
r� c @ s e Zd Zddd�ZdS )r � Invalid filec C s t j| |� d S )N)rs r' )r �messager r r r' L s z InvalidFileException.__init__N)r� )r" r# r$ r' r r r r r K s �B�H�L�Q)r� r rN r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�_BinaryPlistParserc C s || _ || _d S )N)r| r} )r r: r; r r r r' [ s z_BinaryPlistParser.__init__c C s� y||| _ | j jdtj� | j jd�}t|�dkr6t� �tjd|�\}| _ }}}| j j|� | j
||�| _ t g| | _
| j|�S tttjttfk
r� t� �Y nX d S )N� z>6xBBQQQi��)�_fp�seek�os�SEEK_END�readrR r �struct�unpack� _ref_size�
_read_ints�_object_offsets�
_undefined�_objects� _read_object�OSError�
IndexError�errorr� rs )r r0 �trailer� offset_size� num_objects�
top_object�offset_table_offsetr r r r� _ s
z_BinaryPlistParser.parsec C sL |dkrH| j jd�d 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� �fdd�td�| ��D ��S d S )Nrp c 3 s& | ] }t j� ||� � d �V qdS )�bigN)rd �
from_bytes)�.0rX )r> �sizer r � <genexpr>� s z0_BinaryPlistParser._read_ints.<locals>.<genexpr>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 || j�S )N)r� r� )r r� r r r �
_read_refs� s z _BinaryPlistParser._read_refsc s� � j | }|tk r|S � j| }� jj|� � jjd�d }|d@ |d@ }}|dkr^d }�n�|dkrnd}�n||dkr~d}�nl|dkr�d }�n\|d
kr�tj� jjd|> �d |d kd
�}�n0|dkr�tj d� jjd��d }�n |dk�rtj d� jjd��d }�n�|dk�rDtj d� jjd��d }t
j
ddd�t
j |d� }�n�|dk�r�� j |�}� jj|�}t
|�|k�rxt� �� j�s�t|�}�n^|dk�rΈ j |�}� jj|�} t
| �|k�r�t� �| jd�}�n |dk�r� j |�d }� jj|�} t
| �|k�rt� �| jd�}n�|d k�rX� j |�}� j|�}
g }|� j |< |j� fd d �|
D �� n�|d k�r� j |�}� j|�} � j|�}
� j� }|� j |< y2x,t| |
�D ] \} }
� j|
�|� j| �< �q�W W n tk
�r� t� �Y nX nt� �|� j |< |S ) Nr� r �� r� r� F� TrP r� r� rO )�signed�" z>frN �# z>d�3 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.<locals>.<genexpr>�� )r� r� r� r� r� r� rd r� r� r� re Z timedeltar� rR r r| r r� r� �extendr} �zipr� rB )r �ref�result�offset�tokenZtokenHr� r@ rV r> Zobj_refsZkey_refs�k�or )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�
_getrefnum�tellr� �pack)
r r r� �objr� r� r� Z
offset_formatZ sort_versionr� r r r r7
s(
z_BinaryPlistWriter.writec C s~ t |t�r"t|�|f| jkrZd S n8t |t�rHt|j�|jf| jkrZd S nt|�| jkrZd S t| j �}| j j
|� t |t�r�|| jt|�|f< n0t |t�r�|| jt|j�|jf< n|| jt|�< t |t ��rPg }g }|j � }| j
r�t|�}xB|D ]:\}}t |t��s| j�r q�td��|j
|� |j
|� q�W xLtj||�D ]}| j|� �q:W n*t |ttf��rzx|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 � itertools�chainr r� r� ) r r Zrefnum�keys�valuesr� r
�vr r r r r 7 sB
z_BinaryPlistWriter._flattenc C sN t |t�r | jt|�|f S t |t�r<| 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� �n�t|t��rl|dk r�y� 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|t��r � jdt|j�� � jj|j� �n�t|ttf��r0� jdt|�� � jj|� �n�t|t��r�y |jd �}� jd t|�� W n4 tk
�r� |jd �}� jd t|�d � Y nX � jj|� �nDt|ttf��r�� fd!d"�|D �}t|�}� jd#|� � jjtjd$� j| f|�� � n�t|t��r�g g }}� j �r&t |j � �} n|j � } xR| D ]J\}
} t|
t��s\� j �rT�q4t 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 �
<listcomp>� s z4_BinaryPlistWriter._write_object.<locals>.<listcomp>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� �info�P�pr r r r � s
c C s t | �}t||||d�S )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 �enum�ior 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 �<module>1 sx
0
h&d! #
S