#284 Fix: attribute as import refactor issue

hakancelikdev · Mar 30, 2023 · e6d156a · e6d156a
1 parent 1a36aa8
commit e6d156a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/unimport/refactor.py b/src/unimport/refactor.py
@@ -64,7 +64,10 @@ def leave_import_alike(
         # already handled by leave_ImportFrom
         for column, import_alias in enumerate(names):
             if isinstance(import_alias.name, cst.Attribute):
-                import_name = self.get_import_name_from_attr(attr_node=import_alias.name)
+                if import_alias.asname:
+                    import_name = import_alias.asname.name.value
+                else:
+                    import_name = self.get_import_name_from_attr(attr_node=import_alias.name)
             else:
                 raw_import = import_alias.asname or import_alias
                 raw_import_name = cst.ensure_type(raw_import.name, cst.Name)