o DDi@szddlZddlmZddlmZddlZddlmZmZddl m Z GdddeeZ Gdd d e Z Gd d d e Z dS) N)ABC)Pattern)CodemodContextVisitorBasedCodemodCommand)insert_header_commentscs`eZdZUeed<deddffdd Zdejddfdd Z d ej d ej dej fd d Z Z S)AddPyreDirectiveCommandPYRE_TAGcontextreturnNcs,t|td|jd|_d|_dS)Nz ^#\s+pyre-z\s*$T)super__init__recompiler_regex_pattern needs_add)selfr  __class__c/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/commands/add_pyre_directive.pyr s   z AddPyreDirectiveCommand.__init__nodecCs|j|jr d|_dSdS)NF)rsearchvaluer)rrrrr visit_Comments z%AddPyreDirectiveCommand.visit_Comment original_node updated_nodecCs|js|St|d|jgS)Nz# pyre-)rrr)rrrrrr leave_Modulesz$AddPyreDirectiveCommand.leave_Module) __name__ __module__ __qualname__str__annotations__rr libcstCommentrModuler __classcell__rrrrrs rc@*eZdZUdZdZeed<dZeed<dS)AddPyreStrictCommandzc Given a source file, we'll add the strict tag if the file doesn't already contain it. strictrz&Add the 'pyre-strict' tag to a module. DESCRIPTIONNrrr__doc__rr r!r)rrrrr''  r'c@r&)AddPyreUnsafeCommandzc Given a source file, we'll add the unsafe tag if the file doesn't already contain it. unsaferz&Add the 'pyre-unsafe' tag to a module.r)Nr*rrrrr-2r,r-)r abcrtypingrr"libcst.codemodrrlibcst.helpersrrr'r-rrrrs