ideas [skip-ci]

py2wasm/python.py

@@ -16,32 +16,47 @@ class Visitor(ast.NodeVisitor):
     #         ))
 
     def visit_FunctionDef(self, node):
+        is_export = False
+
         if node.decorator_list:
             # TODO: Support normal decorators
             assert 1 == len(node.decorator_list)
+
             call = node.decorator_list[0]
-            assert 'external' == call.func.id
+            if not isinstance(call, ast.Name):
-            assert 1 == len(call.args)
+                assert isinstance(call, ast.Call)
-            assert isinstance(call.args[0].value, str)
 
-            import_ = Import(
+                assert 'external' == call.func.id
-                call.args[0].value,
+                assert 1 == len(call.args)
-                node.name,
+                assert isinstance(call.args[0].value, str)
-                node.name,
-            )
 
-            import_.params = [
+                import_ = Import(
-                arg.annotation.id
+                    call.args[0].value,
-                for arg in node.args.args
+                    node.name,
-            ]
+                    node.name,
+                )
 
-            self.module.imports.append(import_)
+                import_.params = [
-            return
+                    arg.annotation.id
+                    for arg in node.args.args
+                ]
+
+                self.module.imports.append(import_)
+                return
+
+            assert call.id == 'export'
+            is_export = True
 
         func = Function(
             node.name,
+            is_export,
         )
 
+        for arg in node.args.args:
+            func.params.append(
+                
+            )
+
         self._stack.append(func)
         self.generic_visit(node)
         self._stack.pop()

py2wasm/wasm.py

@@ -25,6 +25,7 @@ class Function:
     def __init__(self, name, exported=True):
         self.name = name
         self.exported = exported # TODO: Use __all__!
+        self.params = []
         self.statements = []
 
     def generate(self):
@@ -39,7 +40,7 @@ class Module:
         self.functions = functions or []
 
     def generate(self):
-        return '(module\n  {}\n  {})'.format(
+        return '(module\n  {}\n  {})\n'.format(
             '\n  '.join(x.generate() for x in self.imports),
             '\n  '.join(x.generate() for x in self.functions),
         )

tests/integration/test_fib.py

@@ -75,8 +75,12 @@ def test_fib():
 def logInt32(value: i32) -> None:
     ...
 
+def helper(arg: i32) -> i32:
+    return arg + 1
+
+@export
 def testEntry():
-    logInt32(13 + 13 * 123)
+    logInt32(13 + 13 * helper(122))
 """
 
     result = Suite(code_py, 'test_fib').run_code()