U f@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 CsZt|}||jkrdSd}i}g}|j|D]}t|jtjr||j}t|tjr||krt ||||stt | ddd|| ddd<| |n,||jj t|d|j_ |jjs|||<d}q.t|jtjr.|j|kr.t ||j||st | ddd|| ddd<| |jd}q.t|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 isinstancenodeastaliasparent ImportFrom_rename_name_in_importfromAssertionErrorrsplitappendnamelenZasnamesix iteritems _rename_reads) told_namenew_namesc has_changedrenamesZalready_changedrefrZ rename_oldZ rename_newr!;/tmp/pip-target-lpfmz8o1/lib/python/pasta/augment/rename.pyrename_externals4        r#c Cs||kr dS|jd}|d}|d}|dt||kr`d||t|d|_dS|jD]}|j|dkrfqqfdS|d|_||ddkrt|jdkrt|||}d|dd|_nd|dd|_dS)NFrTr r )modulesplitrjoinnamesrrZ split_import) rr rr module_partsZ old_partsZ new_partsZalias_to_changeZ new_importr!r!r"r^s&     rc Cs|d}z.|j|d}|ddD]}|j|}q&Wntk rNYdSXd}|jD]<}t|tjtjfrZt | ||t |j djd}qZ|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'attrsKeyErrorZreadsr r Name AttributerZ replace_childrparsebodyvalue) rrrrZ name_partsrpartrZref_noder!r!r"rs  r)__doc__ __future__rrrr rZ pasta.augmentrZ pasta.baserrr#rrr!r!r!r"s      A!