o DDiN@sHddlZddlmZmZddlmZmZddlmZGdddeZ dS)N) GeneratorType)CodemodMagicArgsCodemodCommand)AddImportsVisitorc@sNeZdZUdZeed<edejddfddZ de e e ddffdd Z dS) EnsureImportPresentCommandzjGiven a module and possibly an entity in that module, add an import as long as one does not already exist. DESCRIPTION arg_parserreturnNcCsF|jddddtdd|jddd d tdd |jd d ddtdd dS)Nz--modulemoduleMODULEzModule that should be imported.T)destmetavarhelptyperequiredz--entityentityENTITYz[Entity that should be imported from module. If left empty, entire module will be imported.)r rrrdefaultz--aliasaliasALIASzcAlias that will be used for the imported module or entity. If left empty, no alias will be applied.) add_argumentstr)r rf/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/commands/ensure_import_present.pyadd_argss0  z#EnsureImportPresentCommand.add_argsccs6t|j|jjd|jjd|jjdtVdS)Nr rr)radd_needed_importcontextscratch)selfrrrget_transforms4s    z)EnsureImportPresentCommand.get_transforms)__name__ __module__ __qualname__rr__annotations__ staticmethodargparseArgumentParserrrrrr rrrrr s   r) r&typingrrlibcst.codemodrrlibcst.codemod.visitorsrrrrrrs