o DDi @sddlmZddlmZmZddlmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZddlmZe rLdd lmZdd lmZeed d Gd ddZdddededddef ddZdddede eddde ef ddZdddede eefddde eeff ddZdddedeedddeef ddZ dddede eddde ef dd Z!dddede edddeef d!d"Z"dddede eddde ef d#d$Z#d%S)&)contextmanager) dataclassfield)IterableIteratorListOptionalSequence TYPE_CHECKINGUnion) add_slots)FlattenSentinel) MaybeSentinel)RemovalSentinel)CSTNodeT)CSTNode) CSTVisitorTF)frozenc @seZdZUeed<eed<dZeed<eedZ e eed<eedZ e eed<ded dfd d Z dd d Z dddZded dfddZd ddZd ddZdddZeddddddeddedd edfddZdS)! CodegenStatedefault_indentdefault_newlineNprovider)default_factory indent_tokenstokensvaluereturncC|j|dSN)rappendselfrr"O/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_nodes/internal.pyincrease_indent"zCodegenState.increase_indentcCs|jdSr)rpopr!r"r"r#decrease_indent%szCodegenState.decrease_indentcCs|j|jdSr)rextendrr'r"r"r#add_indent_tokens(szCodegenState.add_indent_tokenscCrr)rrr r"r"r# add_token+r%zCodegenState.add_tokennodercCdSrr"r!r,r"r"r#before_codegen.zCodegenState.before_codegencCr-rr"r.r"r"r# after_codegen1r0zCodegenState.after_codegencCs t|jdkr|jdSdS)z Called by :meth:`libcst.Module._codegen_impl` at the end of the file to remove the last token (a trailing newline), assuming the file isn't empty. rN)lenrr&r'r"r"r#pop_trailing_newline4sz!CodegenState.pop_trailing_newline) start_nodeend_noder4r5ccs dVdSrr")r!r,r4r5r"r"r#record_syntactic_position>s z&CodegenState.record_syntactic_position)rN)r,rrN)__name__ __module__ __qualname__str__annotations__robjectrlistrrrr$r(r*r+r/r1r3rrrr6r"r"r"r#rs2        rparentr fieldnamer,visitorrrcCsr|||||}t|trtdt|jddt|tr1tdt|jddd||||S)z| Given a node, visits the node using `visitor`. If removal is attempted by the visitor, an exception is raised. z*We got a RemovalSentinel while visiting a . This z.node's parent does not allow it to be removed.*We got a FlattenSentinel while visiting a Cnode's parent does not allow for it to be it to be replaced with a sequence.) on_visit_attributevisit isinstancer TypeErrortyper7r on_leave_attributer>r?r,r@resultr"r"r#visit_requiredIs$     rMcCs|dur||||||dS|||||}t|tr1tdt|jddd|||t|tr>dS|S)zx Given an optional node, visits the node if it exists with `visitor`. If the node is removed, returns None. NrBrArCrD) rErJrFrGr rHrIr7rrKr"r"r#visit_optionalbs       rNcCst|tr||||||tjS|||||}t|tr3tdt|j ddd|||t|t rAtjS|S)z Given a node that can be a real value or a sentinel value, visits the node if it is real with `visitor`. If the node is removed, returns MaybeSentinel. rBrArCrD) rGrrErJDEFAULTrFr rHrIr7rrKr"r"r#visit_sentinelys      rPchildrenccsX||||D]}||}t|tr|EdHq t|ts#|Vq |||dS)z Given an iterable of children, visits each child with `visitor`, and yields the new children with any `RemovalSentinel` values removed. N)rErFrGr rrJ)r>r?rQr@child new_childr"r"r#visit_iterables     rTcCtt||||S)zh A convenience wrapper for `visit_iterable` that returns a sequence instead of an iterable. )tuplerTr>r?rQr@r"r"r#visit_sequence rXccs||||D]0}||}t|trq t|tr-|D]}|s(|r(q|Vqq |s6|r6q |Vq |||dS)z Similar to visit_iterable above, but capable of discarding empty SimpleStatementLine nodes in order to preserve correct pass insertion behavior. N)rErFrGrr _is_removablerJ)r>r?rQr@rRrSchild_r"r"r#visit_body_iterables     r\cCrU)zm A convenience wrapper for `visit_body_iterable` that returns a sequence instead of an iterable. )rVr\rWr"r"r#visit_body_sequencerYr]N)$ contextlibr dataclassesrrtypingrrrrr r r libcst._add_slotsr libcst._flatten_sentinelr libcst._maybe_sentinelrlibcst._removal_sentinelr libcst._typesrlibcst._nodes.baserlibcst._visitorsrrr:rMrNrPrTrXr\r]r"r"r"r#s $       0         %