o o^@sxdZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z d d Z d d Z d dZdS)z Rename names in a python module.)absolute_import)division)print_functionN) import_utils) ast_utils)scopec CsTt|}||jvr dSd}i}g}|j|D]}t|jtjrh||j}t|tjrO||vrOt ||||s9J| ddd|| ddd<| |n||jj t |d|j_ |jjse|||<d}qt|jtjr|j|vrt ||j||s~J| ddd|| ddd<| |jd}qt|D] \} } t||| | q|S)aRename an imported name in a module. This will rewrite all import statements in `tree` that reference the old module as well as any names in `tree` which reference the imported name. This may introduce new import statements, but only if necessary. For example, to move and rename the module `foo.bar.utils` to `foo.bar_utils`: > rename_external(tree, 'foo.bar.utils', 'foo.bar_utils') - import foo.bar.utils + import foo.bar_utils - from foo.bar import utils + from foo import bar_utils - from foo.bar import logic, utils + from foo.bar import logic + from foo import bar_utils Arguments: t: (ast.Module) Module syntax tree to perform the rename in. This will be updated as a result of this function call with all affected nodes changed and potentially new Import/ImportFrom nodes added. old_name: (string) Fully-qualified path of the name to replace. new_name: (string) Fully-qualified path of the name to update to. Returns: True if any changes were made, False otherwise. F.NT)rZanalyzeZexternal_references isinstancenodeastaliasparentZ ImportFrom_rename_name_in_importfromrsplitappendnamelenasnamesixZ iteritems _rename_reads) told_namenew_namesc has_changedrenamesZalready_changedrefrZ rename_oldZ rename_newr4lib/python3.10/site-packages/pasta/augment/rename.pyrename_externals6        r!c Cs||krdS|jd}|d}|d}|dt||kr0d||t|d|_dS|jD] }|j|dkr>nq3dS|d|_||ddkrrt|jdkrht|||}d|dd|_dSd|dd|_dS)NFrTr r )modulesplitrjoinnamesrrZ split_import) rr rrZ module_partsZ old_partsZ new_partsZalias_to_changeZ new_importrrr r^s*     rc Cs|d}z|j|d}|ddD]}|j|}qWn ty&YdSwd}|jD]}t|tjtjfrJt | ||t |j djd}q,|S)apUpdates all locations in the module where the given name is read. Arguments: sc: (scope.Scope) Scope to work in. This should be the scope of `t`. t: (ast.AST) The AST to perform updates in. old_name: (string) Dotted name to update. new_name: (string) Dotted name to replace it with. Returns: True if any changes were made, False otherwise. rrr NFT)r#r%ZattrsKeyErrorZreadsr r NameZ AttributerZ replace_childrparsebodyvalue) rrrrZ name_partsrpartrZref_noderrr rs$    r)__doc__Z __future__rrrr rZ pasta.augmentrZ pasta.baserrr!rrrrrr s      A !