o DDiN@s,ddlmZddlmZGdddeZdS)) CodemodTest) RenameCommandc@seZdZeZdcddZdcddZdcddZdcd d Zdcd d Z dcd dZ dcddZ dcddZ dcddZ dcddZdcddZdcddZdcddZdcddZdcdd Zdcd!d"Zdcd#d$Zdcd%d&Zdcd'd(Zdcd)d*Zdcd+d,Zdcd-d.Zdcd/d0Zdcd1d2Zdcd3d4Zdcd5d6Zdcd7d8Zdcd9d:Z dcd;d<Z!dcd=d>Z"dcd?d@Z#dcdAdBZ$dcdCdDZ%dcdEdFZ&dcdGdHZ'dcdIdJZ(dcdKdLZ)dcdMdNZ*dcdOdPZ+dcdQdRZ,dcdSdTZ-dcdUdVZ.dcdWdXZ/dcdYdZZ0dcd[d\Z1dcd]d^Z2dcd_d`Z3dcdadbZ4dS)dTestRenameCommandreturnNcCd}d}|j||ddddS)Nza from foo import bar def test() -> None: bar(5) a from baz import qux def test() -> None: qux(5) foo.barbaz.quxold_namenew_name assertCodemodselfbeforeafterrb/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/commands/tests/test_rename.pytest_rename_namesz"TestRenameCommand.test_rename_namecCr)NzK from typing import List x: List[int] = [] z' x: list[int] = [] z typing.Listz builtins.listr r rrrrtest_rename_to_builtins  z(TestRenameCommand.test_rename_to_builtincCr)Nzh from foo import bar as bla def test() -> None: bla(5) rrr r r rrrrtest_rename_name_asname, z)TestRenameCommand.test_rename_name_asnamecCr)Nzl from foo import foo as bla def test() -> None: bla.bla(5) ze from baz import qux def test() -> None: qux.bla(5) foo.foor r r rrrr%test_rename_repeated_name_with_asnameA z7TestRenameCommand.test_rename_repeated_name_with_asnamecCr)NzZ import a.b def test() -> None: a.b.c(5) zZ import d.e def test() -> None: d.e.f(5) a.b.czd.e.fr r rrrrtest_rename_attrUrz"TestRenameCommand.test_rename_attrcCr)Nzc import foo as bar def test() -> None: bar.qux(5) z] import baz def test() -> None: baz.quux(5) foo.quxbaz.quuxr r rrrrtest_rename_attr_asnamejrz)TestRenameCommand.test_rename_attr_asnamecCr)Nze import foo.qux as bar def test() -> None: bar.z(5) zd import baz.quux def test() -> None: baz.quux.z(5) rrr r rrrrtest_rename_attr_asname_2rz+TestRenameCommand.test_rename_attr_asname_2cCr)NT import a.b class Foo(a.b.C): pass T import c.b class Foo(c.b.C): pass a.bc.br r rrrrtest_rename_module_importrz+TestRenameCommand.test_rename_module_importcCr)Nr"r#acr r rrrrtest_rename_module_import_2rz-TestRenameCommand.test_rename_module_import_2cCd}|j||ddddS)Nr"br%r r rrrrr#test_rename_module_import_no_changes z5TestRenameCommand.test_rename_module_import_no_changecCr)NW from a import b class Foo(b.C): pass W from c import b class Foo(b.C): pass r$r%r r rrrrtest_rename_module_import_fromrz0TestRenameCommand.test_rename_module_import_fromcCr)Nr.r/r'r(r r rrrr test_rename_module_import_from_2rz2TestRenameCommand.test_rename_module_import_from_2cCr)Nzi from a.b import some_class class Foo(some_class): pass zi from c.b import some_class class Foo(some_class): pass a.b.some_classzc.b.some_classr r rrrrtest_rename_classrz#TestRenameCommand.test_rename_classcCr)Nzl from a.b import Class_1, Class_2 class Foo(Class_1): pass zl from a.b import Class_3, Class_2 class Foo(Class_3): pass z a.b.Class_1z a.b.Class_3r r rrrr"test_rename_importfrom_same_module rz4TestRenameCommand.test_rename_importfrom_same_modulecCr)Nz from a.b import module_1, module_2 class Foo(module_1.Class_1): pass class Fooo(module_2.Class_2): pass z from a.b import module_2 from a.b.module_3 import Class_3 class Foo(Class_3): pass class Fooo(module_2.Class_2): pass za.b.module_1.Class_1za.b.module_3.Class_3r r rrrr$test_rename_importfrom_same_module_2s  z6TestRenameCommand.test_rename_importfrom_same_module_2cCr)Nz@ import logging logging.warn(1) zC import logging logging.warning(1) z logging.warnzlogging.warningr r rrrrtest_import_same_module8 z)TestRenameCommand.test_import_same_modulecCr)Nz8 import a.b a.b.warn(1) z; import a.b a.b.warning(1) za.b.warnz a.b.warningr r rrrrtest_import_same_dotted_moduleHr7z0TestRenameCommand.test_import_same_dotted_modulecCr)Nz1 x = 5 y = 5 + x z1 z = 5 y = 5 + z xzr r rrrrtest_rename_local_variableXs z,TestRenameCommand.test_rename_local_variablecCr)NzU from a import b class Foo(b): pass zU from a import c class Foo(c): pass r$za.cr r rrrrtest_module_does_not_changeiz-TestRenameCommand.test_module_does_not_changecCr)Nw import a, b, c class Foo(a.z): bar: b.bar baz: c.baz zw import d, b, c class Foo(d.z): bar: b.bar baz: c.baz za.zzd.zr r rrrrtest_other_imports_untouchedx z.TestRenameCommand.test_other_imports_untouchedcCr)Nr>zw import a, b, d class Foo(a.z): bar: b.bar baz: d.baz zc.bazzd.bazr r rrrrtest_comma_importr@z#TestRenameCommand.test_comma_importcCr)N| from a import b, c, d class Foo(b): bar: c.bar baz: d.baz z from a import c, d from f import b class Foo(b): bar: c.bar baz: d.baz r$zf.br r rrrr!test_other_import_froms_untouched z3TestRenameCommand.test_other_import_froms_untouchedcCr)NrBz from a import b, c from f import d class Foo(b): bar: c.bar baz: d.baz a.df.dr r rrrrtest_comma_import_fromrDz(TestRenameCommand.test_comma_import_fromcCr)Nz from a import ( b, c, d, ) from x import (y,) class Foo(b): bar: c.bar baz: d.baz z from a import ( b, c, ) from x import (y,) from f import d class Foo(b): bar: c.bar baz: d.baz rErFr r rrrrtest_comma_import_from_parenss   z/TestRenameCommand.test_comma_import_from_parenscCr)Nz import a class Foo(a.b): pass class Foo2(a.c): pass z import a, z class Foo(z.b): pass class Foo2(a.c): pass r$zz.br r rrrr test_no_removal_of_import_in_use z2TestRenameCommand.test_no_removal_of_import_in_usecCr)Nz import a.b class Foo(a.b.c): pass class Foo2(a.b.d): pass z import a.b, z.b class Foo(z.b.c): pass class Foo2(a.b.d): pass rzz.b.cr r rrrr'test_no_removal_of_dotted_import_in_use rJz9TestRenameCommand.test_no_removal_of_dotted_import_in_usecCr)Nzs from a import b class Foo(b.some_class): bar: b.some_other_class z from a import b from blah import some_class class Foo(some_class): bar: b.some_other_class r2zblah.some_classr r rrrr%test_no_removal_of_import_from_in_use"rz7TestRenameCommand.test_no_removal_of_import_from_in_usecCr)Nzg import a import b class Foo(a.obj): pass zg import c import b class Foo(c.obj): pass za.objzc.objr r rrrr#test_other_unused_imports_untouched7r@z5TestRenameCommand.test_other_unused_imports_untouchedcCr)N] from a.b.c import d class Foo(d.e.f): pass Y from g.h.i import j class Foo(j): pass a.b.c.d.e.fg.h.i.jr r rrrrtest_complex_module_renameMr=z,TestRenameCommand.test_complex_module_renamecCr)Nzf from a.b.c import d as ddd class Foo(ddd.e.f): pass rOrPrQr r rrrr&test_complex_module_rename_with_asname\r=z8TestRenameCommand.test_complex_module_rename_with_asnamecCr)Nz` from aa import aaaa class Foo(aaaa.Bar): pass zY from b import c class Foo(c.Bar): pass zaa.aaaazb.cr r rrrr#test_names_with_repeated_substringskrz5TestRenameCommand.test_names_with_repeated_substringscCr)NzX from foo import foo def bar(): foo(5) zX from qux import qux def bar(): qux(5) rzqux.quxr r rrrrtest_repeated_namerz$TestRenameCommand.test_repeated_namecCr*)NzX from foo import bar def baz(): bar(5) barquxr r r,rrrtest_no_codemods z!TestRenameCommand.test_no_codemodcCr)Nz$ import a.b.c.d z$ import x.y.c.d r$x.yr r rrrrtest_rename_import_prefix z+TestRenameCommand.test_rename_import_prefixcCr)Nz- from a.b.c.d import foo z- from x.y.c.d import foo r$rYr r rrrrtest_rename_import_from_prefixr[z0TestRenameCommand.test_rename_import_from_prefixcCr)Nz from a import b class Foo(b.some_class): pass class Foobar(b.some_class): pass z from c.d import some_class class Foo(some_class): pass class Foobar(some_class): pass r2zc.d.some_classr r rrrr test_rename_multiple_occurrencess  z2TestRenameCommand.test_rename_multiple_occurrencescCr)Nz import a from a import b from a.c import d class Foo(d): pass class Fooo(b.some_class): pass class Foooo(a.some_class): pass z import z from z import b from z.c import d class Foo(d): pass class Fooo(b.some_class): pass class Foooo(z.some_class): pass r'r:r r rrrrtest_rename_multiple_importss  z.TestRenameCommand.test_rename_multiple_importscCr)NrNzY from g.h import i class Foo(i.j): pass rPzg.h:i.jr r rrrrtest_input_with_colon_sepr=z+TestRenameCommand.test_input_with_colon_sepcCr)N[ from a.b.c import d class Foo(d.e): pass z\ import g.h.i.j class Foo(g.h.i.j.e): pass a.b.c.dzg.h.i.j:r r rrrr$test_input_with_colon_sep_at_the_endr=z6TestRenameCommand.test_input_with_colon_sep_at_the_endcCr)Nr`z[ from g.h.i import j class Foo(j.e): pass raz:g.h.i.jr r rrrr&test_input_with_colon_sep_at_the_front sz8TestRenameCommand.test_input_with_colon_sep_at_the_frontcCr*)Nz= from foo import bar bar(42) zbaz.barzqux.barr r r,rrr)test_no_change_because_no_match_was_foundsz;TestRenameCommand.test_no_change_because_no_match_was_foundcCr)NzA from a.b import qux print(qux) z? from a import b print(b.qux) za.b.quxza:b.quxr r rrrrtest_rename_single_with_colon"s z/TestRenameCommand.test_rename_single_with_coloncCr)N9 import a a.b.c(a.b.c.d) z9 from z import c c(c.d) rz.cr r rrrrtest_import_parent_module4z+TestRenameCommand.test_import_parent_modulecCr)Nz? import a.b a.b.c.d(a.b.c.d.x) zE from z import c c(c.x) rargr r rrrrtest_import_parent_module_2@riz-TestRenameCommand.test_import_parent_module_2cCr)Nrfz8 import z.c z.c(z.c.d) rzz.c:r r rrrrtest_import_parent_module_3Lriz-TestRenameCommand.test_import_parent_module_3cCr)NzH import a.b as alias alias.c(alias.c.d) z5 import z z.c(z.c.d) rrgr r rrrr test_import_parent_module_asnameXz2TestRenameCommand.test_import_parent_module_asnamecCr)Nz6 import foo foo.baz() @ import quux.foo quux.foo.baz() fooquux.foor r rrrrtest_push_down_toplevel_namescrmz/TestRenameCommand.test_push_down_toplevel_namescCr)Nz= import foo as bar bar.baz() rnrorpr r rrrr)test_push_down_toplevel_names_with_asnamenrmz;TestRenameCommand.test_push_down_toplevel_names_with_asname)rN)5__name__ __module__ __qualname__r TRANSFORMrrrrrr r!r&r)r-r0r1r3r4r5r6r8r;r<r?rArCrGrHrIrKrLrMrRrSrTrUrXrZr\r]r^r_rbrcrdrerhrjrkrlrqrrrrrrr sd                                           rN)libcst.codemodrlibcst.codemod.commands.renamerrrrrrs