Fix crash

JelleZijlstra · May 3, 2023 · 2601427 · 2601427
1 parent e9678f4
commit 2601427
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/blib2to3/pgen2/tokenize.py b/src/blib2to3/pgen2/tokenize.py
@@ -599,7 +599,7 @@ def generate_tokens(
                     if token[-1] == "\n":  # continued string
                         strstart = (lnum, start)
                         maybe_endprog = (
-                            endprogs[initial]
+                            endprogs.get(initial)
                             or endprogs.get(token[1])
                             or endprogs.get(token[2])
                         )

diff --git a/tests/data/simple_cases/fstring.py b/tests/data/simple_cases/fstring.py
@@ -7,6 +7,8 @@
 f"\"{f'{nested} inner'}\" outer"
 f"space between opening braces: { {a for a in (1, 2, 3)}}"
 f'Hello \'{tricky + "example"}\''
+f"Tried directories {str(rootdirs)} \
+but none started with prefix {parentdir_prefix}"
 
 # output
 
@@ -19,3 +21,5 @@
 f"\"{f'{nested} inner'}\" outer"
 f"space between opening braces: { {a for a in (1, 2, 3)}}"
 f'Hello \'{tricky + "example"}\''
+f"Tried directories {str(rootdirs)} \
+but none started with prefix {parentdir_prefix}"