fix issue with multiline comments

ricffb · Jun 24, 2021 · 5f9524e · 5f9524e
1 parent c8d2280
commit 5f9524e
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/whilelang/parse.py b/whilelang/parse.py
@@ -5,8 +5,15 @@
 pad = regex(r'[^\S\r\n]')
 
 Newline = regex(r"\n")
-Next = (Newline.at_least(1)
-        | eof).desc("at least one newline or the end of file")
+_Next = (Newline.at_least(1)
+         | eof).desc("at least one newline or the end of file")
+
+
+@generate
+def Next():
+    yield Comment | _Next >> Comment | _Next
+
+
 Comment = string("#") >> (regex(r'[\S]') | pad).many().concat() >> Next
 
 woc = whitespace << Comment.optional() << whitespace