Explorar o código

fix: functions

Timothy J. Baek hai 10 meses
pai
achega
aa88022624
Modificáronse 1 ficheiros con 8 adicións e 7 borrados
  1. 8 7
      backend/apps/webui/models/functions.py

+ 8 - 7
backend/apps/webui/models/functions.py

@@ -107,7 +107,7 @@ class FunctionsTable:
             Session.commit()
             Session.commit()
             Session.refresh(result)
             Session.refresh(result)
             if result:
             if result:
-                return FunctionModel(**result.model_dump())
+                return FunctionModel(**result.__dict__)
             else:
             else:
                 return None
                 return None
         except Exception as e:
         except Exception as e:
@@ -117,19 +117,20 @@ class FunctionsTable:
     def get_function_by_id(self, id: str) -> Optional[FunctionModel]:
     def get_function_by_id(self, id: str) -> Optional[FunctionModel]:
         try:
         try:
             function = Session.get(Function, id)
             function = Session.get(Function, id)
-            return FunctionModel(**function)
+            return FunctionModel(**function.__dict__)
         except:
         except:
             return None
             return None
 
 
     def get_functions(self, active_only=False) -> List[FunctionModel]:
     def get_functions(self, active_only=False) -> List[FunctionModel]:
         if active_only:
         if active_only:
             return [
             return [
-                FunctionModel(**function)
+                FunctionModel(**function.__dict__)
                 for function in Session.query(Function).filter_by(is_active=True).all()
                 for function in Session.query(Function).filter_by(is_active=True).all()
             ]
             ]
         else:
         else:
             return [
             return [
-                FunctionModel(**function) for function in Session.query(Function).all()
+                FunctionModel(**function.__dict__)
+                for function in Session.query(Function).all()
             ]
             ]
 
 
     def get_functions_by_type(
     def get_functions_by_type(
@@ -137,20 +138,20 @@ class FunctionsTable:
     ) -> List[FunctionModel]:
     ) -> List[FunctionModel]:
         if active_only:
         if active_only:
             return [
             return [
-                FunctionModel(**function)
+                FunctionModel(**function.__dict__)
                 for function in Session.query(Function)
                 for function in Session.query(Function)
                 .filter_by(type=type, is_active=True)
                 .filter_by(type=type, is_active=True)
                 .all()
                 .all()
             ]
             ]
         else:
         else:
             return [
             return [
-                FunctionModel(**function)
+                FunctionModel(**function.__dict__)
                 for function in Session.query(Function).filter_by(type=type).all()
                 for function in Session.query(Function).filter_by(type=type).all()
             ]
             ]
 
 
     def get_global_filter_functions(self) -> List[FunctionModel]:
     def get_global_filter_functions(self) -> List[FunctionModel]:
         return [
         return [
-            FunctionModel(**function)
+            FunctionModel(**function.__dict__)
             for function in Session.query(Function)
             for function in Session.query(Function)
             .filter_by(type="filter", is_active=True, is_global=True)
             .filter_by(type="filter", is_active=True, is_global=True)
             .all()
             .all()