ITQC/pydase
24
0
Fork 0
mirror of https://github.com/tiqi-group/pydase.git synced 2025-04-22 09:10:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: updating serialization of functions

Browse Source
This commit is contained in:
Mose Müller 2023-08-10 15:48:45 +02:00
parent 06f93c110e
commit d94b85f305
1 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/pydase/data_service/data_service.py
View File

@ -3,7 +3,7 @@ import inspect
import json
import os
from enum import Enum
from typing import Any, Optional, Union, cast, get_args, get_type_hints
from typing import Any, Optional, cast, get_type_hints
import rpyc
from loguru import logger
@ -265,22 +265,13 @@ class DataService(rpyc.Service, AbstractDataService):
}
elif inspect.isfunction(value) or inspect.ismethod(value):
sig = inspect.signature(value)
# Store parameters and their anotations in a dictionary
parameters: dict[str, Optional[str]] = {}
for k, v in sig.parameters.items():
annotation = v.annotation
if annotation is not inspect._empty:
if (
hasattr(annotation, "__origin__")
and annotation.__origin__ is Union
):
# Handle union types
union_types = get_args(annotation)
type_names = [
t.__name__ if isinstance(t, type) else str(t)
for t in union_types
]
parameters[k] = " | ".join(type_names)
elif isinstance(annotation, type):
if isinstance(annotation, type):
# Handle regular types
parameters[k] = annotation.__name__
else: