xzbautomarkapi/admin/api/endpoints/sysdata/role.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from typing import Optional

from fastapi import APIRouter, Depends, Query
from sqlalchemy import text
from sqlalchemy.ext.asyncio import AsyncSession

from crud.sysdata.role import crud_role, crud_permission
from models.sysdata.role import Role, Permission
from models.user import Admin
from schemas.base import ReturnField, OrderByField
from schemas.sysdata.role import RoleList, RoleDetail, NewRoleInfo, UpdateRoleInfo
from utils.depends import get_async_db, get_current_user

router = APIRouter()


@router.post("/role", summary="新增角色", response_model=RoleDetail)
async def create_role(info: NewRoleInfo,
                      db: AsyncSession = Depends(get_async_db),
                      current_user: Admin = Depends(get_current_user)):
    info = info.dict(exclude_none=True)
    if not info:
        return {"errcode": 400, "mess": "请求参数为空!"}
    db_role = await crud_role.find_one(db, filters={"name": info["name"]})
    if db_role:
        return {"errcode": 400, "mess": "角色名称重复!"}
    info["creator_id"] = current_user.id
    info["creator_name"] = current_user.username
    db_role = await crud_role.insert_one(db, info)

    return {"data": db_role}


@router.put("/role/{rid}", summary="修改角色", response_model=RoleDetail)
async def update_role(rid: int,
                      info: UpdateRoleInfo,
                      db: AsyncSession = Depends(get_async_db),
                      current_user: Admin = Depends(get_current_user)):
    info_dict = info.dict(exclude_none=True)
    if not info_dict:
        return {"errcode": 400, "mess": "请求参数为空!"}
    db_role = await crud_role.find_one(db, filters={"id": rid})
    if not db_role:
        return {"errcode": 400, "mess": "角色不存在!"}

    info_dict["editor_id"] = current_user.id
    info_dict["editor_name"] = current_user.username
    db_role = await crud_role.update(db, db_role, info_dict)

    return {"data": db_role}


@router.delete("/role/{rid}", summary="删除角色", response_model=RoleDetail)
async def delete_role(rid: int,
                      db: AsyncSession = Depends(get_async_db),
                      current_user: Admin = Depends(get_current_user)):
    # 不允许删除初始管理员角色，ID=1
    if rid == 1:
        return {"errcode": 400, "mess": "该角色不允许删除!"}

    db_role = await crud_role.count(db, filters={"id": rid})
    if not db_role:
        return {"errcode": 404, "mess": "角色不存在!"}
    await crud_role.delete(db, obj_id=rid)
    return {"data": None}


@router.get("/role/{rid}", summary="角色详情", response_model=RoleDetail)
async def get_role(rid: int,
                   db: AsyncSession = Depends(get_async_db),
                   current_user: Admin = Depends(get_current_user)):
    db_role = await crud_role.find_one(db, filters={"id": rid})
    if not db_role:
        return {"errcode": 404, "mess": "角色不存在!"}
    return {"data": db_role}


@router.get("/roles", summary="角色列表", response_model=RoleList)
async def get_role_list(page: Optional[int] = None,
                        size: Optional[int] = None,
                        platform: str = "",
                        name: str = "",
                        res: ReturnField = Query("", description="控制返回字段,字段逗号分隔"),
                        order: OrderByField = Query("-created_at",
                                                    description="排序字段,用逗号分隔,升降序以-判断,默认-created_at"),
                        db: AsyncSession = Depends(get_async_db),
                        current_user: Admin = Depends(get_current_user)):
    _q = []
    if name:
        _q.append(Role.name.like(f"%{name}%"))
    if platform:
        _q.append(Role.platform == platform)
    if ((page is not None) and page >= 1) and ((size is not None) and size >= 1):
        offset = (page - 1) * size
    else:
        size = offset = None

    if isinstance(order, str):
        order = [text(order)]
    total, db_roles = await crud_role.find_all(db,
                                               filters=_q,
                                               offset=offset,
                                               limit=size,
                                               order_by=order,
                                               return_fields=res)
    return {"data": db_roles, "total": total}


@router.get("/permissions", summary="权限类别")
async def get_permission_all(name: str = "",
                             res: ReturnField = Query("", description="控制返回字段,字段逗号分隔"),
                             db: AsyncSession = Depends(get_async_db),
                             current_user: Admin = Depends(get_current_user)):
    _q = []
    if name:
        _q.append(Permission.name.like(f"%{name}%"))
    total, db_perms = await crud_permission.find_all(db, filters=_q, return_fields=res)
    db_perms = [{
        "name": "首页",
        "value": "menu.*",
        "children": [{
            "name": "首页",
            "value": "menu.menu",
            "children": []
        }]
    }, {
        "name":
            "学校管理",
        "value":
            "menu2.*",
        "children": [{
            "name": "学校管理",
            "value": "menu2.menu2-1",
            "children": []
        }, {
            "name": "班级管理",
            "value": "menu2.menu2-2",
            "children": []
        }]
    }, {
        "name":
            "师生信息",
        "value":
            "menu3.*",
        "children": [{
            "name": "教师管理",
            "value": "menu3.menu3-1",
            "children": []
        }, {
            "name": "学生管理",
            "value": "menu3.menu3-2",
            "children": []
        }]
    }, {
        "name":
            "分类管理",
        "value":
            "menu8.*",
        "children": [{
            "name": "作业分类",
            "value": "menu8.menu8-1",
            "children": []
        }, {
            "name": "试卷分类",
            "value": "menu8.menu8-2",
            "children": []
        }]
    }, {
        "name":
            "资源管理",
        "value":
            "menu4.*",
        "children": [{
            "name": "作业资源",
            "value": "menu4.menu4-1",
            "children": []
        }, {
            "name": "考试资源",
            "value": "menu4.menu4-2",
            "children": []
        }]
    }, {
        "name":
            "阅卷管理",
        "value":
            "menu5.*",
        "children": [{
            "name": "作业管理",
            "value": "menu5.menu5-1",
            "children": []
        }, {
            "name": "作业批阅",
            "value": "menu5.menu5-2",
            "children": []
        }, {
            "name": "考试批阅",
            "value": "menu5.menu5-3",
            "children": []
        }, {
            "name": "考试管理",
            "value": "menu5.menu5-4",
            "children": []
        }]
    }, {
        "name":
            "错题管理",
        "value":
            "menu6.*",
        "children": [{
            "name": "班级错题",
            "value": "menu6.menu6-1",
            "children": []
        }, {
            "name": "学生错题",
            "value": "menu6.menu6-2",
            "children": []
        }, {
            "name": "错题打印",
            "value": "menu6.menu6-3",
            "children": []
        }]
    }, {
        "name":
            "系统管理",
        "value":
            "menu7.*",
        "children": [{
            "name": "账号管理",
            "value": "menu7.menu7-1",
            "children": []
        }, {
            "name": "角色管理",
            "value": "menu7.menu7-2",
            "children": []
        }]
    }]

    return {"data": db_perms, "total": total}