From 63c8e58e4269670d58b0f4fa7c9bda7a62c21949 Mon Sep 17 00:00:00 2001
From: Erez Shinan <erezshin+git@gmail.com>
Date: Wed, 15 Feb 2017 10:00:01 +0200
Subject: [PATCH] Fixed calc example

---
 examples/calc.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/examples/calc.py b/examples/calc.py
index dc936cf..c9da74d 100644
--- a/examples/calc.py
+++ b/examples/calc.py
@@ -4,6 +4,12 @@
 
 from lark import Lark, InlineTransformer
 
+try:
+    input = raw_input
+except NameError:
+    pass
+
+
 calc_grammar = """
     ?start: sum
           | NAME "=" sum    -> assign_var
@@ -47,7 +53,7 @@ calc = calc_parser.parse
 def main():
     while True:
         try:
-            s = raw_input('> ')
+            s = input('> ')
         except EOFError:
             break
         print(calc(s))
@@ -58,6 +64,6 @@ def test():
     print(calc("1+a*-3"))
 
 if __name__ == '__main__':
-    test()
-    # main()
+    # test()
+    main()