Check for broken symlinks before checking file data (#202)

diff --git a/black.py b/black.py
index 913fe8dfefaf496e57373e94d96b272caebf486d..a1131693cb9b2b9ab17b1cd53e283c5169b7b3e1 100644 (file)
--- a/black.py
+++ b/black.py
@@ -2494,7 +2494,7 @@ def gen_python_files_in_dir(path: Path) -> Iterator[Path]:
 
             yield from gen_python_files_in_dir(child)
 
-        elif child.suffix in PYTHON_EXTENSIONS:
+        elif child.is_file() and child.suffix in PYTHON_EXTENSIONS:
             yield child
 
 

diff --git a/tests/test_black.py b/tests/test_black.py
index cf209455e4aceb38b0429a071491287f82a7b342..bc133cab4aca84f69de1a6eab504134cd7ed1065 100644 (file)
--- a/tests/test_black.py
+++ b/tests/test_black.py
@@ -695,6 +695,13 @@ class BlackTestCase(unittest.TestCase):
             result = CliRunner().invoke(black.main, [])
             self.assertEqual(result.exit_code, 0)
 
+    def test_broken_symlink(self) -> None:
+        with cache_dir() as workspace:
+            symlink = workspace / "broken_link.py"
+            symlink.symlink_to("nonexistent.py")
+            result = CliRunner().invoke(black.main, [str(workspace.resolve())])
+            self.assertEqual(result.exit_code, 0)
+
     def test_read_cache_line_lengths(self) -> None:
         with cache_dir() as workspace:
             path = (workspace / "file.py").resolve()