Added AP get repository #52

Merged
merge-bot merged 3 commits from Qubasa-main into main 2024-01-14 15:01:37 +00:00
Showing only changes of commit 52987625c8 - Show all commits

View File

@@ -72,8 +72,11 @@ class Entity(EntityBase):
# define a custom getter function for roles
@validator("roles", pre=True)
def get_roles(cls, v: List[EntityRoles]) -> List[Role]:
return [x.role for x in v]
def get_roles(cls, v: List[EntityRoles | Role]) -> List[Role]:
if isinstance(v, list) and len(v) > 0 and isinstance(v[0], EntityRoles):
return [x.role for x in v] # type: ignore
else:
return v # type: ignore
#########################