Added support for numbers.

1 files changed, 10 insertions, 0 deletions

diff --git a/bparser.py b/bparser.py
index ce01011..1611988 100755
--- a/bparser.py
+++ b/bparser.py
@@ -39,6 +39,7 @@ tokens = ['LPAREN',
           'COMMA',
           'DOT',
           'STRING',
+          'INT',
           'EOL_CONTINUE',
           'EOL',
           ] + list(reserved.values())
@@ -68,6 +69,11 @@ def t_STRING(t):
     t.value = t.value[1:-1]
     return t
 
+def t_INT(t):
+    '[0-9]+'
+    t.value = int(t.value)
+    return t
+
 def t_EOL(t):
     r'\n'
     t.lexer.lineno += 1
@@ -103,6 +109,10 @@ def p_expression_atom(t):
     'expression : ATOM'
     t[0] = nodes.AtomExpression(t[1], t.lineno(1))
 
+def p_expression_int(t):
+    'expression : INT'
+    t[0] = nodes.IntExpression(t[1], t.lineno(1))
+
 def p_expression_bool(t):
     '''expression : TRUE
                   | FALSE'''