!w:[ � @ s] d d g Z d d l m Z d d l Td d l TGd d � d e � Z Gd d � d � Z d S)�Composer�
ComposerError� )�MarkedYAMLError)�*c @ s e Z d Z d S)r N)�__name__�
__module__� __qualname__� r r � /composer.pyr s c @ s| e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d S)r c C s
i | _ d S)N)�anchors)�selfr r r
�__init__
s zComposer.__init__c C s' | j t � r | j � | j t � S)N)� check_eventZStreamStartEvent� get_event�StreamEndEvent)r r r r
�
check_node s
zComposer.check_nodec C s | j t � s | j � Sd S)N)r r �compose_document)r r r r
�get_node s zComposer.get_nodec C so | j � d } | j t � s+ | j � } | j t � sa | j � } t d | j d | j � � | j � | S)Nz(expected a single document in the streamzbut found another document)r r r r r �
start_mark)r Zdocument�eventr r r
�get_single_node s
zComposer.get_single_nodec C s3 | j � | j d d � } | j � i | _ | S)N)r � compose_noder )r �noder r r
r 2 s
zComposer.compose_documentc C s. | j t � rZ | j � } | j } | | j k rO t d d d | | j � � | j | S| j � } | j } | d k r� | | j k r� t d | | j | j d | j � � | j | | � | j t � r� | j
| � } n? | j t � r| j | � } n | j t
� r | j | � } | j � | S)Nzfound undefined alias %rz*found duplicate anchor %r; first occurencezsecond occurence)r Z
AliasEventr �anchorr r r Z
peek_eventZdescend_resolverZ ScalarEvent�compose_scalar_nodeZSequenceStartEvent�compose_sequence_nodeZMappingStartEvent�compose_mapping_nodeZascend_resolver)r �parent�indexr r r r r r
r ? s.
zComposer.compose_nodec C s� | j � } | j } | d k s- | d k rH | j t | j | j � } t | | j | j | j d | j �} | d k r� | | j | <| S)N�!�style)
r �tag�resolveZ
ScalarNode�value�implicitr �end_markr r )r r r r! r r r r
r X s
z Composer.compose_scalar_nodec C s� | j � } | j } | d k s- | d k rE | j t d | j � } t | g | j d d | j �} | d k r | | j | <d } x9 | j t � s� | j
j | j | | � � | d 7} q� W| j � } | j
| _
| S)Nr �
flow_style� r )r r! r" Z SequenceNoder$ r r& r r ZSequenceEndEventr# �appendr r% )r r � start_eventr! r r � end_eventr r r
r c s
z Composer.compose_sequence_nodec C s� | j � } | j } | d k s- | d k rE | j t d | j � } t | g | j d d | j �} | d k r | | j | <xM | j t � s� | j
| d � } | j
| | � } | j j | | f � q� W| j � } | j
| _
| S)Nr r& )r r! r" Z MappingNoder$ r r& r r ZMappingEndEventr r# r( r% )r r r) r! r Zitem_keyZ
item_valuer* r r r
r u s
z Composer.compose_mapping_nodeN) r r r r
r r r r r r r r r r r r
r s
N)�__all__�errorr �eventsZnodesr r r r r r
�<module> s