Fix check for python 3.12

10sr · Oct 18, 2023 · 5a37b4c · 5a37b4c
1 parent 7de98d1
commit 5a37b4c
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/flake8_no_implicit_concat/__init__.py b/flake8_no_implicit_concat/__init__.py
@@ -29,6 +29,27 @@
 }
 
 
+def _is_ok_lt_py312(a: tokenize.TokenInfo, b: tokenize.TokenInfo) -> bool:
+    return not a.type == b.type == tokenize.STRING
+
+
+def _is_ok(a: tokenize.TokenInfo, b: tokenize.TokenInfo) -> bool:
+    if a.type == b.type == tokenize.STRING:
+        return False
+    if a.type == tokenize.STRING and b.type == tokenize.FSTRING_START:  # type: ignore[attr-defined]
+        return False
+    if a.type == tokenize.FSTRING_END and b.type == tokenize.STRING:  # type: ignore[attr-defined]
+        return False
+    if a.type == tokenize.FSTRING_END and b.type == tokenize.FSTRING_START:  # type: ignore[attr-defined]
+        return False
+    return True
+
+
+if sys.version_info < (3, 12):  # pragma: no cover
+    # E811 redefinition of unused '_is_ok' from line 36
+    _is_ok = _is_ok_lt_py312  # noqa
+
+
 def _check(tokens: Iterable[tokenize.TokenInfo]) -> Iterable[_ERROR]:
     tokens_wo_ws = (
         e
@@ -43,7 +64,7 @@ def _check(tokens: Iterable[tokenize.TokenInfo]) -> Iterable[_ERROR]:
         )
     )
     for a, b in pairwise(tokens_wo_ws):
-        if not (a.type == b.type == tokenize.STRING):
+        if _is_ok(a, b):
             continue
 
         on_one_line = a.end[0] == b.start[0]