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

Author: Xian-Yu0

app/api/v1/endpoints/group.py

@@ -5,8 +5,8 @@
 
 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, crud_modify_basic_info, crud_modify_admin_list, crud_remove_member, crud_leave_group
-from app.schemas.group import ApplyToEnter, LeaveGroup
+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, crud_get_basic_info, crud_get_people_info, crud_get_my_level
+from app.schemas.group import ApplyToEnter, LeaveGroup, GetBasicInfo
 
 router = APIRouter()
 
@@ -55,7 +55,7 @@ async def modify_basic_info(group_id: int = Query(...), group_name: str | None =
     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的文件列表，最多有一个元素
+        old_avatar = glob.glob(os.path.join("/lhcos-data/group-avatar", f"{group_id}.*"))   # 基本名为group_id的文件列表，最多有一个元素
         if old_avatar:
             os.remove(old_avatar[0])
         # 存储新头像，文件名为 {group_id}.上传文件的扩展名
@@ -83,4 +83,22 @@ async def leave_group(model: LeaveGroup, db: AsyncSession = Depends(get_db), use
     await crud_leave_group(group_id, user_id, db)
     return {"msg": "You successfully left the group"}
 
-# 写返回个人文件树的后端时记得加 visible = True
+@router.get("/getBasicInfo", response_model=dict)
+async def get_basic_info(model: GetBasicInfo, db: AsyncSession = Depends(get_db)):
+    group_id = model.group_id
+    name, desc = crud_get_basic_info(group_id, db)
+    find = glob.glob(os.path.join("/lhcos-data/group-avatar", f"{group_id}.*"))
+    avatar = 'default.png' if not find else find[0].removeprefix("/lhcos-data/group-avatar\\\\")
+    avatar = '/lhcos-data/group-avatar/' + avatar
+    return {"avatar": avatar, "name": name, "desc": desc}
+
+@router.get("/getPeopleInfo", response_model=dict)
+async def get_people_info(group_id: int = Query(...), db: AsyncSession = Depends(get_db)):
+    leader, admins, members = await crud_get_people_info(group_id, db)
+    return {"leader": leader, "admins": admins, "members": members}
+
+@router.get("/getMyLevel", response_model=dict)
+async def get_my_level(group_id: int = Query(...), db: AsyncSession = Depends(get_db), user: dict = Depends(get_current_user)):
+    user_id = user.get("id")
+    level = await crud_get_my_level(user_id, group_id, db)
+    return {"level": level}

app/curd/group.py

@@ -92,4 +92,59 @@ 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()
+    await db.commit()
+
+async def crud_get_basic_info(group_id: int, db: AsyncSession):
+    query = select(Group.name, Group.description).where(Group.id == group_id)
+    result = await db.execute(query)
+    group = result.first()
+    return group.name, group.description
+
+async def crud_get_people_info(group_id: int, db: AsyncSession):
+    # 创建者信息
+    query = select(Group.leader).where(Group.id == group_id)
+    result = await db.execute(query)
+    leader_id = result.scalar_one_or_none()
+    query = select(User).where(User.id == leader_id)
+    result = await db.execute(query)
+    user = result.scalar_one_or_none()
+    leader = {"id": user.id, "name": user.username, "avatar": user.avatar}
+
+    # 管理者信息
+    query = select(user_group.c.user_id).where(user_group.c.group_id == group_id, user_group.c.is_admin == True)
+    result = await db.execute(query)
+    admin_ids = result.scalars().all()
+    query = select(User).where(User.id.in_(admin_ids))
+    result = await db.execute(query)
+    users = result.all()
+    admins = [{"id": user.id, "name": user.username, "avatar": user.avatar} for user in users]
+    
+    # 普通成员信息
+    query = select(user_group.c.user_id).where(user_group.c.group_id == group_id, user_group.c.is_admin == False)
+    result = await db.execute(query)
+    member_ids = result.scalars.all()
+    query = select(User).where(User.id.in_(member_ids))
+    result = await db.execute(query)
+    users = result.all()
+    members = [{"id": user.id, "name": user.username, "avatar": user.avatar} for user in users]
+    
+    return leader, admins, members
+
+async def crud_get_my_level(user_id: int, group_id: int, db: AsyncSession):
+    # 是否是创建者
+    query = select(Group.leader).where(Group.id == group_id)
+    result = await db.execute(query)
+    leader_id = result.scalar_one_or_none()
+    if user_id == leader_id:
+        return 1
+    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 relation and relation["is_admin"]:
+        return 2
+    # 是否是普通成员
+    if relation: # and not relation["is_admin"]:
+        return 3
+    # 未加入组织
+    return 4

app/schemas/group.py

@@ -4,4 +4,7 @@ class ApplyToEnter(BaseModel):
     group_id: int
 
 class LeaveGroup(BaseModel):
+    group_id: int
+
+class GetBasicInfo(BaseModel):
     group_id: int