开篇闲聊
昨天在JD买了个跳绳的玩意，打算后面好好锻炼下身体。平时久坐又加上不怎么运动，导致各种小毛病还是挺多的，建议大家也在学习之余，多放松下身体。比如像某榜一大哥带妹跑步，带妹羽毛球，都是很不错的锻炼方式。对于我这种懒人，就买根绳子跳一跳吧，不然可能上个6楼都得喘气了！
回顾
上一节咱们定制好了简单的权限系统，也设计了相关的表概念。但是表只有对应的model，并没有编写实际的操作接口，所以今天我们来完善他们！
上次我们编写的TestCase表存在一点问题，首先是没有params字段，也没有body字段，当然可能还缺少很多字段，我们可以在后续的开发过程中陆续补充。
在dao/init.py加入新创建的表
image
修改TestCase表
加入params和body字段
编写项目和项目角色的增删改查接口
首先咱们明确一个要求，因为项目有公有和私有的概念，所以对于用户能看到的项目应该是这样的逻辑:
- 普通用户
能看到公有项目和私有项目(用户在该成员中，成员或创建者) - 管理员
能看到所有项目由于我们需要通过用户id拿到对应所属的project，所以我们先编写project_role相关接口
设计表的时候漏掉了，这里补上user_id
编写app/dao/project/ProjectRoleDao.py
from app.models import db from app.models.project_role import ProjectRole from app.utils.logger import Log class ProjectRoleDao(object): log = Log("ProjectRoleDao") @staticmethod def list_project_by_user(user_id): """ 通过user_id获取项目列表 :param user_id: :return: """ try: projects = ProjectRole.query.filter_by(user_id=user_id, deleted_at=None).all() return [p.id for p in projects], None except Exception as e: ProjectRoleDao.log.error(f"查询用户: {user_id}项目失败, {e}") return [], f"查询项目失败, {e}" @staticmethod def add_project_role(user_id, project_id, project_role, user): """ 为项目添加用户 :param user_id: 用户id :param project_id: 项目id :param project_role: 用户角色 :param user: 创建人 :return: """ try: role = ProjectRole.query.filter_by(user_id=user_id, project_id=project_id, project_role=project_role, deleted_at=None).first() if role is not None: # 说明角色已经存在了 return "该用户已存在" role = ProjectRole(user_id, project_id, project_role, user) db.session.add(role) db.session.commit() except Exception as e: ProjectRoleDao.log.error(f"添加项目用户失败, {e}") return f"添加项目用户失败, {e}" return None
这边暂时编写了2个接口，1个是通过用户获取到用户的项目，另一个是为项目添加用户，代码很简单，也写了一些注释。由于暂时我们需要快速做一个展示，所以暂时提供，增和查功能即可，后续的接口再慢慢补充。
编写app/dao/project/ProjectRole.py
由于我们已经可以通过user_id获取到他有权限的项目，那么接下来我们把它和共有项目做一个整合就可以了。因为project不可能太多，所以这边我们直接用in操作符，先预想一下sql:
select * from project where deleted_at is null and (owner = user or project_id in (已经查出的用户项目) or private is false);
首先必须是未删除的，其次是owner=user的，再就是非私有或者用户有权限的。
如果对于超级管理员来说的话，那么他们对于所有项目都是可见的。
from sqlalchemy import or_ from app import pity from app.dao.project.ProjectRoleDao import ProjectRoleDao from app.models import db from app.models.project import Project from app.utils.logger import Log class ProjectDao(object): log = Log("ProjectDao") @staticmethod def list_project(user, role, page, size, name=None): """ 查询/获取项目列表 :param user: 当前用户 :param role: 当前用户角色 :param page: 当前页码 :param size: 当前size :param name: 项目名称 :return: """ try: search = [Project.deleted_at == None] if role != pity.config.get("ADMIN"): project_list = ProjectRoleDao.list_project_by_user(user) search.append(or_(Project.id in project_list, Project.owner == user, Project.private == False)) if name: search.append(Project.name.ilike("%{}%".format(name))) data = Project.query.filter(*search) total = data.count() return data.order_by(Project.created_at.desc()).paginate(page, per_page=size).items, total, None except Exception as e: ProjectDao.log.error(f"获取用户: {user}项目列表失败, {e}") return 0, 0, f"获取用户: {user}项目列表失败, {e}" @staticmethod def add_project(name, owner, user, private): try: data = Project.query.filter_by(name=name, deleted_at=None).first() if data is None: return "项目已存在" pr = Project(name, owner, user, private) db.session.add(pr) db.session.commit(pr) except Exception as e: ProjectDao.log.error(f"新增项目: {name}失败, {e}") return 0, 0, f"新增项目: {name}失败, {e}"
这里重点关注list_project，里面先判断用户是不是ADMIN，如果不是，则查询一下这个用户拥有的项目，等于说是转换了我们刚才写的sql。
最后通过项目的创建时间进行了排序，并进行了分页并返回了项目总数，第三个参数是error信息。
编写具体的接口
- /project/insertProject
- /project/listProject
- /project/insertProjectRole
编写获取分页的相关方法app/handler/page.py
from flask import request # 默认页数和页码 PAGE = 1 SIZE = 10 class PageHandler(object): @staticmethod def page(): """ 获取page和size :return: """ page = request.args.get("page") if page is None or not page.isdigit(): page = PAGE size = request.args.get("size") if size is None or not size.isdigit(): size = SIZE return int(page), int(size)
如果没有获取到page/size或者说他们不是非数字，那么我们就给个默认值。
