o DDi[ @s dZddlZddlmZddlmZddlmZmZm Z ddl m Z m Z ddl mZedd Gd d d Zedd Gd d d Zedd GdddZGdddeZedd GdddZGdddeZe eeeefZdddededeedefddZdS)z5 Provides everything needed to run a CodemodCommand. N) dataclass)Enum)OptionalSequenceUnion) parse_modulePartialParserConfig)CodemodT)frozenc@s&eZdZUdZeeed<eed<dS)TransformSuccessz A :class:`~libcst.codemod.TransformResult` used when the codemod was successful. Stores all the information we might need to display to the user upon success, as well as the transformed file contents. warning_messagescodeN)__name__ __module__ __qualname____doc__rstr__annotations__rrO/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/_runner.pyr s   r c@s.eZdZUdZeeed<eed<eed<dS)TransformFailurez A :class:`~libcst.codemod.TransformResult` used when the codemod failed. Stores all the information we might need to display to the user upon a failure. r error traceback_strN)rrrrrrr Exceptionrrrrr&s   rc@s"eZdZUdZdZeeed<dS) TransformExitz A :class:`~libcst.codemod.TransformResult` used when the codemod was interrupted by the user (e.g. KeyboardInterrupt). rr N)rrrrr rrrrrrrr7s rc@seZdZdZdZdZdZdS) SkipReasonzD An enumeration of all valid reasons for a codemod to skip. generated blacklistedotherN)rrrr GENERATED BLACKLISTEDOTHERrrrrrCs rc@s2eZdZUdZeed<eed<dZeeed<dS) TransformSkipz A :class:`~libcst.codemod.TransformResult` used when the codemod requested to be skipped. This could be because it's a generated file, or due to filename blacklist, or because the transform raised :class:`~libcst.codemod.SkipFile`. skip_reasonskip_descriptionrr N) rrrrrrrr rrrrrr"Us r"c@seZdZdZdS)SkipFilez Raise this exception to skip codemodding the current file. The exception message should be the reason for skipping. N)rrrrrrrrr%gsr%python_version transformerr r'returnc Cszt||dur t|dntd}||}t|j|jjdWSty+tYSt yH}zt t |t j |jjdWYd}~Sd}~wtyd}zt|t|jjdWYd}~Sd}~ww)a Given a module as represented by a string and a :class:`~libcst.codemod.Codemod` that we wish to run, execute the codemod on the code and return a :class:`~libcst.codemod.TransformResult`. This should never raise an exception. On success, this returns a :class:`~libcst.codemod.TransformSuccess` containing any generated warnings as well as the transformed code. If the codemod is interrupted with a Ctrl+C, this returns a :class:`~libcst.codemod.TransformExit`. If the codemod elected to skip by throwing a :class:`~libcst.codemod.SkipFile` exception, this will return a :class:`~libcst.codemod.TransformSkip` containing the reason for skipping as well as any warnings that were generated before the codemod decided to skip. If the codemod throws an unexpected exception, this will return a :class:`~libcst.codemod.TransformFailure` containing the exception that occured as well as any warnings that were generated before the codemod crashed. Nr&)config)r r )r$r#r )rrr )rrtransform_moduler r contextwarningsKeyboardInterruptrr%r"rrr!rr traceback format_exc)r(r r' input_tree output_treeexrrrr+ts8     r+)rr/ dataclassesrenumrtypingrrrlibcstrrlibcst.codemod._codemodr r rrrr"rr%TransformResultrr+rrrrs<