o DDi4@sjddlmZmZddlmZddlmZddlmZddl m Z m Z m Z ddl mZGddde eZd S) )ABCabstractmethod)contextmanager)replace) Generator)MetadataDependentMetadataWrapperModule)CodemodContextc@seZdZdZdeddfddZdefddZd eddfd d Z e de fd d Z e de de fddZede dee ddffddZde de fddZdS)Codemoda Abstract base class that all codemods must subclass from. Classes wishing to perform arbitrary, non-visitor-based mutations on a tree should subclass from this class directly. Classes wishing to perform visitor-based mutation should instead subclass from :class:`~libcst.codemod.ContextAwareTransformer`. Note that a :class:`~libcst.codemod.Codemod` is a subclass of :class:`~libcst.MetadataDependent`, meaning that you can declare metadata dependencies with the :attr:`~libcst.MetadataDependent.METADATA_DEPENDENCIES` class property and while you are executing a transform you can call :meth:`~libcst.MetadataDependent.get_metadata` to retrieve the resolved metadata. contextreturnNcCst|||_dS)N)r__init__r )selfr rP/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/_codemod.pyrs  zCodemod.__init__cCdS)z Override this and return ``True`` to allow your transform to be called repeatedly until the tree doesn't change between passes. By default, this is off, and should suffice for most transforms. Fr)rrrrshould_allow_multiple_passes"sz$Codemod.should_allow_multiple_passeswarningcCs|jj|dS)z\ Emit a warning that is displayed to the user who has invoked this codemod. N)r warningsappend)rrrrrwarn*sz Codemod.warncCs,|jj}|durtd|jjdd|S)av Reference to the currently-traversed module. Note that this is only available during the execution of a codemod. The module reference is particularly handy if you want to use :meth:`libcst.Module.code_for_node` or :attr:`libcst.Module.config_for_parsing` and don't wish to track a reference to the top-level module manually. NzAttempted access of z.module outside of ztransform_module().)r module Exception __class____name__)rrrrrr0s zCodemod.moduletreecCr)z Override this with your transform. You should take in the tree, optionally mutate it and then return the mutated version. The module reference and all calculated metadata are available for the lifetime of this function. Nr)rrrrrtransform_module_implAszCodemod.transform_module_implrc cs|jj}|jj}|jj}|dur|r||}t||d}nt|}||)t|j|d|_z|jVWt|j|d|_n t|j|d|_wWddS1sTwYdS)N)cache)wrapper) r rmetadata_managerfilenameget_cache_for_pathrresolverr)rr oldwrapperr r!rrrrr_handle_metadata_referenceJs    &"z"Codemod._handle_metadata_referencecCs|s|| }||WdS1swY|} || }||}Wdn1s6wY||rC |S|}q!)a  Transform entrypoint which handles multi-pass logic and metadata calculation for you. This is the method that you should call if you wish to invoke a codemod directly. This is the method that is called by :func:`~libcst.codemod.transform_module`. N)rr%r deep_equals)rrtree_with_metadatapreviousrrrtransform_modulebs     zCodemod.transform_module)r __module__ __qualname____doc__r rboolrstrrpropertyr rrrrrr%r)rrrrr s   r N)abcrr contextlibr dataclassesrtypingrlibcstrrr libcst.codemod._contextr r rrrrs