Replace SimpleNamespaces with dataclasses. Add base class _Typing in labeler.py to convert dictionaries to dataclasses. Remove SimpleNamespace references.

This commit is contained in:
Davide Casale
2023-01-16 16:30:06 +01:00
parent 903f68c6e3
commit 0bb9f65a19
3 changed files with 66 additions and 36 deletions

View File

@@ -2,10 +2,13 @@ from .exceptions import LabelerSerializerException
from typing import Generic, TypeVar, Iterable, Optional, List, Tuple, Any, cast
from types import SimpleNamespace
T = TypeVar("T")
class _Typing(object):
def __init__(self, **kwargs):
for key, value in kwargs.items():
self.__setattr__(key,value)
class _Serializer(Generic[T]):
def __init__(self, name: str, labels: List[str], IGNORE: List[str] = [ "_PLACEHOLDER" ]):
self.name, self.__labels, self.__IGNORE = name, labels, IGNORE
@@ -21,4 +24,4 @@ class _Serializer(Generic[T]):
yield label, args[index]
def parse(self, *values: Any, skip: Optional[List[str]] = None) -> T:
return cast(T, SimpleNamespace(**dict(self._serialize(*values, skip=skip))))
return cast(T, _Typing(**dict(self._serialize(*values, skip=skip))))