refac/enh: group add/remove users endpoints

2025-07-17 01:50:37 +04:00
parent 51242f4484
commit 9634df4347
2 changed files with 104 additions and 1 deletions
@@ -83,10 +83,14 @@ class GroupForm(BaseModel):
    permissions: Optional[dict] = None


-class GroupUpdateForm(GroupForm):
+class UserIdsForm(BaseModel):
    user_ids: Optional[list[str]] = None


+class GroupUpdateForm(GroupForm, UserIdsForm):
+    pass
+
+
 class GroupTable:
    def insert_new_group(
        self, user_id: str, form_data: GroupForm
@@ -275,5 +279,53 @@ class GroupTable:
                log.exception(e)
                return False

+    def add_users_to_group(
+        self, id: str, user_ids: Optional[list[str]] = None
+    ) -> Optional[GroupModel]:
+        try:
+            with get_db() as db:
+                group = db.query(Group).filter_by(id=id).first()
+                if not group:
+                    return None
+
+                if not group.user_ids:
+                    group.user_ids = []
+
+                for user_id in user_ids:
+                    if user_id not in group.user_ids:
+                        group.user_ids.append(user_id)
+
+                group.updated_at = int(time.time())
+                db.commit()
+                db.refresh(group)
+                return GroupModel.model_validate(group)
+        except Exception as e:
+            log.exception(e)
+            return None
+
+    def remove_users_from_group(
+        self, id: str, user_ids: Optional[list[str]] = None
+    ) -> Optional[GroupModel]:
+        try:
+            with get_db() as db:
+                group = db.query(Group).filter_by(id=id).first()
+                if not group:
+                    return None
+
+                if not group.user_ids:
+                    return GroupModel.model_validate(group)
+
+                for user_id in user_ids:
+                    if user_id in group.user_ids:
+                        group.user_ids.remove(user_id)
+
+                group.updated_at = int(time.time())
+                db.commit()
+                db.refresh(group)
+                return GroupModel.model_validate(group)
+        except Exception as e:
+            log.exception(e)
+            return None
+

 Groups = GroupTable()