fix(AtomParser): add new production rule to support loops

aadhithya · May 16, 2022 · ab91aad · ab91aad
1 parent 7c951eb
commit ab91aad
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rajinipp/parser/atom_parser.py b/rajinipp/parser/atom_parser.py
@@ -7,6 +7,7 @@ class AtomParser:
     """parse all atomic components."""
 
     def parse(self, pg):
+        @pg.production("forvar : NUMBER")
         @pg.production("expression : NUMBER")
         def number_expr(p):
             logger.debug("Parser --> number")
@@ -22,6 +23,7 @@ def string_expr(p):
             logger.debug("Parser --> string")
             return String(p[0].value)
 
+        @pg.production("forvar : WORD")
         @pg.production("expression : WORD")
         @pg.production("variable : WORD")
         def word_expr(p):