Fixed bug where the end of the string was not matched

ECN · Jun 28, 2021 · 5738191 · 5738191
1 parent 7ee915f
commit 5738191
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/src/webqueue2api/parser/parser.py b/src/webqueue2api/parser/parser.py
@@ -214,10 +214,9 @@ def error_handler(original_string, match_start_index, tokens):
     (pp.Literal(action_end_delimiter) + pp.LineEnd()).suppress() +
     pp.Group(
         pp.SkipTo(
-            pp.Literal(reply_from_user_start_delimiter) 
+            pp.Literal(reply_from_user_start_delimiter)
             | pp.Literal(action_start_delimiter)
-            | pp.StringEnd()
-        )
+        ) | pp.SkipTo(pp.StringEnd(), include=True)
     ).setResultsName("content") 
 ).leaveWhitespace().setParseAction(parse_section_by_type("reply_to_user"))
 
@@ -231,8 +230,7 @@ def error_handler(original_string, match_start_index, tokens):
         pp.SkipTo(
             pp.Literal(reply_from_user_start_delimiter)
             | pp.Literal(action_start_delimiter)
-            | pp.stringEnd()
-        )
+        ) | pp.SkipTo(pp.StringEnd(), include=True)
     ).setResultsName("content") 
 ).leaveWhitespace().setParseAction(parse_section_by_type("edit"))
 
@@ -244,10 +242,9 @@ def error_handler(original_string, match_start_index, tokens):
     (pp.Literal(action_end_delimiter) + pp.LineEnd()).suppress() +
     pp.Group(
         pp.SkipTo(
-            pp.Literal(reply_from_user_start_delimiter) 
+            pp.Literal(reply_from_user_start_delimiter)
             | pp.Literal(action_start_delimiter)
-            | pp.StringEnd()
-        ) 
+        ) | pp.SkipTo(pp.StringEnd(), include=True)
     ).setResultsName("content") 
 ).leaveWhitespace().setParseAction(parse_section_by_type("status"))