o DDi@srddlZddlmZddlmZmZddlZddlmZm Z Gddde eZ Gddde Z Gd d d e Z dS) N)ABC)PatternUnion)CodemodContextVisitorBasedCodemodCommandcsVeZdZUeed<deddffdd Zdejdejde ejej ffd d Z Z S) RemovePyreDirectiveCommandPYRE_TAGcontextreturnNcs&t|td|jd|_dS)Nz ^#\s+pyre-z\s*$)super__init__recompiler_regex_pattern)selfr  __class__f/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/commands/remove_pyre_directive.pyr s   z#RemovePyreDirectiveCommand.__init__ original_node updated_nodecCs4|jdust|jt|jtjjs|StS)N) commentboolrsearchlibcst ensure_typeCommentvalueRemoveFromParent)rrrrrrleave_EmptyLines z*RemovePyreDirectiveCommand.leave_EmptyLine) __name__ __module__ __qualname__str__annotations__rr r EmptyLinerRemovalSentinelr __classcell__rrrrrs rc@*eZdZUdZdZeed<dZeed<dS)RemovePyreStrictCommandzc Given a source file, we'll remove the any strict tag if the file already contains it. z,Removes the 'pyre-strict' tag from a module. DESCRIPTIONstrictrNr r!r"__doc__r*r#r$rrrrrr)%  r)c@r()RemovePyreUnsafeCommandzc Given a source file, we'll remove the any unsafe tag if the file already contains it. z,Removes the 'pyre-unsafe' tag from a module.r*unsaferNr,rrrrr/0r.r/) r abcrtypingrrrlibcst.codemodrrrr)r/rrrrs