Fix an infinite loop when handling # fmt: on/off.

psf · Jul 6, 2022 · b420d19 · b420d19
1 parent 7af77d1
commit b420d19
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/black/comments.py b/src/black/comments.py
@@ -241,6 +241,7 @@ def generate_ignored_nodes(
                 if contains_fmt_on_at_column(child, leaf.column, preview=preview):
                     return
                 yield child
+            return
         else:
             yield container
             container = container.next_sibling

diff --git a/tests/data/simple_cases/fmtonoff5.py b/tests/data/simple_cases/fmtonoff5.py
@@ -0,0 +1,11 @@
+# Regression test for https://github.com/psf/black/issues/3129.
+setup(
+    entry_points={
+        # fmt: off
+        "console_scripts": [
+            "foo-bar"
+            "=foo.bar.:main",
+        # fmt: on
+        ]
+    },
+)