
    jj"                        d dl mZ d dlZd dlmZmZ d dlmZmZ d dl	m
Z
mZmZmZmZmZmZmZ d dlmZmZmZ 	 g d	Z G d
 de          Z G d de          Z G d d          ZdS )    )annotationsN)MarkedYAMLErrorReusedAnchorWarning)nprintnprintf)StreamStartEventStreamEndEventMappingStartEventMappingEndEventSequenceStartEventSequenceEndEvent
AliasEventScalarEvent)MappingNode
ScalarNodeSequenceNodeF)AnyDictOptionalList)ComposerComposerErrorMaxDepthExceededErrorc                      e Zd ZdS )r   N__name__
__module____qualname__     d/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/ruamel/yaml/composer.pyr   r              Dr    r   c                      e Zd ZdS )r   Nr   r   r    r!   r   r      r"   r    r   c                      e Zd ZdddZedd            Ze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 dZd!dZdS )"r   Nloaderr   returnNonec                    || _         | j         "t          | j         dd           | | j         _        i | _        d| _        d| _        d S )N	_composerTr   )r%   getattrr)   anchorswarn_double_anchorsdepth)selfr%   s     r!   __init__zComposer.__init__$   sH    ;"wt{K'N'N'V$(DK!')#' 


r    c                \    t          | j        d          r| j        j         | j        j        S Ntyp)hasattrr%   parser_parserr.   s    r!   r4   zComposer.parser,   s,    4;&& 	K{""r    c                \    t          | j        d          r| j        j         | j        j        S r1   )r3   r%   resolver	_resolverr6   s    r!   r8   zComposer.resolver2   s.     4;&& 	!K  {$$r    c                    | j                             t                    r| j                                          | j                             t                     S N)r4   check_eventr   	get_eventr	   r6   s    r!   
check_nodezComposer.check_node9   sJ    ;""#344 	$K!!### ;**>::::r    c                l    | j                             t                    s|                                 S d S r;   )r4   r<   r	   compose_documentr6   s    r!   get_nodezComposer.get_nodeA   s6    {&&~66 	+((***	+ 	+r    c                |   | j                                          d }| j                             t                    s|                                 }| j                             t                    s5| j                                         }t          d|j        d|j                  | j                                          |S )Nz(expected a single document in the streamzbut found another document)r4   r=   r<   r	   r@   r   
start_mark)r.   documentevents      r!   get_single_nodezComposer.get_single_nodeF   s     {&&~66 	/,,..H {&&~66 	K))++E:#, 	   	r    r.   c                    i | _         | j                                         |                     d d           }| j                                         |S r;   )r+   r4   r=   compose_node)r.   nodes     r!   r@   zComposer.compose_document^   sO       t,, 	r    ac                    |S r;   r   )r.   rJ   s     r!   return_aliaszComposer.return_aliask   s    r    parentindexc                L   | j                             t                    rc| j                                         }|j        }|| j        vrt          d d d||j                  |                     | j        |                   S | xj	        dz  c_	        | j         
                                }| j        j        r5| j	        | j        j        k    r t          d d d| j	         d|j                  |j        }|L| j        rE|| j        v r<d|d| j        |         j         d|j         }t          j        |t"          d	           | j                            ||           | j                             t(                    r|                     |          }ni| j                             t,                    r|                     |          }n4| j                             t0                    r|                     |          }| j                                         | xj	        dz  c_	        |S )
Nzfound undefined alias    z*maximum depth of data structure exceeded (z)), if necessary increase YAML().max_depthz
found duplicate anchor z
first occurrence z
second occurrence    )
stacklevel)r4   r<   r   r=   anchorr+   r   rC   rL   r-   
peek_eventr%   	max_depthr   r,   warningswarnr   r8   descend_resolverr   compose_scalar_noder   compose_sequence_noder
   compose_mapping_nodeascend_resolver)r.   rM   rN   rE   aliasrS   wsrI   s           r!   rH   zComposer.compose_noden   s@   ;"":.. 	:K))++ELEDL((#$ B B BEDT   $$T\%%8999

a

&&((;  	TZ$+2G%G%G'9TZ 9 9 9    ' EFdl,B,B< < <(,V(<(G< <).)9< < 
 b"5!DDDD&&vu555;"";// 	5++F33DD[$$%788 	5--f55DD[$$%677 	5,,V44D%%'''

a

r    rS   c           	     j   | j                                         }|j        }|t          |          dk    rB| j                            t          |j        |j                  }t          |t                    rJ t          ||j        |j
        |j        |j        |j        |          }|
|| j        |<   |S )N!)stylecommentrS   )r4   r=   ctagstrr8   resolver   valueimplicit
isinstancerC   end_markra   rb   r+   )r.   rS   rE   tagrI   s        r!   rY   zComposer.compose_scalar_node   s    %%''j;#c((c//-''
EKPPC!#s+++++KN+M
 
 
 #'DL r    c           	     $   | j                                         }|j        }|t          |          dk    r=| j                            t          d |j                  }t          |t                    rJ t          |g |j	        d |j
        |j        |          }|
|| j        |<   d}| j                             t                    sR|j                            |                     ||                     |dz  }| j                             t                    R| j                                         }|j
        du r5|j        .|j        |j
        }t%          d| dd           |j        |_        |j        |_        |                     ||           |S )	Nr`   
flow_stylerb   rS   r   rP   Tz8Warning: unexpected end_event commment in sequence node 
zC    if possible, please report an issue with reproducable data/code)r4   r=   rc   rd   r8   re   r   rg   rh   rC   rm   rb   r+   r<   r   rf   appendrH   r   ri   check_end_doc_comment)r.   rS   start_eventrj   rI   rN   	end_eventxs           r!   rZ   zComposer.compose_sequence_node   s   k++--;#c((c//-''dK<PQQC!#s+++++""-'
 
 
 #'DL +))*:;; 	Jd//e<<===QJE +))*:;; 	 K))++	?d""y'8'D|'OTqTTTY   %,DL!*""9d333r    c           	        | j                                         }|j        }|t          |          dk    r=| j                            t          d |j                  }t          |t                    rJ t          |g |j	        d |j
        |j        |          }|
|| j        |<   | j                             t                    sg|                     |d           }|                     ||          }|j                            ||f           | j                             t                    g| j                                         }|j
        du r|j        |j        |_        |j        |_        |                     ||           |S )Nr`   rl   T)r4   r=   rc   rd   r8   re   r   rg   rh   rC   rm   rb   r+   r<   r   rH   rf   ro   ri   rp   )r.   rS   rq   rj   rI   item_key
item_valuerr   s           r!   r[   zComposer.compose_mapping_node   sz   k++--;#c((c//-''T;;OPPC!#s+++++""-'
 
 
 #'DL +))/:: 		6((t44H
 **4::JJx4555 +))/:: 		6 K))++	?d""y'8'D$,DL!*""9d333r    rr   rI   c                    |j         re|j         d         rZ|j         	d d g|_         t          |t                    rJ |j                             |j         d                    d |j         d<   d S d S d S )NrP   )rb   rh   r   ro   )r.   rr   rI   s      r!   rp   zComposer.check_end_doc_comment   s     	(!21!5 	(|# $d|!$44444 L	 1! 4555#'Ia   	( 	( 	( 	(r    r;   )r%   r   r&   r'   )r&   r   )r.   r   r&   r   )rJ   r   r&   r   )rM   r   rN   r   r&   r   )rS   r   r&   r   )rr   r   rI   r   r&   r'   )r   r   r   r/   propertyr4   r8   r>   rA   rF   r@   rL   rH   rY   rZ   r[   rp   r   r    r!   r   r   #   s8            # # # X#
 % % % X%; ; ; ;+ + + +
   0      % % % %N   (       D       D	( 	( 	( 	( 	( 	(r    r   )
__future__r   rV   ruamel.yaml.errorr   r   ruamel.yaml.compatr   r   ruamel.yaml.eventsr   r	   r
   r   r   r   r   r   ruamel.yaml.nodesr   r   r   typingr   r   r   r   __all__r   r   r   r   r    r!   <module>r      sx   " " " " " "  B B B B B B B B . . . . . . . .	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 D C C C C C C C C C1 A
@
@	 	 	 	 	O 	 	 		 	 	 	 	O 	 	 	S( S( S( S( S( S( S( S( S( S(r    