From 8eda8decea3b7e243dd2f6cc0f68a190b4f6a4d4 Mon Sep 17 00:00:00 2001
From: Dave Halter <davidhalter88@gmail.com>
Date: Sat, 7 Apr 2018 15:31:30 +0200
Subject: [PATCH] Fix whitespace issues with prefixes

---
 parso/python/tokenize.py | 24 +++++++++++++++++++++---
 test/test_fstring.py     | 12 +++++++++++-
 2 files changed, 32 insertions(+), 4 deletions(-)

diff --git a/parso/python/tokenize.py b/parso/python/tokenize.py
index 8d9e7cc..31f081d 100644
--- a/parso/python/tokenize.py
+++ b/parso/python/tokenize.py
@@ -333,6 +333,19 @@ def tokenize(code, version_info, start_pos=(1, 0)):
     return tokenize_lines(lines, version_info, start_pos=start_pos)
 
 
+def _print_tokens(func):
+    """
+    A small helper function to help debug the tokenize_lines function.
+    """
+    def wrapper(*args, **kwargs):
+        for token in func(*args, **kwargs):
+            print(token)
+            yield token
+
+    return wrapper
+
+
+# @_print_tokens
 def tokenize_lines(lines, version_info, start_pos=(1, 0)):
     """
     A heavily modified Python standard library tokenizer.
@@ -394,7 +407,10 @@ def tokenize_lines(lines, version_info, start_pos=(1, 0)):
                 if string:
                     yield PythonToken(
                         FSTRING_STRING, string,
-                        fstring_stack[-1].last_string_start_pos, ''
+                        fstring_stack[-1].last_string_start_pos,
+                        # Never has a prefix because it can start anywhere and
+                        # include whitespace.
+                        prefix=''
                     )
                     fstring_stack[-1].previous_lines = ''
                     continue
@@ -410,8 +426,9 @@ def tokenize_lines(lines, version_info, start_pos=(1, 0)):
                         FSTRING_END,
                         fstring_stack[fstring_index].quote,
                         (lnum, pos),
-                        prefix=''
+                        prefix=additional_prefix,
                     )
+                    additional_prefix = ''
                     del fstring_stack[fstring_index:]
                     pos += end
                     continue
@@ -461,12 +478,13 @@ def tokenize_lines(lines, version_info, start_pos=(1, 0)):
                 if fstring_index is not None:
                     if end != 0:
                         yield PythonToken(ERRORTOKEN, token[:end], spos, prefix)
+                        prefix = ''
 
                     yield PythonToken(
                         FSTRING_END,
                         fstring_stack[fstring_index].quote,
                         (lnum, spos[1] + 1),
-                        prefix=''
+                        prefix=prefix
                     )
                     del fstring_stack[fstring_index:]
                     pos -= len(token) - end
diff --git a/test/test_fstring.py b/test/test_fstring.py
index 59debd9..4c85de8 100644
--- a/test/test_fstring.py
+++ b/test/test_fstring.py
@@ -1,4 +1,5 @@
 import pytest
+from textwrap import dedent
 
 from parso import load_grammar, ParserSyntaxError
 from parso.python.tokenize import tokenize
@@ -75,5 +76,14 @@ def test_invalid(code, grammar):
 )
 def test_tokenize_start_pos(code, positions):
     tokens = list(tokenize(code, version_info=(3, 6)))
-    print(tokens)
     assert positions == [p.start_pos for p in tokens]
+
+
+def test_roundtrip(grammar):
+    code = dedent("""\
+        f'''s{
+           str.uppe
+        '''
+        """)
+    tree = grammar.parse(code)
+    assert tree.get_code() == code