o DDi" @sddlZddlmZmZmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZmZe r4ddlmZedgdfZeeeefZGdd d e e Z  dd ed eed eed eedef ddZd eedefddZGdddeZdS)N)CallablecastIterableListMappingMutableMappingOptional TYPE_CHECKING)MetadataDependent)CSTTypedVisitorFunctions)CSTNodeT CSTVisitor)CSTNoderc@s&eZdZdZdeeeffddZdS)BatchableCSTVisitoraq The low-level base visitor class for traversing a CST as part of a batched set of traversals. This should be used in conjunction with the :func:`~libcst.visit_batched` function or the :func:`~libcst.MetadataWrapper.visit_batched` method from :class:`~libcst.MetadataWrapper` to visit a tree. Instances of this class cannot modify the tree. returncCst|dd}t|S)a Returns a mapping of all the ``visit_``, ``visit__``, ``leave_`` and `leave__`` methods defined by this visitor, excluding all empty stubs. cSs0t|o|jdp|jdot|dd S)Nvisit_leave_ _is_no_opF)inspectismethod__name__ startswithgetattr)mrP/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_batched_visitor.py1s z2BatchableCSTVisitor.get_visitors..)r getmembersdict)selfmethodsrrr get_visitors's  z BatchableCSTVisitor.get_visitorsN)r __module__ __qualname____doc__rstr VisitorMethodr!rrrrrs rnodebatchable_visitors before_visit after_leavercCs&t|}t|||d}tt||S)a3 Do a batched traversal over ``node`` with all ``visitors``. ``before_visit`` and ``after_leave`` are provided as optional hooks to execute before the ``visit_`` and after the ``leave_`` methods from each visitor in ``visitor`` are executed by the batched visitor. This function does not handle metadata dependency resolution for ``visitors``. See :func:`~libcst.MetadataWrapper.visit_batched` from :class:`~libcst.MetadataWrapper` for batched traversal with metadata dependency resolution. r)r*)_get_visitor_methods_BatchedCSTVisitorrr visit)r'r(r)r*visitor_methodsbatched_visitorrrr visit_batched?s r1cCs:i}|D]}|D] \}}||g|q q|S)z Gather all ``visit_``, ``visit__``, ``leave_`` amd `leave__`` methods from ``batchabled_visitors``. )r!items setdefaultappend)r(r/bvnamefnrrrr,Ys r,c seZdZUdZeed<eeed<eeed<ddddedeedeeddffdd Zd d de fd d Z dddZ d d de ddfddZ dd de ddfddZZS)r-zJ Internal visitor class to perform batched traversal over a tree. r/r)r*Nr+rcs t||_||_||_dS)N)super__init__r/r)r*)rr/r)r* __class__rrr9qs  z_BatchedCSTVisitor.__init__r'rcCsF|j}|dur ||t|j}|jd|gD]}||qdS)zR Call appropriate visit methods on node before visiting children. NrT)r)typerr/get)rr'r) type_namevrrron_visit}s  z_BatchedCSTVisitor.on_visit original_nodecCsJt|j}|jd|gD]}||q|j}|dur#||dSdS)zQ Call appropriate leave methods on node after visiting children. rN)r<rr/r=r*)rrAr>r?r*rrron_leaves   z_BatchedCSTVisitor.on_leave attributecC6t|j}|jd|d|gD]}||qdS)zp Call appropriate visit attribute methods on node before visiting attribute's children. r_Nr<rr/r=)rr'rCr>r?rrron_visit_attribute  z%_BatchedCSTVisitor.on_visit_attributecCrD)zo Call appropriate leave attribute methods on node after visiting attribute's children. rrENrF)rrArCr>r?rrron_leave_attributerHz%_BatchedCSTVisitor.on_leave_attribute)rArrN)rr"r#r$_VisitorMethodCollection__annotations__rr&r9boolr@rBr%rGrI __classcell__rrr:rr-hs(      r-)NN)rtypingrrrrrrrr libcst._metadata_dependentr libcst._typed_visitorr libcst._visitorsr r libcst._nodes.baserr&r%rJrr1r,r-rrrrs:(  %