o DDi^@s|ddlmZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z Gdd d ee ZGd d d e eZdS) )MappingN)MetadataDependent)Codemod)CodemodContext)MatcherDecoratableTransformerMatcherDecoratableVisitor) ProviderTc@s8eZdZdZdeddfddZdejdejfdd ZdS) ContextAwareTransformera A transformer which visits using LibCST. Allows visitor-based mutation of a tree. Classes wishing to do arbitrary non-visitor-based mutation on a tree should instead subclass from :class:`Codemod` and implement :meth:`~Codemod.transform_module_impl`. This is a subclass of :class:`~libcst.matchers.MatcherDecoratableTransformer` so all features of matchers as well as :class:`~libcst.CSTTransformer` are available to subclasses of this class. contextreturnNcCst||t|dSN)r__init__r)selfr rP/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/_visitor.pyr s z ContextAwareTransformer.__init__treecCs ||Sr )visit)rrrrrtransform_module_impls z-ContextAwareTransformer.transform_module_impl) __name__ __module__ __qualname____doc__rr cstModulerrrrrr s r c@sHeZdZdZdeddfddZdeddfdd Zede j fd d Z dS) ContextAwareVisitora A visitor which visits using LibCST. Allows visitor-based collecting of info on a tree. All codemods which wish to implement an information collector should subclass from this instead of directly from :class:`~libcst.matchers.MatcherDecoratableVisitor` or :class:`~libcst.CSTVisitor` since this provides access to the current codemod context. As a result, this class allows access to metadata which was calculated in a parent :class:`~libcst.codemod.Codemod` through the :meth:`~libcst.MetadataDependent.get_metadata` method. Note that you cannot directly run a :class:`~libcst.codemod.ContextAwareVisitor` using :func:`~libcst.codemod.transform_module` because visitors by definition do not transform trees. However, you can instantiate a :class:`~libcst.codemod.ContextAwareVisitor` inside a codemod and pass it to the :class:`~libcst.CSTNode.visit` method on any node in order to run information gathering with metadata and context support. Remember that a :class:`~libcst.codemod.ContextAwareVisitor` is a subclass of :class:`~libcst.MetadataDependent`, meaning that you still need to declare your metadata dependencies with :attr:`~libcst.MetadataDependent.METADATA_DEPENDENCIES` before you can retrieve metadata using :meth:`~libcst.MetadataDependent.get_metadata`, even if the parent codemod has listed its own metadata dependencies. Note also that the dependencies listed on this class must be a strict subset of the dependencies listed in the parent codemod. r r Ncst|t|||_|}|rZ|jjdur)td|jjddd|D]"}|j vrMtd|jdddd |jjd |jd q+fd d |D|_ dSdS)NzAttempting to instantiate z outside of z:an active transform. This means that metadata hasn't been z:calculated and we cannot successfully create this visitor.zAttempting to access metadata z that was not a z:declared dependency of parent transform! This means it is z;not possible to compute this value. Please ensure that all zparent transforms of z declare z as a metadata dependency.csi|]}|j|qSr) _metadata).0depwrapperrr Vs z0ContextAwareVisitor.__init__..) rr rr get_inherited_dependenciesr Exception __class__rrmetadata)rr dependenciesrrrrr ?s@       zContextAwareVisitor.__init__warningcCs|jj|dS)z\ Emit a warning that is displayed to the user who has invoked this codemod. N)r warningsappend)rr&rrrwarnZszContextAwareVisitor.warncCs,|jj}|durtd|jjdd|S)z Reference to the currently-traversed module. Note that this is only available during a transform itself. NzAttempted access of z.module outside of ztransform_module().)r moduler"r#r)rr*rrrr*`szContextAwareVisitor.module) rrrrrr strr)propertyrrr*rrrrr#s r)typingrlibcstrrlibcst.codemod._codemodrlibcst.codemod._contextrlibcst.matchersrrlibcst.metadatarr rrrrrs