feat(api): support the change user group-quota API

Author: BobAnkh

src/pynetmito/__init__.py

@@ -38,6 +38,8 @@
     CreateGroupReq,
     ChangeGroupStorageQuotaReq,
     GroupStorageQuotaResp,
+    ChangeUserGroupQuota,
+    UserGroupQuotaResp,
     WorkerQueryInfo,
     WorkerQueryResp,
     WorkersQueryReq,
@@ -99,6 +101,8 @@
     "CreateGroupReq",
     "ChangeGroupStorageQuotaReq",
     "GroupStorageQuotaResp",
+    "ChangeUserGroupQuota",
+    "UserGroupQuotaResp",
     "WorkerQueryInfo",
     "WorkerQueryResp",
     "WorkersQueryReq",

src/pynetmito/client.py

@@ -29,6 +29,8 @@
     CreateGroupReq,
     ChangeGroupStorageQuotaReq,
     GroupStorageQuotaResp,
+    ChangeUserGroupQuota,
+    UserGroupQuotaResp,
     WorkerQueryResp,
     WorkersQueryReq,
     WorkersQueryResp,
@@ -569,6 +571,21 @@ def admin_update_group_storage_quota(
                 f"Failed to update storage quota for group {group_name}, status code: {resp.status_code}, error: {resp.text}"
             )
 
+    def admin_update_user_group_quota(
+        self, username: str, req: ChangeUserGroupQuota
+    ) -> UserGroupQuotaResp:
+        url = self._get_url(f"admin/users/{username}/group-quota")
+        headers = {"Authorization": f"Bearer {self.credential}"}
+        resp = self.http_client.post(url, headers=headers, json=req.to_dict())
+        if resp.status_code == 200:
+            r = UserGroupQuotaResp.model_validate(resp.json())
+            return r
+        else:
+            self.logger.error(resp.text)
+            raise Exception(
+                f"Failed to update group quota for user {username}, status code: {resp.status_code}, error: {resp.text}"
+            )
+
     def query_attachments_by_filter(
         self, group_name: str, req: AttachmentsQueryReq
     ) -> AttachmentsQueryResp:

src/pynetmito/schemas.py

@@ -657,6 +657,14 @@ class GroupStorageQuotaResp(BaseAPIModel):
     storage_quota: int
 
 
+class ChangeUserGroupQuota(BaseAPIModel):
+    group_quota: str
+
+
+class UserGroupQuotaResp(BaseAPIModel):
+    group_quota: int
+
+
 class WorkerQueryInfo(BaseAPIModel):
     worker_id: UUID4
     creator_username: str