sqlmodel: avoid using __dict__

adsharma · Jan 11, 2025 · ab5a43b · ab5a43b
1 parent 5d72c36
commit ab5a43b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/fquery/sqlmodel.py b/fquery/sqlmodel.py
@@ -44,7 +44,8 @@ def model(table: bool = True, table_name: str = None, global_id: bool = False):
     """
 
     def sqlmodel(self) -> SQLModel:
-        return self.__sqlmodel__(**self.__dict__)
+        attrs = {name: getattr(self, name) for name in self.__sqlmodel__.__fields__}
+        return self.__sqlmodel__(**attrs)
 
     def decorator(cls):
         # Check if the class is a dataclass