Refactoring Hub Firms routes

2025-04-11 21:58:04 +02:00
parent ef9ae99cb6
commit d1718becde
3 changed files with 39 additions and 33 deletions
--- a/api/rpk-api/hub/user/schemas.py
+++ b/api/rpk-api/hub/user/schemas.py
@@ -8,5 +8,11 @@ from hub.firm import FirmRead
 class UserSchema(BaseUser[PydanticObjectId]):
    firms: list[FirmRead] = Field()

+    def belongs_to(self, firm):
+        for f in self.firms:
+            if f.instance == firm.instance and f.firm == firm.firm :
+                return True
+        return False
+
 class UserUpdateSchema(BaseUserUpdate):
    pass