Changing message for TokenError

psf · Jul 21, 2021 · fc40194 · fc40194
1 parent 705e027
commit fc40194
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/black/parsing.py b/src/black/parsing.py
@@ -85,18 +85,20 @@ def lib2to3_parse(src_txt: str, target_versions: Iterable[TargetVersion] = ()) -
             result = drv.parse_string(src_txt, True)
             break
 
-        except (TokenError, ParseError) as err:
-            if isinstance(err, ParseError):
-                lineno, column = err.context[1]
-            else:
-                lineno, column = err.args[1]
+        except ParseError as pe:
+            lineno, column = pe.context[1]
             lines = src_txt.splitlines()
             try:
                 faulty_line = lines[lineno - 1]
             except IndexError:
                 faulty_line = "<line number missing in source>"
             exc = InvalidInput(f"Cannot parse: {lineno}:{column}: {faulty_line}")
 
+        except TokenError as te:
+            lineno, column = te.args[1]
+            lines = src_txt.splitlines()
+            exc = InvalidInput(f"Cannot parse: {lineno}:{column}: Unexpected EOF")
+
     else:
         raise exc from None