3
�L�j#~ � @ 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� 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.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�
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\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 )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 ) Nzz>)�_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)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 |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 )Nzz)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� 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 � � 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 || 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..�� )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 �
� 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� �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 �1 sx
0
h&d! #
S