From 9ebdf48459be9c16bf343e6807dfa4ad9cb4f9b5 Mon Sep 17 00:00:00 2001
From: Luis-Hebendanz <consulting@qube.email>
Date: Sun, 14 Jan 2024 15:23:43 +0100
Subject: [PATCH] Added AP get repository

---
 pkgs/clan-cli/clan_cli/webui/schemas.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/pkgs/clan-cli/clan_cli/webui/schemas.py b/pkgs/clan-cli/clan_cli/webui/schemas.py
index a203d0f..033d168 100644
--- a/pkgs/clan-cli/clan_cli/webui/schemas.py
+++ b/pkgs/clan-cli/clan_cli/webui/schemas.py
@@ -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
 
 
 #########################