a ,=ic3@sdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z dd d Z d d ZdddZddZdddZdS)z-Functions for dealing with import statements.)absolute_import)division)print_functionN)errors) ast_utils)scopeTc st||jvr:tddj|Dd}|r:|jjSd}d}fdd}|rd|vr|dd\} || |} |rtfdd|jDd} | r| j | | j p| j St j | gd d }|s|||d } t j| gd }|jt|rdnd || j p| j S) a9Adds an import to the module. This function will try to ensure not to create duplicate imports. If name_to_import is already imported, it will return the existing import. This is true even if asname is set (asname will be ignored, and the existing name will be returned). If the import would create a name that already exists in the scope given by tree, this function will "import as", and append "_x" to the asname where x is the smallest positive integer generating a unique name. Arguments: tree: (ast.Module) Module AST to modify. name_to_import: (string) The absolute name to import. asname: (string) The alias for the import ("import name_to_import as asname") from_import: (boolean) If True, import the name using an ImportFrom node. merge_from_imports: (boolean) If True, merge a newly inserted ImportFrom node into an existing ImportFrom node, if applicable. Returns: The name (as a string) that can be used to reference the imported name. This can be the fully-qualified name, the basename, or an alias name. css|]}|jdur|VqdS)N)name_ref).0refr k/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pasta/augment/import_utils.py 9s zadd_import..NcsHtj||d}|p|}d}|jvrD|d7}d|p4||f}|_q|S)N)nameasnamerz%s_%d)astaliasnamesr) alias_namer new_aliasZ imported_namecounter)scr r make_safe_alias_nodeAs z(add_import..make_safe_alias_node.rc3s2|]*}t|tjr|jkr|jdkr|VqdS)rN) isinstancer ImportFrommodulelevel)r node) from_moduler r r Usr)rrr)rr)r)ranalyzeZexternal_referencesnextridrsplitbodyrappendrrrrImportinsertrZ has_docstring) treeZname_to_importrZ from_importZmerge_from_importsZ existing_ref import_nodeZ added_namerrrZexisting_from_importr )rrr add_imports:       r*cCs||}d}dD]*}t||r|t||vrt||}qPqtd||f||}t|}|g|_|j || |d||S)aSplit an import node by moving the given imported alias into a new import. Arguments: sc: (scope.Scope) Scope computed on whole tree of the code being modified. node: (ast.Import|ast.ImportFrom) An import node to split. alias_to_remove: (ast.alias) The import alias node to remove. This must be a child of the given `node` argument. Raises: errors.InvalidAstError: if `node` is not appropriately contained in the tree represented by the scope `sc`. N)r$orelse finalbodyz:Unable to find list containing import %r on parent node %rr) parenthasattrgetattrrZInvalidAstErrorindexcopydeepcopyrremover')rrZalias_to_remover-Z parent_listaidxZ new_importr r r split_importis     r6cCs|durt|}t}t|D]X}t|tjr"|jdurB|jn|j}||j vrn|j |}|j sz| |q"t d|q"|S)a.Get the import aliases that aren't used. Arguments: tree: (ast.AST) An ast to find imports in. sc: A scope.Scope representing tree (generated from scratch if not provided). Returns: A list of ast.alias representing imported aliases that aren't referenced in the given tree. NzGImported name %s not found in scope (perhaps it's imported dynamically))rr setrwalkrrrrrZreadsaddloggingwarning)r(rZunused_aliasesrZstr_namerr r r get_unused_import_aliasess      r<cCs@||}t|jdkr0||}t||n t||dS)zRemove an alias and if applicable remove their entire import. Arguments: sc: (scope.Scope) Scope computed on whole tree of the code being modified. node: (ast.Import|ast.ImportFrom|ast.alias) The node to remove. rN)r-lenrrZ remove_child)rrr)Z import_parentr r r remove_import_alias_nodes   r>cCs|durt|}d}t}|jD]}t|tjtjfr"t|j D]}| |}t|tjrd|j }n4|j rdd|j |j |j f}ndd|j |j f}|d|jp|j 7}||vrt||d}qB||qBq"|S)a3Remove duplicate imports, where it is safe to do so. This does NOT remove imports that create new aliases Arguments: tree: (ast.AST) An ast to modify imports in. sc: A scope.Scope representing tree (generated from scratch if not provided). Returns: Whether any changes were made. NFz%s%s.%srz%s%s:T)rr r7r$rrr&rlistrr-rrrrr>r9)r(rmodifiedZ seen_namesrrr)Z full_namer r r remove_duplicatess*       rB)NTT)N)N)__doc__ __future__rrrrr1r:Z pasta.augmentrZ pasta.baserrr*r6r<r>rBr r r r s       K