Don't gather dirs ending .py

Instagram · Aug 24, 2023 · 8b62f45 · 8b62f45
1 parent 9eab2f0
commit 8b62f45
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/libcst/codemod/_cli.py b/libcst/codemod/_cli.py
@@ -90,7 +90,10 @@ def gather_files(
             ret.extend(
                 str(p)
                 for p in Path(fd).rglob("*.py*")
-                if str(p).endswith("py") or (include_stubs and str(p).endswith("pyi"))
+                if Path.is_file(p)
+                and (
+                    str(p).endswith("py") or (include_stubs and str(p).endswith("pyi"))
+                )
             )
     return sorted(ret)
 

diff --git a/libcst/tests/test_e2e.py b/libcst/tests/test_e2e.py
@@ -48,6 +48,9 @@ def test_leaky_codemod(self) -> None:
             # File that should not be modified
             other = tmp / "other.py"
             other.touch()
+            # Just a dir named "dir.py", should be ignored
+            adir = tmp / "dir.py"
+            adir.mkdir()
 
             # Run command
             command_instance = PrintToPPrintCommand(CodemodContext())