Enable support for custom functions definition

2019-07-12 00:13:20 +02:00
parent af3cb7027a
commit eb28976704
4 changed files with 143 additions and 137 deletions
--- a/smnp/ast/node/program.py
+++ b/smnp/ast/node/program.py
@@ -1,3 +1,4 @@
+from smnp.ast.node.function import FunctionDefinitionParser
 from smnp.ast.node.imports import ImportParser
 from smnp.ast.node.model import Node, ParseResult
 from smnp.ast.node.statement import StatementParser
@@ -15,6 +16,7 @@ def ProgramParser(input):
            result = Parser.oneOf(
                # Start Symbol
                ImportParser,
+                FunctionDefinitionParser,
                StatementParser,
                exception=RuntimeError("Nie znam tego wyrazenia")
            )(input)