[feat]: 组织管理基本功能

Author: Xian-Yu0

app/api/v1/endpoints/group.py

@@ -1,11 +1,12 @@
 from fastapi import APIRouter, Query, Body, UploadFile, File, Depends, HTTPException
 from sqlalchemy.ext.asyncio import AsyncSession
 import os
+import glob
 
 from app.utils.get_db import get_db
 from app.utils.auth import get_current_user
-from app.curd.group import crud_create, crud_apply_to_enter, crud_get_applications, crud_reply_to_enter
-from app.schemas.group import ApplyToEnter
+from app.curd.group import crud_create, crud_apply_to_enter, crud_get_applications, crud_reply_to_enter, crud_modify_basic_info, crud_modify_admin_list, crud_remove_member, crud_leave_group
+from app.schemas.group import ApplyToEnter, LeaveGroup
 
 router = APIRouter()
 
@@ -17,6 +18,7 @@ async def create(group_name: str = Query(...), group_desc: str = Query(...), gro
     if len(group_desc) > 200:
         raise HTTPException(status_code=405, detail="Invalid group description, longer than 200")
     group_id = await crud_create(user.get("id"), group_name, group_desc, db)
+    # 存储头像，文件名为 {group_id}.上传文件的扩展名
     if group_avatar:
         os.makedirs("/lhcos-data/group-avatar", exist_ok=True)
         ext = os.path.splitext(group_avatar.filename)[1]
@@ -39,8 +41,46 @@ async def get_applications(group_id: int = Query(...), db: AsyncSession = Depend
     return {"users": users}
 
 @router.post("/replyToEnter", response_model=dict)
-async def reply_to_enter(user_id: int = Body(...), group_id: int = Body(...), reply: int = Body(...), db: AsyncSession = Depends(get_db)):
-    if reply != 0 and reply != 1:
-        raise HTTPException(status_code=405, detail="Wrong parameter, reply should be either 0 or 1")
+async def reply_to_enter(user_id: int = Body(...), group_id: int = Body(...), reply: bool = Body(...), db: AsyncSession = Depends(get_db)):
     msg = await crud_reply_to_enter(user_id, group_id, reply, db)
-    return {"msg": msg}
+    return {"msg": msg}
+
+@router.post("/modifyBasicInfo", response_model=dict)
+async def modify_basic_info(group_id: int = Query(...), group_name: str | None = Query(None), group_desc: str | None = Query(None), group_avatar: UploadFile | None = File(None), db: AsyncSession = Depends(get_db)):
+    if group_name and len(group_name) > 30:
+        raise HTTPException(status_code=405, detail="Invalid group name, longer than 30")
+    if group_desc and len(group_desc) > 200:
+        raise HTTPException(status_code=405, detail="Invalid group description, longer than 200")
+    await crud_modify_basic_info(db=db, id=group_id, name=group_name, desc=group_desc)
+    if group_avatar:
+        os.makedirs("/lhcos-data/group-avatar", exist_ok=True)
+        # 若之前存储了旧头像，则将其删除；若之前就没头像，则不做处理
+        old_avatar = glob.glob(os.path.join("/lhcos-data/group-avatar", group_id + ".*"))   # 基本名为group_id的文件列表，最多有一个元素
+        if old_avatar:
+            os.remove(old_avatar[0])
+        # 存储新头像，文件名为 {group_id}.上传文件的扩展名
+        ext = os.path.splitext(group_avatar.filename)[1]
+        path = os.path.join("/lhcos-data/group-avatar", f"{group_id}{ext}")
+        with open(path, "wb") as f:
+            content = await group_avatar.read()
+            f.write(content)
+    return {"msg": "Basic info modified successfully"}
+
+@router.post("/modifyAdminList", response_model=dict)
+async def modify_admin_list(group_id: int = Body(...), user_id: int = Body(...), add_admin: bool = Body(...), db: AsyncSession = Depends(get_db)):
+    msg = await crud_modify_admin_list(group_id, user_id, add_admin, db)
+    return {"msg": msg}
+
+@router.post("/removeMember", response_model=dict)
+async def remove_member(group_id: int = Body(...), user_id: int = Body(...), db: AsyncSession = Depends(get_db)):
+    await crud_remove_member(group_id, user_id, db)
+    return {"msg": "Member removed successfully"}
+
+@router.post("/leaveGroup", response_model=dict)
+async def leave_group(model: LeaveGroup, db: AsyncSession = Depends(get_db), user: dict = Depends(get_current_user)):
+    group_id = model.group_id
+    user_id = user.get("id")
+    await crud_leave_group(group_id, user_id, db)
+    return {"msg": "You successfully left the group"}
+
+# 写返回个人文件树的后端时记得加 visible = True

app/curd/group.py

@@ -1,7 +1,7 @@
 from fastapi import HTTPException
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.exc import IntegrityError
-from sqlalchemy import select, insert, delete
+from sqlalchemy import select, insert, delete, update
 from app.models.model import User, Group, Folder, Article, Note, Tag, user_group, enter_application
 
 async def crud_create(leader: int, name: str, description: str, db: AsyncSession):
@@ -41,18 +41,55 @@ async def crud_get_applications(group_id: int, db: AsyncSession):
     users = result.all()
     return [{"user_id": user.id, "user_name": user.username} for user in users]
 
-async def crud_reply_to_enter(user_id: int, group_id: int, reply: int, db: AsyncSession):
+async def crud_reply_to_enter(user_id: int, group_id: int, reply: bool, db: AsyncSession):
     # 答复后，需要从待处理申请的表中删除表项
     query = delete(enter_application).where(enter_application.c.user_id == user_id, enter_application.c.group_id == group_id)
     result = await db.execute(query)
     if result.rowcount == 0:  # 如果没有删除任何行，说明不存在该项
         raise HTTPException(status_code=405, detail="Application is not existed or already handled")
     await db.commit()
 
-    if reply == 1:
+    if reply:
         new_relation = insert(user_group).values(user_id=user_id, group_id=group_id)
         await db.execute(new_relation)
         await db.commit()
         return "Add new member successfully"
 
     return "Refuse the application successfully"
+
+async def crud_modify_basic_info(db: AsyncSession, id: int, name: str | None = None, desc: str | None = None):
+    update_data = {}
+    if name:
+        update_data["name"] = name
+    if desc:
+        update_data["description"] = desc
+    query = update(Group).where(Group.id == id).values(**update_data)
+    await db.execute(query)
+    await db.commit()
+
+async def crud_modify_admin_list(group_id: int, user_id: int, add_admin: bool, db: AsyncSession):
+    # 检查组织中是否有该成员
+    query = select(user_group).where(user_group.c.user_id == user_id, user_group.c.group_id == group_id)
+    result = await db.execute(query)
+    relation = result.first()
+    if not relation:
+        raise HTTPException(status_code=405, detail="User currently not in the group")
+    
+    # 将该成员设为或取消管理员
+    query = update(user_group).where(user_group.c.group_id == group_id, user_group.c.user_id == user_id).values(is_admin=add_admin)
+    await db.execute(query)
+    await db.commit()
+
+    return "The user is an admin now" if add_admin else "The user is not an admin now"
+
+async def crud_remove_member(group_id: int, user_id: int, db: AsyncSession):    
+    # 不必先检查组织中是否有该成员，若没有则再执行一次delete也不会报错
+    query = delete(user_group).where(user_group.c.group_id == group_id, user_group.c.user_id == user_id)
+    await db.execute(query)
+    await db.commit()
+
+async def crud_leave_group(group_id: int, user_id: int, db: AsyncSession):
+    # 不必先检查组织中是否有该成员，若没有则再执行一次delete也不会报错
+    query = delete(user_group).where(user_group.c.group_id == group_id, user_group.c.user_id == user_id)
+    await db.execute(query)
+    await db.commit()

app/schemas/group.py

@@ -1,4 +1,7 @@
 from pydantic import BaseModel
 
 class ApplyToEnter(BaseModel):
+    group_id: int
+
+class LeaveGroup(BaseModel):
     group_id: int