Added DbObjects

src/app.py

@@ -4,6 +4,8 @@ from fasthtml import serve
 from fasthtml.components import *
 
 from myfasthtml.controls.Layout import Layout
+from myfasthtml.controls.helpers import Ids
+from myfasthtml.core.instances import InstancesManager
 from myfasthtml.myfastapp import create_app
 
 logging.basicConfig(
@@ -22,7 +24,7 @@ app, rt = create_app(protect_routes=True,
 
 @rt("/")
 def index(session):
-  layout = Layout(session, "Testing Layout")
+  layout = InstancesManager.get(session, Ids.Layout, Layout, "Testing Layout")
   layout.set_footer("Goodbye World")
   for i in range(1000):
     layout.left_drawer.append(Div(f"Left Drawer Item {i}"))

src/myfasthtml/controls/Layout.py

@@ -13,7 +13,8 @@ from myfasthtml.controls.BaseCommands import BaseCommands
 from myfasthtml.controls.UserProfile import UserProfile
 from myfasthtml.controls.helpers import mk, Ids
 from myfasthtml.core.commands import Command
-from myfasthtml.core.instances import MultipleInstance, InstancesManager
+from myfasthtml.core.dbmanager import DbObject
+from myfasthtml.core.instances import InstancesManager, SingleInstance
 from myfasthtml.icons.fluent import panel_left_expand20_regular as left_drawer_icon
 from myfasthtml.icons.fluent_p2 import panel_right_expand20_regular as right_drawer_icon
 
@@ -21,7 +22,10 @@ logger = logging.getLogger("LayoutControl")
 
 
 @dataclass
-class LayoutState:
+class LayoutState(DbObject):
+  def __init__(self, session, owner):
+    super().__init__(session, owner.get_id())
+  
   left_drawer_open: bool = True
   right_drawer_open: bool = False
 
@@ -31,7 +35,7 @@ class Commands(BaseCommands):
     return Command("ToggleDrawer", "Toggle main layout drawer", self._owner.toggle_drawer, "left")
 
 
-class Layout(MultipleInstance):
+class Layout(SingleInstance):
   """
   A responsive layout component with header, footer, main content area,
   and optional collapsible side drawers.
@@ -70,7 +74,7 @@ class Layout(MultipleInstance):
     self._header_content = None
     self._footer_content = None
     self._main_content = None
-    self._state = LayoutState()
+    self._state = LayoutState(session, self)
     self.commands = Commands(self)
     self.left_drawer = self.DrawerContent(self, "left")
     self.right_drawer = self.DrawerContent(self, "right")

src/myfasthtml/controls/UserProfile.py

@@ -2,7 +2,7 @@ from fasthtml.components import *
 
 from myfasthtml.controls.helpers import Ids, mk
 from myfasthtml.core.instances import SingleInstance
-from myfasthtml.core.utils import get_user_info
+from myfasthtml.core.utils import retrieve_user_info
 from myfasthtml.icons.material import dark_mode_filled, person_outline_sharp
 from myfasthtml.icons.material_p1 import light_mode_filled, alternate_email_filled
 
@@ -12,7 +12,7 @@ class UserProfile(SingleInstance):
     super().__init__(session, Ids.UserProfile)
   
   def render(self):
-    user_info = get_user_info(self._session)
+    user_info = retrieve_user_info(self._session)
     return Div(
       Div(user_info['username'],
           tabindex="0",

src/myfasthtml/controls/helpers.py

@@ -3,10 +3,14 @@ from fasthtml.components import *
 from myfasthtml.core.bindings import Binding
 from myfasthtml.core.commands import Command
 from myfasthtml.core.utils import merge_classes
+
+
 class Ids:
+  DbManager = "mf-dbmanager"
   Layout = "mf-layout"
   UserProfile = "mf-user-profile"
 
+
 class mk:
   
   @staticmethod

src/myfasthtml/core/dbmanager.py

@@ -0,0 +1,62 @@
+from dbengine.dbengine import DbEngine
+
+from myfasthtml.controls.helpers import Ids
+from myfasthtml.core.instances import SingleInstance, InstancesManager
+from myfasthtml.core.utils import retrieve_user_info
+
+
+class DbManager(SingleInstance):
+  def __init__(self, session, root=".myFastHtmlDb", auto_register: bool = True):
+    super().__init__(session, Ids.DbManager, auto_register=auto_register)
+    self.db = DbEngine(root=root)
+  
+  def save(self, entry, obj):
+    self.db.save(self.get_tenant(), self.get_user(), entry, obj)
+  
+  def load(self, entry):
+    return self.db.load(self.get_tenant(), entry)
+  
+  def exists_entry(self, entry):
+    return self.db.exists(self.get_tenant(), entry)
+  
+  def get_tenant(self):
+    return retrieve_user_info(self._session)["id"]
+  
+  def get_user(self):
+    return retrieve_user_info(self._session)["email"]
+
+
+class DbObject:
+  """
+  When you set the attribute, it persists in DB
+  It loads from DB at startup
+  """
+  
+  def __init__(self, session, name=None, db_manager=None):
+    self._session = session
+    self._name = name or self.__class__.__name__
+    self._db_manager = db_manager or InstancesManager.get(self._session, Ids.DbManager, DbManager)
+    
+    # init is possible
+    if self._db_manager.exists_entry(self._name):
+      props = self._db_manager.load(self._name)
+      for k, v in props.items():
+        setattr(self, k, v)
+    else:
+      self._save_self()
+  
+  def __setattr__(self, name: str, value: str):
+    if name.startswith("_"):
+      super().__setattr__(name, value)
+      return
+    
+    old_value = getattr(self, name, None)
+    if old_value == value:
+      return
+    
+    super().__setattr__(name, value)
+    self._save_self()
+  
+  def _save_self(self):
+    props = {k: getattr(self, k) for k, v in self.__class__.__dict__.items() if not k.startswith("_")}
+    self._db_manager.save(self._name, props)

src/myfasthtml/core/instances.py

@@ -11,10 +11,11 @@ class BaseInstance:
   Base class for all instances (manageable by InstancesManager)
   """
   
-  def __init__(self, session: dict, _id: str):
+  def __init__(self, session: dict, _id: str, auto_register: bool = True):
     self._session = session
     self._id = _id
-    InstancesManager.register(session, self)
+    if auto_register:
+      InstancesManager.register(session, self)
   
   def get_id(self):
     return self._id
@@ -28,8 +29,8 @@ class SingleInstance(BaseInstance):
   Base class for instances that can only have one instance at a time.
   """
   
-  def __init__(self, session: dict, prefix: str):
-    super().__init__(session, prefix)
+  def __init__(self, session: dict, prefix: str, auto_register: bool = True):
+    super().__init__(session, prefix, auto_register)
     self._instance = None
 
 
@@ -38,8 +39,8 @@ class MultipleInstance(BaseInstance):
   Base class for instances that can have multiple instances at a time.
   """
   
-  def __init__(self, session: dict, prefix: str):
-    super().__init__(session, f"{prefix}-{str(uuid.uuid4())}")
+  def __init__(self, session: dict, prefix: str, auto_register: bool = True):
+    super().__init__(session, f"{prefix}-{str(uuid.uuid4())}", auto_register)
     self._instance = None
 
 

src/myfasthtml/core/utils.py

@@ -191,7 +191,7 @@ def quoted_str(s):
   return str(s)
 
 
-def get_user_info(session: dict):
+def retrieve_user_info(session: dict):
   if not session:
     return {
         "id": "** NOT LOGGED IN **",