chore: 基线提交（重构前快照）

重构前保存现有 apps/ 结构基线，供后续各阶段回滚。 Co-Authored-By: Claude <noreply@anthropic.com>
2026-08-07 09:43:18 +08:00
commit dbcce79536
112 changed files with 9677 additions and 0 deletions
@@ -0,0 +1,177 @@
+import asyncio
+from datetime import datetime, timezone
+
+from fastapi.logger import logger
+from sqlalchemy import and_
+
+from apps.constants.message import PAGE_SIZE
+from apps.forms.role import RoleForm, RoleStatusForm
+from apps.models.role import Role
+from extensions import get_db_session
+from utils import R
+from utils.security import get_user_id
+from utils.request import like_contains, parse_id_list
+from utils.perm_cache import invalidate_role_perms
+
+
+# 查询角色分页数据
+
+def get_role_page(request):
+    db = get_db_session()
+    # 页码
+    page = int(request.query_params.get("pageNo", 1))
+    # 每页数
+    limit = int(request.query_params.get("pageSize", PAGE_SIZE))
+    # 分页查询
+    query = db.query(Role).filter(Role.is_delete == 0)
+    # 角色名称
+    name = request.query_params.get('name')
+    if name:
+        query = query.filter(Role.name.like(like_contains(name), escape='\\'))
+    # 排序
+    query = query.order_by(Role.sort.asc())
+    # 记录总数
+    count = query.count()
+    # 查询分页数据
+    role_list = query.limit(limit).offset((page - 1) * limit).all()
+    # 若当前页超出末页（数据被并发删除），自动回退到第一页
+    if page > 1 and not role_list:
+        page = 1
+        count = query.count()
+        role_list = query.limit(limit).offset(0).all()
+    # 实例化返回对象
+    result = []
+    # 遍历数据源
+    if len(role_list) > 0:
+        for item in role_list:
+            # 对象转字典
+            data = item.to_dict()
+            # 加入列表
+            result.append(data)
+    # 返回结果
+    return R.ok(data=result, count=count)
+# 根据角色ID查询详情
+
+def get_role_detail(role_id):
+    db = get_db_session()
+    # 根据ID查询角色
+    role = db.query(Role).filter(and_(Role.id == role_id, Role.is_delete == 0)).first()
+    # 查询结果判空
+    if not role:
+        return None
+    # 对象转字典
+    data = role.to_dict()
+    # 返回结果
+    return data
+# 添加角色
+
+def add_role(request, data: RoleForm):
+    db = get_db_session()
+    # 唯一性校验
+    if db.query(Role).filter(and_(Role.name == data.name, Role.is_delete == 0)).first():
+        return R.failed("角色名称不能重复")
+    # 表单数据赋值给对象
+    role = Role(**data.model_dump(exclude={'id'}))
+    role.create_user = get_user_id(request)
+    # 插入数据
+    role.save()
+    # 返回结果
+    return R.ok(msg="添加成功")
+
+
+# 更新角色
+
+def update_role(request, data: RoleForm):
+    db = get_db_session()
+    # 记录ID判空
+    if not data.id or data.id <= 0:
+        return R.failed("记录ID不能为空")
+    # 唯一性校验（排除自身）
+    if db.query(Role).filter(and_(Role.name == data.name, Role.id != data.id, Role.is_delete == 0)).first():
+        return R.failed("角色名称不能重复")
+
+    # 根据ID查询记录
+    role = db.query(Role).filter(and_(Role.id == data.id, Role.is_delete == 0)).first()
+    # 查询结果判空
+    if not role:
+        return R.failed("记录不存在")
+    # 构造更新数据（排除 id）
+    updData = data.model_dump(exclude={'id'})
+    updData['update_user'] = get_user_id(request)
+    updData['update_time'] = datetime.now(tz=timezone.utc)
+    result = db.query(Role).filter_by(id=data.id).update(updData)
+    # 提交数据
+    if not result:
+        return R.failed("更新失败")
+    # 返回结果
+    return R.ok(msg="更新成功")
+# 删除角色
+
+async def delete_role(role_id):
+    def _delete():
+        db = get_db_session()
+        # 记录ID为空判断
+        if not role_id:
+            return R.failed("记录ID不存在"), None
+        # 分裂字符串
+        id_list = parse_id_list(role_id)
+        if not id_list:
+            return R.failed("记录ID不存在"), None
+        # 批量查询并校验
+        records = db.query(Role).filter(and_(Role.id.in_(id_list), Role.is_delete == 0)).all()
+        if len(records) != len(id_list):
+            return R.failed("记录不存在"), None
+        # 批量设置删除标识
+        for record in records:
+            record.is_delete = 1
+        return None, (id_list, records)
+
+    # 同步DB操作放入线程池，避免阻塞事件循环
+    err, result = await asyncio.to_thread(_delete)
+    if err:
+        return err
+    id_list, records = result
+
+    # 清理被删角色下所有用户的权限缓存
+    for rid in id_list:
+        await invalidate_role_perms(rid)
+    # 返回结果
+    return R.ok(msg="本次共删除{0}条数据".format(len(records)))
+# 设置状态
+
+def update_role_status(request, data: RoleStatusForm):
+    db = get_db_session()
+    # 根据ID查询记录
+    role = db.query(Role).filter(and_(Role.id == data.id, Role.is_delete == 0)).first()
+    # 查询结果判空
+    if not role:
+        return R.failed("记录不存在")
+
+    # 更新记录
+    result = db.query(Role).filter_by(id=data.id).update({
+        "status": data.status,
+        "update_user": get_user_id(request),
+        "update_time": datetime.now(tz=timezone.utc)
+    })
+    # 提交数据
+    if not result:
+        return R.failed("设置失败")
+    # 返回结果
+    return R.ok(msg="设置成功")
+# 获取角色列表
+
+def get_role_list():
+    db = get_db_session()
+    # 查询角色列表
+    role_rows = db.query(Role).filter(and_(Role.is_delete == 0, Role.status == 1)).order_by(Role.sort.asc()).all()
+    # 实例化角色列表
+    role_list = []
+    # 遍历数据源
+    if role_rows:
+        for v in role_rows:
+            # 对象转字典
+            item = v.to_dict()
+            # 加入数组
+            role_list.append(item)
+    # 返回结果
+    return role_list