From 612e62d06be2d1153b0120bf69ea9dca0a42c7c0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mose=20M=C3=BCller?= Date: Wed, 27 Mar 2024 12:13:12 +0100 Subject: [PATCH] updates ProxyClassFactory (go through handled types before components) --- src/pydase/client/proxy_class_factory.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/pydase/client/proxy_class_factory.py b/src/pydase/client/proxy_class_factory.py index 72095e2..98ca315 100644 --- a/src/pydase/client/proxy_class_factory.py +++ b/src/pydase/client/proxy_class_factory.py @@ -45,6 +45,11 @@ class ProxyClassFactory: "Exception": loads, } + # First go through handled types (as ColouredEnum is also within the components) + handler = type_handler.get(serialized_object["type"]) + if handler: + return handler(serialized_object) + # Custom types like Components or DataService classes component_class = Deserializer.get_component_class(serialized_object["type"]) if component_class: @@ -53,10 +58,6 @@ class ProxyClassFactory: ) proxy_class._sio = self.sio_client return proxy_class - - handler = type_handler.get(serialized_object["type"]) - if handler: - return handler(serialized_object) return None def _deserialize_method(self, serialized_object: SerializedObject) -> Any: