Minor audit cleanups in dython transpiler and runtime

Author: RainingComputers

.vscode/settings.json

@@ -3,6 +3,7 @@
     "alecthomas",
     "allocid",
     "amacneil",
+    "asyncio",
     "buildmode",
     "BYTEA",
     "chancreate",
@@ -11,6 +12,7 @@
     "cloc",
     "clocincludetests",
     "condn",
+    "coro",
     "dataclass",
     "dbconn",
     "dbmate",
@@ -112,6 +114,7 @@
     "XDECREF"
   ],
   "python.defaultInterpreterPath": "./.venv/bin/python",
+  "mypy-type-checker.interpreter": ["./.venv/bin/python"],
   "python.terminal.activateEnvironment": false,
   "[json]": {
     "editor.defaultFormatter": "esbenp.prettier-vscode"

pkg/runtimes/dython/deepslate/__init__.py

@@ -1,5 +1,5 @@
 from deepslate.imports import pure, impure, block
-from deepslate.transpile import transpile_flow, flow
+from deepslate.transpile import flow
 from deepslate.worker import spawn_workers, spawn_inspector, new_postgres
 from deepslate.errors import DSError
 from deepslate import dpslio
@@ -20,7 +20,6 @@
     "sleep",
     "json_marshal",
     "json_unmarshal",
-    "transpile_flow",
     "flow",
     "pure",
     "impure",

pkg/runtimes/dython/deepslate/blocks.py

@@ -30,16 +30,16 @@
 TriggerFunc = Callable[[Context, EventNotify], DSError | None]
 
 
-def get_return(ctx: Context, idx: int, tparam: type = object) -> Any:
-    return ctx.get_var(f"__return_{idx}__", tparam)
+class ChanRef[T](str):
+    pass
 
 
-class ChanRef[T](str):
+class WgRef(str):
     pass
 
 
-WgRef = str
-MutexRef = str
+class MutexRef(str):
+    pass
 
 
 @dataclass
@@ -172,14 +172,14 @@ def impure_block(
     line: int,
     stmt: Callable[[Context], None],
 ) -> AwaitBlock:
-    def _trigger(ctx: Context, notify: EventNotify) -> None:
-        stmt(ctx)
-
     def _start(ctx: Context) -> Event:
         key = "impure-" + ctx.gen_key()
 
         return CheckpointEvent(id=key)
 
+    def _trigger(ctx: Context, notify: EventNotify) -> None:
+        stmt(ctx)
+
     return await_block(line, _start, _trigger)
 
 
@@ -258,19 +258,23 @@ def _trigger(ctx: Context, notify: EventNotify) -> None:
     return await_block(line, _start, _trigger)
 
 
+def get_return(ctx: Context, idx: int, tparam: type = object) -> Any:
+    return ctx.get_var(f"__return_{idx}__", tparam)
+
+
 def recv_any(
     line: int,
     count: int,
     ids: Callable[[Context], list[str]],
-    ref: Callable[[Context, bytes, int], None],
+    ret: Callable[[Context, bytes, int], tuple[None, None]],
 ) -> AwaitBlock:
     def _make_future(idx: int) -> Future:
         def _start(ctx: Context) -> Event:
             return ChannelRecvEvent(id=ids(ctx)[idx])
 
         def _trigger(ctx: Context, notify: EventNotify) -> None:
             if hasattr(notify, "value"):
-                ref(ctx, notify.value, idx)
+                ret(ctx, notify.value, idx)
 
         return Future(start=_start, trigger=_trigger)
 

pkg/runtimes/dython/deepslate/context.py

@@ -11,7 +11,7 @@ class Context:
     vars: dict[str, Any] = field(default_factory=dict)
     vars_ser: dict[str, bytes] = field(default_factory=dict)
 
-    def set_var(self, name: str, value: Any) -> None:
+    def set_var(self, name: str, value: Any) -> Any:
         self.vars[name] = value
 
     def get_var(self, name: str, tparam: type = object) -> Any:

pkg/runtimes/dython/deepslate/imports.py

@@ -23,7 +23,7 @@ def __init__(
     ) -> None:
         self._func = func
         self.metadata = metadata
-        self.impl: Callable[..., Any] = impl if impl is not None else func
+        self.impl = impl
 
         self.__name__ = func.__name__
         self.__type_params__ = getattr(func, "__type_params__", ())
@@ -61,10 +61,9 @@ def decorator(stub_func: Callable[P, R]) -> DeepslateAnnotated[P, R]:
     return decorator
 
 
-def mark_program(func: Callable, impl: Callable | None = None) -> DeepslateAnnotated:
+def mark_program(func: Callable, impl: Callable) -> DeepslateAnnotated:
     meta = _non_block_func_metadata(func, FuncType.PROGRAM, allow_sync_primitives=True)
-    actual = impl if impl is not None else func
-    return DeepslateAnnotated(actual, meta, actual)
+    return DeepslateAnnotated(impl, meta, impl)
 
 
 def _mark_pure[**P, R](func: Callable[P, R]) -> DeepslateAnnotated[P, R]:
@@ -108,8 +107,8 @@ def _non_block_func_metadata(
 def _block_func_metadata(impl_func: Callable, stub_func: Callable) -> FuncMetadata:
     impl_name = impl_func.__name__
     stub_name = stub_func.__name__
-    stub_sig = inspect.signature(stub_func)
     impl_sig = inspect.signature(impl_func)
+    stub_sig = inspect.signature(stub_func)
 
     if _has_decorator(impl_func):
         return FuncMetadata(