Fix parsing for if statement

dlorch · May 5, 2016 · 75c01e7 · 75c01e7
1 parent ec1d74e
commit 75c01e7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/pycep/parser.py b/pycep/parser.py
@@ -833,8 +833,11 @@ def _if_stmt(tokens):
     result.append(tokens.accept(token.OP, ":", result_token=token.COLON))
     result.append(_suite(tokens))
 
-    if tokens.check(token.NAME, "elif"):
-        raise NotImplementedError
+    while tokens.check(token.NAME, "elif"):
+        result.append(tokens.accept(token.NAME, "elif"))
+        result.append(_test(tokens))
+        result.append(tokens.accept(token.OP, ":", result_token=token.COLON))
+        result.append(_suite(tokens))
 
     if tokens.check(token.NAME, "else"):
         result.append(tokens.accept(token.NAME, "else"))

diff --git a/pycep/tests/snippets/pos/astrein.py b/pycep/tests/snippets/pos/astrein.py
@@ -16,6 +16,16 @@ class decorated_class():
         continue
     print "Found a number", num
 
+if x < 0:
+    x = 0
+    print 'Negative changed to zero'
+elif x == 0:
+    print 'Zero'
+elif x == 1:
+    print 'Single'
+else:
+    print 'More'
+
 for x in [1, 2, 3]:
     pass
 del x