3
d?�Z d@dA� Z!dBdC� Z"dDdE� Z#dFdG� Z$dHdI� Z%dJdK� Z&dLdM� Z'dNdO� Z(dPdQ� Z)dRdS� Z*dTdU� Z+dVdW� Z,dXdY� Z-dZd[� Z.d\d]� Z/d^d_� Z0d`da� Z1dbdc� Z2ddde� Z3d�dfdg�Z4dhdi� Z5d�djdk�Z6dldm� Z7dndo� Z8d�dqdr�Z9dsdtdudvdwdxdydzd{d|d}d~dd�d�d��Z:d�d�d��Z;d�d�� Zd�d�d��Z?dS )�r �!z!!)r ztag:yaml.org,2002:Nc C s� || _ d | _g | _| j| _g | _d | _g | _d | _d| _ d| _
d| _ d| _ d| _
d| _d| _d| _d| _d| _|| _|| _d| _|r�d| k o�dk n r�|| _d| _|r�|| jd kr�|| _d| _|d kr�|| _d | _d | _d | _d | _d | _ d S ) N� FT� r �
�P �
�
�
)r r r ) �stream�encoding�states�expect_stream_start�state�events�event�indents�indent�
flow_level� root_context�sequence_context�mapping_context�simple_key_context�line�column�
whitespace� indention�
open_ended� canonical�
allow_unicode� best_indent�
best_width�best_line_break� tag_prefixes�prepared_anchor� prepared_tag�analysis�style)r r r1 r& �widthr2 Z
line_breakr r r
r & sD zEmitter.__init__c C s g | _ d | _d S )N)r r" )r r r r
�disposej s zEmitter.disposec C s: | j j|� x(| j� s4| j jd�| _| j� d | _qW d S )Nr )r# �append�need_more_events�popr$ r" )r r$ r r r
�emito s
z Emitter.emitc C sX | j s
dS | j d }t|t�r(| jd�S t|t�r<| jd�S t|t�rP| jd�S dS d S )NTr r r � F)r# �
isinstance�DocumentStartEvent� need_events�SequenceStartEvent�MappingStartEvent)r r$ r r r
r> x s
zEmitter.need_more_eventsc C sx d}x`| j dd � D ]N}t|ttf�r0|d7 }n&t|ttf�rH|d8 }nt|t�rVd}|dk rdS qW t| j �|d k S )Nr r F�����)r# rB rC �CollectionStartEvent�DocumentEndEventZCollectionEndEvent�StreamEndEvent�len)r �count�levelr$ r r r
rD � s
zEmitter.need_eventsFc C sF | j j| j� | jd kr.|r&| j| _qBd| _n|sB| j| j7 _d S )Nr )r% r= r& r3 )r �flow�
indentlessr r r
�increase_indent� s
zEmitter.increase_indentc C sP t | jt�r>| jjr,t| jd� r,| jj| _| j� | j| _nt d| j ��d S )Nr z%expected StreamStartEvent, but got %s)
rB r$ ZStreamStartEventr �hasattrr �write_stream_start�expect_first_document_startr" r )r r r r
r! � s
zEmitter.expect_stream_startc C s t d| j ��d S )Nz expected nothing, but got %s)r r$ )r r r r
�expect_nothing� s zEmitter.expect_nothingc C s | j dd�S )NT)�first)�expect_document_start)r r r r
rS � s z#Emitter.expect_first_document_startc C sv t | jt��r(| jjs | jjr8| jr8| jdd� | j� | jjrX| j| jj�}| j |� | j
j � | _ | jjr�t
| jjj� �}xB|D ]:}| jj| }|| j |< | j|�}| j|�}| j||� q�W |o�| jj o�| j o�| jj o�| jj o�| j� }|�s | j� | jdd� | j�r | j� | j| _nJt | jt��rd| j�rR| jdd� | j� | j� | j| _ntd| j ��d S )Nz...Tz---z'expected DocumentStartEvent, but got %s)rB r$ rC �versionZtagsr0 �write_indicator� write_indent�prepare_version�write_version_directive�DEFAULT_TAG_PREFIXES�copyr6 �sorted�keys�prepare_tag_handle�prepare_tag_prefix�write_tag_directive�explicitr1 �check_empty_document�expect_document_rootr" rJ �write_stream_endrT r ) r rU � version_textZhandles�handle�prefix� handle_text� prefix_text�implicitr r r
rV � sB
z Emitter.expect_document_startc C sT t | jt�rB| j� | jjr0| jdd� | j� | j� | j| _nt d| j ��d S )Nz...Tz%expected DocumentEndEvent, but got %s)
rB r$ rI rY rc rX � flush_streamrV r" r )r r r r
�expect_document_end� s
zEmitter.expect_document_endc C s | j j| j� | jdd� d S )NT)�root)r r= rn � expect_node)r r r r
re � s z Emitter.expect_document_rootc C s� || _ || _|| _|| _t| jt�r.| j� n�t| jtt f�r�| j
d� | j � t| jt�rf| j � q�t| jt
�r�| js�| js�| jjs�| j� r�| j� q�| j� q�t| jt�r�| js�| js�| jjs�| j� r�| j� q�| j� ntd| j ��d S )N�&z expected NodeEvent, but got %s)r( r) r* r+ rB r$ �
AliasEvent� expect_alias� ScalarEventrH �process_anchor� process_tag�
expect_scalarrE r' r1 Z
flow_style�check_empty_sequence�expect_flow_sequence�expect_block_sequencerF �check_empty_mapping�expect_flow_mapping�expect_block_mappingr )r ro �sequence�mapping�
simple_keyr r r
rp � s,
zEmitter.expect_nodec C s. | j jd krtd��| jd� | jj� | _d S )Nz!anchor is not specified for aliasr )r$ �anchorr ru r r? r" )r r r r
rs s
zEmitter.expect_aliasc C s0 | j dd� | j� | jj� | _| jj� | _d S )NT)rN )rP �process_scalarr% r? r& r r" )r r r r
rw
s zEmitter.expect_scalarc C s6 | j dddd� | jd7 _| jdd� | j| _d S )N�[T)r. r )rN )rX r' rP � expect_first_flow_sequence_itemr" )r r r r
ry s z Emitter.expect_flow_sequencec C sx t | jt�r@| jj� | _| jd8 _| jdd� | jj� | _ n4| j
sR| j | j krZ| j
� | jj| j� | jdd� d S )Nr �]FT)r~ )rB r$ �SequenceEndEventr% r? r& r' rX r r" r1 r- r4 rY r= �expect_flow_sequence_itemrp )r r r r
r� s z'Emitter.expect_first_flow_sequence_itemc C s� t | jt�rZ| jj� | _| jd8 _| jr@| jdd� | j � | jdd� | j
j� | _ n@| jdd� | jsx| j | j
kr�| j � | j
j| j� | jdd� d S )Nr �,Fr� T)r~ )rB r$ r� r% r? r& r' r1 rX rY r r" r- r4 r= r� rp )r r r r
r� $ s z!Emitter.expect_flow_sequence_itemc C s6 | j dddd� | jd7 _| jdd� | j| _d S )N�{T)r. r )rN )rX r' rP � expect_first_flow_mapping_keyr" )r r r r
r| 6 s zEmitter.expect_flow_mappingc C s� t | jt�r@| jj� | _| jd8 _| jdd� | jj� | _ nn| j
sR| j | j krZ| j
� | j
r�| j� r�| jj| j� | jddd� n&| jdd� | jj| j� | jdd� d S )Nr �}FT)r r� �?)r )rB r$ �MappingEndEventr% r? r& r' rX r r" r1 r- r4 rY �check_simple_keyr= � expect_flow_mapping_simple_valuerp �expect_flow_mapping_value)r r r r
r� < s z%Emitter.expect_first_flow_mapping_keyc C s� t | jt�rZ| jj� | _| jd8 _| jr@| jdd� | j � | jdd� | j
j� | _ nz| jdd� | jsx| j | j
kr�| j � | j r�| j� r�| j
j| j� | jddd� n&| jdd� | j
j| j� | jdd� d S ) Nr r� Fr� T)r r� r� )r )rB r$ r� r% r? r& r' r1 rX rY r r" r- r4 r� r= r� rp r� )r r r r
�expect_flow_mapping_keyM s" z Emitter.expect_flow_mapping_keyc C s* | j dd� | jj| j� | jdd� d S )N�:FT)r )rX r r= r� rp )r r r r
r� b s z(Emitter.expect_flow_mapping_simple_valuec C sD | j s| j| jkr| j� | jdd� | jj| j� | jdd� d S )Nr� T)r ) r1 r- r4 rY rX r r= r� rp )r r r r
r� g s
z!Emitter.expect_flow_mapping_valuec C s( | j o | j }| jd|d� | j| _d S )NF)rN rO )r* r/ rP � expect_first_block_sequence_itemr" )r rO r r r
rz p s z Emitter.expect_block_sequencec C s | j dd�S )NT)rU )�expect_block_sequence_item)r r r r
r� u s z(Emitter.expect_first_block_sequence_itemc C sb | r,t | jt�r,| jj� | _| jj� | _n2| j� | j dddd� | jj
| j � | j dd� d S )N�-T)r/ )r~ )
rB r$ r� r% r? r& r r" rY rX r= r� rp )r rU r r r
r� x s z"Emitter.expect_block_sequence_itemc C s | j dd� | j| _d S )NF)rN )rP � expect_first_block_mapping_keyr" )r r r r
r} � s z Emitter.expect_block_mappingc C s | j dd�S )NT)rU )�expect_block_mapping_key)r r r r
r� � s z&Emitter.expect_first_block_mapping_keyc C s� | r,t | jt�r,| jj� | _| jj� | _nX| j� | j � rZ| jj
| j � | j ddd� n*| j
dddd� | jj
| j� | j dd� d S )NT)r r� r� )r/ )r )rB r$ r� r% r? r& r r" rY r� r= �!expect_block_mapping_simple_valuerp rX �expect_block_mapping_value)r rU r r r
r� � s z Emitter.expect_block_mapping_keyc C s* | j dd� | jj| j� | jdd� d S )Nr� FT)r )rX r r= r� rp )r r r r
r� � s z)Emitter.expect_block_mapping_simple_valuec C s6 | j � | jdddd� | jj| j� | jdd� d S )Nr� T)r/ )r )rY rX r r= r� rp )r r r r
r� � s z"Emitter.expect_block_mapping_valuec C s" t | jt�o | jo t | jd t�S )Nr )rB r$ rE r# r� )r r r r
rx � s z Emitter.check_empty_sequencec C s" t | jt�o | jo t | jd t�S )Nr )rB r$ rF r# r� )r r r r
r{ � s zEmitter.check_empty_mappingc C sR t | jt� s| j rdS | jd }t |t�oP|jd koP|jd koP|joP|jdkS )NFr � ) rB r$ rC r# rt r� �tagrl �value)r r$ r r r
rd � s
z Emitter.check_empty_documentc C s
d}t | jt�rD| jjd k rD| jd kr6| j| jj�| _|t| j�7 }t | jttf�r�| jj d k r�| j
d krz| j | jj �| _
|t| j
�7 }t | jt�r�| j d kr�| j
| jj�| _ |t| j j�7 }|dk �ot | jt��pt | jt�r�| j j r�| j j �p| j� �p| j� S )Nr � )rB r$ Z NodeEventr� r7 �prepare_anchorrK rt rH r� r8 � prepare_tagr9 �analyze_scalarr� r rr r
r rx r{ )r Zlengthr r r
r� � s$
zEmitter.check_simple_keyc C sR | j jd krd | _d S | jd kr0| j| j j�| _| jrH| j|| j d� d | _d S )NT)r$ r� r7 r� rX )r � indicatorr r r
ru � s
zEmitter.process_anchorc C s� | j j}t| j t�r�| jd kr(| j� | _| j s8|d krn| jdkrN| j jd sd| jdkrn| j jd rnd | _d S | j jd r�|d kr�d}d | _n"| j s�|d kr�| j jr�d | _d S |d kr�t d��| jd kr�| j
|�| _| jr�| j | jd� d | _d S )Nr� r r r ztag is not specifiedT) r$ r� rB rt r: �choose_scalar_styler1 rl r8 r r� rX )r r� r r r
rv � s,
zEmitter.process_tagc C s� | j d kr| j| jj�| _ | jjdks,| jr0dS | jj r�| jjd r�| joZ| j jpZ| j j r�| j
rl| j j s|| j
r�| j j r�dS | jjr�| jjdkr�| j
r�| j r�| j j
r�| jjS | jj s�| jjdkr�| j jr�| jo�| j j r�dS dS )N�"r r� z|>�')r9 r� r$ r� r: r1 rl r+ r
r r' r r r r )r r r r
r� � s&
zEmitter.choose_scalar_stylec C s� | j d kr| j| jj�| _ | jd kr.| j� | _| j }| jdkrR| j| j j|� n`| jdkrn| j | j j|� nD| jdkr�| j
| j j� n*| jdkr�| j | j j� n| j | j j|� d | _ d | _d S )Nr� r� �>�|)
r9 r� r$ r� r: r� r+ �write_double_quotedr �write_single_quoted� write_folded�
write_literal� write_plain)r �splitr r r
r� s
zEmitter.process_scalarc C s, |\}}|dkr t d||f ��d||f S )Nr z unsupported YAML version: %d.%dz%d.%d)r )r rW �major�minorr r r
rZ s zEmitter.prepare_versionc C s� |s t d��|d dks$|d dkr0t d| ��xt|dd� D ]d}d| koTdkn p�d| kold kn p�d
| ko�d kn p�|d ks>t d
||f ��q>W |S )Nz tag handle must not be emptyr r r z*tag handle must start and end with '!': %r�0�9�A�Z�a�zz-_z*invalid character %r in the tag handle: %rrG rG )r )r rh �chr r r
r` ! s HzEmitter.prepare_tag_handlec C s |s t d��g }d }}|d dkr(d}x�|t|�k r�|| }d| koPdkn s�d| kohdkn s�d | ko�d
kn s�|d kr�|d7 }q*||k r�|j|||� � |d }}|jd �}x |D ]}|jd
t|� � q�W q*W ||k �r|j|||� � dj|�S )Nz tag prefix must not be emptyr r r r� r� r� r� r� r� z-;/?!:@&=+$,_.~*'()[]zutf-8z%%%02Xr� )r rK r= �encode�ord�join)r ri �chunks�start�endr� �datar r r
ra - s( H
zEmitter.prepare_tag_prefixc C s� |s t d��|dkr|S d }|}t| jj� �}xH|D ]@}|j|�r4|dksZt|�t|�k r4| j| }|t|�d � }q4W g }d }}x�|t|�k �r^|| } d| ko�dkn p�d| ko�dkn p�d| ko�d kn p�| d
kp�| dko�|dk�r|d 7 }q�||k �r$|j|||� � |d }}| jd �}
x |
D ]} |jd
t| � � �q@W q�W ||k �r||j|||� � dj |�} |�r�d|| f S d| S d S )Nztag must not be emptyr r r� r� r� r� r� r� z-;/?:@&=+$,_.~*'()[]r zutf-8z%%%02Xr� z%s%sz!<%s>)
r r^ r6 r_ �
startswithrK r= r� r� r� ) r r� rh �suffix�prefixesri r� r� r� r� r� Z suffix_textr r r
r� D s@
H
zEmitter.prepare_tagc C s~ |s t d��xl|D ]d}d| ko(dkn pdd| ko@dkn pdd| koXdkn pd|dkst d ||f ��qW |S )
Nzanchor must not be emptyr� r� r� r� r� r� z-_z&invalid character %r in the anchor: %r)r )r r� r� r r r
r� h s
HzEmitter.prepare_anchorc
C s |s t |dddddddd�S d}d}d}d}d}d}d}d} d}
d} |jd�sX|jd�r`d}d}d} t|�dkpz|d dk}
d}d}d}�x�|t|�k �rv|| }|dkr�|d kr�d}d}|d
kr�d}|
r�d}|d kr�|
r�d}d}n<|d kr�d}|d
k�rd}|
�rd}|dk�r | �r d}d}|dk�r.d}|dk�pPd| k�oLdkn �s�|dk�s�d| k�ordkn �s�d| k�o�dkn �r�|dk�r�d}| j�s�d}nd}|dk�r�|dk�r�d}|t|�d k�r�d}|�r�d}
d}d}nJ|dk�r:|dk�rd}|t|�d k�r&d} |�r0d} d}d}nd}d}|d7 }|dk} |d t|�k�pr||d dk}
q�W d}d}d}d}d}|�s�|�s�|�s�| �r�d }}|�r�d}|
�r�d } }}| �s�|�r�d } } }}|�r�d }}|�r�d}|�rd}t |d||||||d�S )NTF)r r
r r r r r r z---z...r u
…

r z#,[]{}&*!|>'"%@`z?:r� z,?[]{}r� �#u
…

r � �~� …�  u ퟿u u �u )r r� rK r2 )r r Zblock_indicatorsZflow_indicatorsZ line_breaksZspecial_characters�
leading_spaceZ
leading_breakZtrailing_spaceZtrailing_breakZ break_spaceZ space_breakZpreceeded_by_whitespaceZfollowed_by_whitespaceZprevious_spaceZprevious_break�indexr� Zunicode_charactersr r r r r r r r
r� r s�
&&&
zEmitter.analyze_scalarc C s t | jd�r| jj� d S )N�flush)rQ r r� )r r r r
rm s zEmitter.flush_streamc C s* | j r&| j jd�r&| jjdj| j �� d S )Nzutf-16u )r r� r �writer� )r r r r
rR s zEmitter.write_stream_startc C s | j � d S )N)rm )r r r r
rf s zEmitter.write_stream_endc C sf | j s | r|}nd| }|| _ | jo(|| _| jt|�7 _d| _| jrV|j| j�}| jj|� d S )Nr� F) r. r/ r- rK r0 r r� r r� )r r� Zneed_whitespacer. r/ r� r r r
rX s zEmitter.write_indicatorc C s| | j pd}| j s.| j|ks.| j|kr6| j r6| j� | j|k rxd| _d|| j }|| _| jrl|j| j�}| jj|� d S )Nr Tr� ) r& r/ r- r. �write_line_breakr r� r r� )r r& r� r r r
rY - s
zEmitter.write_indentc C sP |d kr| j }d| _d| _| jd7 _d| _| jr@|j| j�}| jj|� d S )NTr r ) r5 r. r/ r, r- r r� r r� )r r� r r r
r� : s zEmitter.write_line_breakc C s2 d| }| j r|j| j �}| jj|� | j� d S )Nz %%YAML %s)r r� r r� r� )r rg r� r r r
r[ E s
z Emitter.write_version_directivec C s6 d||f }| j r |j| j �}| jj|� | j� d S )Nz %%TAG %s %s)r r� r r� r� )r rj rk r� r r r
rb L s
zEmitter.write_tag_directiveTc
C s, | j dd� d}d}d }}�x�|t|�k�rd }|t|�k rF|| }|r�|d ksZ|dkr�|d |kr�| j| jkr�|r�|dkr�|t|�kr�| j� n<|||� }| jt|�7 _| jr�|j| j�}| jj|� |}n�|�rH|d ks�|dk�r�|| dk�r| j � x4|||� D ]$} | dk�r*| j � n
| j | � �qW | j� |}nj|d k�sf|d k�sf|dk�r�||k �r�|||� }| jt|�7 _| j�r�|j| j�}| jj|� |}|dk�r�d
}| jd 7 _| j�r�|j| j�}| jj|� |d }|d k �r|dk}|dk}|d7 }q W | j dd� d S ) Nr� TFr r� r u
…

r u
…

z''r )
rX rK r- r4 rY r r� r r� r� )
r �textr� �spaces�breaksr� r� r� r� �brr r r
r� U sb
zEmitter.write_single_quotedr� r� �b�t�n�v�f�r�er� �\�N�_�L�P)� ��� r � � r �r� r� r� r� u 
u 
c C s� | j dd� d }}�xt|t|�k�r�d }|t|�k r>|| }|d ks�|dks�d| ko`dkn p�| jo�d| ko~dkn p�d | ko�d
kn �r�||k r�|||� }| jt|�7 _| jr�|j| j�}| jj|� |}|d k �r�|| jk�rd | j| }n<|d k�r&d
t |� }n$|dk�r>dt |� }n dt |� }| jt|�7 _| j�rp|j| j�}| jj|� |d }d| k �o�t|�d k n �r�|dk�s�||k�r�| j|| | j
k�r�|�r�|||� d }||k �r�|}| jt|�7 _| j�r |j| j�}| jj|� | j � d| _ d| _
|| dk�r�d }| jt|�7 _| j�rt|j| j�}| jj|� |d7 }qW | j dd� d S )Nr� Tr u
"\…

r� r� r� u ퟿u u �r� � ÿz\x%02Xu ￿z\u%04Xz\U%08Xr F)rX rK r2 r- r r� r r� �ESCAPE_REPLACEMENTSr� r4 rY r. r/ )r r� r� r� r� r� r� r r r
r� � sf
8
zEmitter.write_double_quotedc C s\ d}|rX|d dkr"|t | j�7 }|d dkr8|d7 }n t|�dksP|d
dkrX|d7 }|S ) Nr� r u
…

r u
…

r� r �+rG �����)�strr3 rK )r r� �hintsr r r
�determine_block_hints� s
z Emitter.determine_block_hintsc C s | j |�}| jd| d� |d d � dkr0d| _| j� d}d}d}d }}�x�|t|�k�rd }|t|�k rv|| }|�r
|d ks�|dk�r�| r�|d k r�|dkr�|| d kr�| j� |dk}x0|||� D ] } | d kr�| j� q�| j| � q�W |d k �r| j� |}n�|�r�|dk�r�|d |k�r@| j| jk�r@| j� n>|||� }
| jt|
�7 _| j�rr|
j | j�}
| j
j |
� |}nh|d k�s�|d
k�r�|||� }
| jt|
�7 _| j�r�|
j | j�}
| j
j |
� |d k�r�| j� |}|d k �r|dk}|dk}|d7 }qPW d S ) Nr� Tr r� Fr u
…

r� r u
…

rG ) r� rX r0 r� rK rY r- r4 r r� r r� ) r r� r� r� r� r� r� r� r� r� r� r r r
r� � sb
zEmitter.write_foldedc C s< | j |�}| jd| d� |dd � dkr0d| _| j� d}d }}x�|t|�k�r6d }|t|�k rl|| }|r�|d ks�|dkr�x0|||� D ] }|dkr�| j� q�| j|� q�W |d k r�| j� |}nR|d ks�|dk�r|||� }| jr�|j| j�}| jj |� |d k�r| j� |}|d k �r,|dk}|d7 }qFW d S ) Nr� Tr r� r u
…

r rG )
r� rX r0 r� rK rY r r� r r� ) r r� r� r� r� r� r� r� r� r r r
r� s>
zEmitter.write_literalc
C s | j r d| _|sd S | jsNd}| jt|�7 _| jrB|j| j�}| jj|� d| _d| _ d}d}d }}�x�|t|�k�r
d }|t|�k r�|| }|�r|dk�r�|d |kr�| j| j
kr�|r�| j � d| _d| _ n>|||� }| jt|�7 _| j�r|j| j�}| jj|� |}n�|�r�|dk�r�|| dk�r@| j � x4|||� D ]$} | dk�rf| j � n
| j | � �qNW | j � d| _d| _ |}nV|d k�s�|dk�r�|||� }| jt|�7 _| j�r�|j| j�}| jj|� |}|d k �r |dk}|dk}|d7 }qnW d S ) NTr� Fr r u
…

r u
…

)
r( r0 r. r- rK r r� r r� r/ r4 rY r� )
r r� r� r� r� r� r� r� r� r� r r r
r� 7 sl
zEmitter.write_plain)NNNNN)FF)F)FFFF)F)F)FF)N)T)T)T)@r r r r\ r r< r@ r> rD rP r! rT rS rV rn re rp rs rw ry r� r� r| r� r� r� r� rz r� r� r} r� r� r� r� rx r{ rd r� ru rv r� r� rZ r` ra r� r� r� rm rR rf rX rY r� r[ rb r� r� r� r� r� r� r� r r r r
r s�
C
$
$
"
7
6 6#N)�__all__�errorr r# r r r r r r r
� s