very initial attempt at native compiled code; see #71

Author: craigbarratt

custom_components/pyscript/eval.py

@@ -972,6 +972,22 @@ async def ast_classdef(self, arg):
 
     async def ast_functiondef(self, arg):
         """Evaluate function definition."""
+        for dec in arg.decorator_list:
+            if isinstance(dec, ast.Name):
+                if dec.id != "pyscript_compile":
+                    continue
+                arg.decorator_list = []
+                local_var = None
+                if arg.name in self.sym_table and isinstance(self.sym_table[arg.name], EvalLocalVar):
+                    local_var = self.sym_table[arg.name]
+                code = compile(ast.Module(body=[arg], type_ignores=[]), filename=self.filename, mode="exec")
+                exec(code, self.global_sym_table, self.sym_table)  # pylint: disable=exec-used
+                if local_var:
+                    func = self.sym_table[arg.name]
+                    self.sym_table[arg.name] = local_var
+                    self.sym_table[arg.name].set(func)
+                return
+
         func = EvalFunc(arg, self.code_list, self.code_str, self.global_ctx)
         await func.eval_defaults(self)
         await func.eval_decorators(self)
@@ -1374,7 +1390,7 @@ async def ast_name(self, arg):
             # a two-dot name for state.attr needs to exist
             #
             if num_dots == 1 or (num_dots == 2 and State.exist(arg.id)):
-                return await State.get(arg.id)
+                return State.get(arg.id)
             #
             # Couldn't find it, so return just the name wrapped in EvalName to
             # distinguish from a string variable value.  This is to support

custom_components/pyscript/state.py

@@ -249,7 +249,7 @@ def exist(cls, var_name):
         return False
 
     @classmethod
-    async def get(cls, var_name):
+    def get(cls, var_name):
         """Get a state variable value or attribute from hass."""
         parts = var_name.split(".")
         if len(parts) != 2 and len(parts) != 3:

tests/test_state.py

@@ -21,7 +21,7 @@ async def test_service_call(hass):
         State.init(hass)
         await State.get_service_params()
 
-        func = await State.get("test.entity.test")
+        func = State.get("test.entity.test")
         await func(context=Context(id="test"), blocking=True, limit=1, other_service_data="test")
         assert call.called
         assert call.call_args[0] == (
@@ -32,7 +32,7 @@ async def test_service_call(hass):
         assert call.call_args[1] == {"context": Context(id="test"), "blocking": True, "limit": 1}
         call.reset_mock()
 
-        func = await State.get("test.entity.test")
+        func = State.get("test.entity.test")
         await func(context=Context(id="test"), blocking=False, other_service_data="test")
         assert call.called
         assert call.call_args[0] == (