From 9663dea79dd46393318d502e0229938b6ca0ce22 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mose=20M=C3=BCller?= Date: Wed, 31 Jul 2024 08:44:23 +0200 Subject: [PATCH] feat: allow monkey patching serilization methods for api endpoints --- src/pydase/server/web_server/api/v1/endpoints.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/pydase/server/web_server/api/v1/endpoints.py b/src/pydase/server/web_server/api/v1/endpoints.py index 070b489..f09a8cf 100644 --- a/src/pydase/server/web_server/api/v1/endpoints.py +++ b/src/pydase/server/web_server/api/v1/endpoints.py @@ -1,12 +1,15 @@ from typing import Any +import pydase.utils.serialization.deserializer +import pydase.utils.serialization.serializer from pydase.data_service.state_manager import StateManager from pydase.server.web_server.sio_setup import TriggerMethodDict, UpdateDict from pydase.utils.helpers import get_object_attr_from_path -from pydase.utils.serialization.deserializer import loads -from pydase.utils.serialization.serializer import Serializer, dump from pydase.utils.serialization.types import SerializedObject +loads = pydase.utils.serialization.deserializer.loads +Serializer = pydase.utils.serialization.serializer.Serializer + def update_value(state_manager: StateManager, data: UpdateDict) -> None: path = data["access_path"] @@ -32,4 +35,4 @@ def trigger_method(state_manager: StateManager, data: TriggerMethodDict) -> Any: serialized_kwargs = data.get("kwargs", None) kwargs: dict[str, Any] = loads(serialized_kwargs) if serialized_kwargs else {} - return dump(method(*args, **kwargs)) + return Serializer.serialize_object(method(*args, **kwargs))