Add local storage wrapper

Author: apiad

examples/quickstart.py

@@ -71,7 +71,7 @@ async def handle_change(event: Event):
 
 
 @app.startup
-def init_counter():
+async def init_counter():
     """
     Runs automatically when the page loads (Client-Side).
     Restores the counter from Local Storage.
@@ -82,7 +82,7 @@ def init_counter():
     # Check if we have a saved count
     saved_count = store.count
 
-    if saved_count:
+    if saved_count is not None:
         Document.find("display").text = str(saved_count)
         print(f"Restored count: {saved_count}")
 

violetear/app.py

@@ -61,6 +61,9 @@ def get_bundle():
 
     def client(self, func: Callable):
         """Decorator to mark a function to be compiled to the client."""
+        if not inspect.iscoroutinefunction(func):
+            raise ValueError("func must be async")
+
         self.client_functions[func.__name__] = func
         return func
 
@@ -69,6 +72,9 @@ def startup(self, func: Callable):
         Decorator to mark a function to run automatically when the client loads.
         Also registers it as a client function.
         """
+        if not inspect.iscoroutinefunction(func):
+            raise ValueError("func must be async")
+
         self.client(func)
         self.startup_functions.append(func.__name__)
         return func
@@ -78,6 +84,9 @@ def server(self, func: Callable):
         Decorator to expose a function as a server-side RPC endpoint.
         The client bundle will receive a stub that calls this endpoint via fetch.
         """
+        if not inspect.iscoroutinefunction(func):
+            raise ValueError("func must be async")
+
         self.server_functions[func.__name__] = func
 
         # 1. Introspect the function to find arguments
@@ -206,7 +215,7 @@ def _generate_bundle(self) -> str:
         Generates the Python bundle to run in the browser.
         """
         # 1. Mock the 'app' object
-        header = "class MockApp:\n    def client(self, f): return f\n    def server(self, f): return f\napp = MockApp()\n\nclass Event: pass"
+        header = "class Event: pass"
 
         # 2. Inject violetear.dom module
         # This allows 'from violetear.dom import Document' to work in the browser
@@ -251,7 +260,9 @@ def _generate_bundle(self) -> str:
         # 4. Extract User Functions
         user_code = []
         for name, func in self.client_functions.items():
-            user_code.append(inspect.getsource(func))
+            code = inspect.getsource(func).split("\n")
+            code = [c for c in code if not c.startswith("@")] # remove decorators
+            user_code.append("\n".join(code))
 
         # 5. Generate Server Stubs
         server_stubs = self._generate_server_stubs()
@@ -261,7 +272,7 @@ def _generate_bundle(self) -> str:
 
         # Run startup functions
         for func_name in self.startup_functions:
-            init_code += f"\n{func_name}()"
+            init_code += f"\nawait {func_name}()"
 
         return "\n\n".join(
             [

violetear/storage.py

@@ -6,6 +6,7 @@
 
 if IS_BROWSER:
     from js import localStorage, sessionStorage
+    from pyodide.ffi import JsNull
 
 
 class Thing:
@@ -126,10 +127,7 @@ def _del_raw(self, key):
 
     def __getattr__(self, key: str) -> Any:
         # Enables 'store.user' syntax
-        try:
-            return self[key]
-        except KeyError:
-            return None
+        return self[key]
 
     def __setattr__(self, key: str, value: Any):
         if key.startswith("_"):
@@ -139,8 +137,8 @@ def __setattr__(self, key: str, value: Any):
 
     def __getitem__(self, key: str) -> Any:
         raw = self._get_raw(key)
-        if raw is None:
-            raise KeyError(key)
+        if isinstance(raw, JsNull):
+            return None
 
         try:
             data = json.loads(raw)