o DDiQ!@sddlZddlZddlmZmZddlmZmZmZm Z m Z ddl m Z ddl mZddlmZddlmZddlmZdd lmZe d ed ZGd d d eeZGdddeeeZGdddeeZdS)N)ABCabstractmethod)Dict GeneratorListTypeTypeVar)Module)Codemod)CodemodContext)ContextAwareTransformer)AddImportsVisitor)RemoveImportsVisitor_Codemod)boundcseZdZUdZdZeed<edej ddfddZ d e e d e de fd d Zed e de fd dZd e de ffdd ZZS)CodemodCommanda* A :class:`~libcst.codemod.Codemod` which can be invoked on the command-line using the ``libcst.tool codemod`` utility. It behaves like any other codemod in that it can be instantiated and run identically to a :class:`~libcst.codemod.Codemod`. However, it provides support for providing help text and command-line arguments to ``libcst.tool codemod`` as well as facilities for automatically running certain common transforms after executing your :meth:`~libcst.codemod.Codemod.transform_module_impl`. The following list of transforms are automatically run at this time: - :class:`~libcst.codemod.visitors.AddImportsVisitor` (adds needed imports to a module). - :class:`~libcst.codemod.visitors.RemoveImportsVisitor` (removes unreferenced imports from a module). zNo description. DESCRIPTION arg_parserreturnNcCdS)a Override this to add arguments to the CLI argument parser. These args will show up when the user invokes ``libcst.tool codemod`` with ``--help``. They will also be presented to your class's ``__init__`` method. So, if you define a command with an argument 'foo', you should also have a corresponding 'foo' positional or keyword argument in your class's ``__init__`` method. N)rrrP/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/_command.pyadd_args*s zCodemodCommand.add_args transformtreecCs||j}||SN)contexttransform_module)selfrrinstrrr_instantiate_and_run7s  z#CodemodCommand._instantiate_and_runcCr)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_impl;sz$CodemodCommand.transform_module_implcsJt|}tjttjti}|D]\}}||jjvr"|||}q|Sr) superrr CONTEXT_KEYritemsrscratchr )rrsupported_transformskeyr __class__rrrDs   zCodemodCommand.transform_module)__name__ __module__ __qualname____doc__rstr__annotations__ staticmethodargparseArgumentParserrrrr r rr!r __classcell__rrr(rrs   rc@seZdZdZdS)VisitorBasedCodemodCommanda[ A command that acts identically to a visitor-based transform, but also has the support of :meth:`~libcst.codemod.CodemodCommand.add_args` and running supported helper transforms after execution. See :class:`~libcst.codemod.CodemodCommand` and :class:`~libcst.codemod.ContextAwareTransformer` for additional documentation. N)r*r+r,r-rrrrr4]sr4cs~eZdZdZdedeeefddffdd Ze de e e ddffdd Z d e edefd d Zd edefddZZS)MagicArgsCodemodCommanda- A "magic" args command, which auto-magically looks up the transforms that are yielded from :meth:`~libcst.codemod.MagicArgsCodemodCommand.get_transforms` and instantiates them using values out of the context. Visitors yielded in :meth:`~libcst.codemod.MagicArgsCodemodCommand.get_transforms` must have constructor arguments that match a key in the context :attr:`~libcst.codemod.CodemodContext.scratch`. The easiest way to guarantee that is to use :meth:`~libcst.codemod.CodemodCommand.add_args` to add a command arg that will be parsed for each of the args. However, if you wish to chain transforms, adding to the scratch in one transform will make the value available to the constructor in subsequent transforms as well as the scratch for subsequent transforms. rkwargsrNc st||jj|dSr)r"__init__rr%update)rrr6r(rrr7xs z MagicArgsCodemodCommand.__init__cCr)ai A generator which yields one or more subclasses of :class:`~libcst.codemod.Codemod`. In the general case, you will usually yield a series of classes, but it is possible to programmatically decide which classes to yield depending on the contents of the context :attr:`~libcst.codemod.CodemodContext.scratch`. Note that you should yield classes, not instances of classes, as the point of :class:`~libcst.codemod.MagicArgsCodemodCommand` is to instantiate them for you with the contents of :attr:`~libcst.codemod.CodemodContext.scratch`. Nr)rrrrget_transforms|sz&MagicArgsCodemodCommand.get_transformsrc Cst|j}g}i}t|jt|jpd}t|jD]/\}}|dvr$q||jjvrA||kr/qt d|j d|dddd| |jj|q|j pOi}|j D]*} | |jjvrq| |vrqt d|j d | dddd|jj| || || <qS||jg|Ri|S) Nr)rrzVisitor z requires positional arg z but z?it is not in our context nor does it have a default! It should z?be provided by an argument returned from the 'add_args' method z:or populated into context.scratch by a previous transform!z requires keyword arg )inspectgetfullargspecr7lenargsdefaults enumeraterr%KeyErrorr*appendkwonlydefaults kwonlyargsget) rrargspecr=r6last_default_argiargrBkwargrrr _instantiatesD    z$MagicArgsCodemodCommand._instantiatercCs&|D] }||}||}q|Sr)r9rJr)rrrrrrrr!s   z-MagicArgsCodemodCommand.transform_module_impl)r*r+r,r-r rr.objectr7rrrr r9rrJr r!r3rrr(rr5is"$r5)r1r:abcrrtypingrrrrrlibcstr libcst.codemod._codemodr libcst.codemod._contextr libcst.codemod._visitorr $libcst.codemod.visitors._add_importsr 'libcst.codemod.visitors._remove_importsrrrr4r5rrrrs       H