mirrors/github.com_openSUSE_osc
1
0
Fork 0
mirror of https://github.com/openSUSE/osc.git synced 2025-09-08 05:58:43 +02:00
Code Activity

Add support for List[BaseModel] type to Field class

Browse Source
This commit is contained in:
Daniel Mach
2024-01-23 20:05:38 +01:00
parent ea0bf1bb60
commit 7903ade2b4
2 changed files with 86 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
osc/util/models.py
View File

@@ -130,6 +130,23 @@ class Field(property):
return get_origin(types[0]) or types[0]
return origin_type
@property
def inner_type(self):
if self.is_optional:
types = [i for i in self.type.__args__ if i != type(None)]
type_ = types[0]
else:
type_ = self.type
if get_origin(type_) != list:
return None
if not hasattr(type_, "__args__"):
return None
inner_type = [i for i in type_.__args__ if i != type(None)][0]
return inner_type
@property
def is_optional(self):
origin_type = get_origin(self.type) or self.type
@@ -139,6 +156,10 @@ class Field(property):
def is_model(self):
return inspect.isclass(self.origin_type) and issubclass(self.origin_type, BaseModel)
@property
def is_model_list(self):
return inspect.isclass(self.inner_type) and issubclass(self.inner_type, BaseModel)
def validate_type(self, value, expected_types=None):
if not expected_types and self.is_optional and value is None:
return True
@@ -255,6 +276,15 @@ class Field(property):
# initialize a model instance from a dictionary
klass = self.origin_type
value = klass(**value) # pylint: disable=not-callable
elif self.is_model_list and isinstance(value, list):
new_value = []
for i in value:
if isinstance(i, dict):
klass = self.inner_type
new_value.append(klass(**i))
else:
new_value.append(i)
value = new_value
self.validate_type(value)
obj._values[self.name] = value
@@ -342,6 +372,8 @@ class BaseModel(metaclass=ModelMeta):
value = getattr(self, name)
if value is not None and field.is_model:
result[name] = value.dict()
if value is not None and field.is_model_list:
result[name] = [i.dict() for i in value]
else:
result[name] = value