Added check to ignore __init__.py files

pixee · Oct 26, 2023 · ace5ea6 · ace5ea6
1 parent aa6d0f5
commit ace5ea6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/core_codemods/remove_unused_imports.py b/src/core_codemods/remove_unused_imports.py
@@ -45,6 +45,9 @@ def __init__(self, codemod_context: CodemodContext, *codemod_args):
         BaseCodemod.__init__(self, *codemod_args)
 
     def transform_module_impl(self, tree: cst.Module) -> cst.Module:
+        # Do nothing in __init__.py files
+        if self.file_context.file_path.name == "__init__.py":
+            return tree
         gather_unused_visitor = GatherUnusedImportsVisitor(self.context)
         tree.visit(gather_unused_visitor)
         # filter the gathered imports by line excludes/includes

diff --git a/tests/codemods/base_codemod_test.py b/tests/codemods/base_codemod_test.py
@@ -23,7 +23,7 @@ def setup_method(self):
         self.file_context = None
 
     def run_and_assert(self, tmpdir, input_code, expected):
-        tmp_file_path = tmpdir / "code.py"
+        tmp_file_path = Path(tmpdir / "code.py")
         self.run_and_assert_filepath(tmpdir, tmp_file_path, input_code, expected)
 
     def run_and_assert_filepath(self, root, file_path, input_code, expected):