fix: future import insertion error opening quotes on a line

frostming · Dec 2, 2022 · f355b95 · f355b95
1 parent b39b431
commit f355b95
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/fix_future_annotations/_main.py b/fix_future_annotations/_main.py
@@ -51,7 +51,11 @@ def _add_future_annotations(content: str) -> str:
 
         if in_doc:
             code = line.split("#")[0].rstrip()
-            if code.endswith(doc_quote) and not _escaped(code[: -len(doc_quote)]):
+            if (
+                code != doc_quote
+                and code.endswith(doc_quote)
+                and not _escaped(code[: -len(doc_quote)])
+            ):
                 in_doc = False
     first_code = lines[i].lstrip()
     if not first_code.startswith("from __future__ import") and i == insert_pos: