开篇闲聊
昨天在JD买了个跳绳的玩意，打算后面好好锻炼下身体。平时久坐又加上不怎么运动，导致各种小毛病还是挺多的，建议大家也在学习之余，多放松下身体。比如像某榜一大哥带妹跑步，带妹羽毛球，都是很不错的「锻炼方式」。对于我这种懒人，就买根绳子跳一跳吧，不然可能上个6楼都得喘气了！
回顾
上一节咱们定制好了简单的权限系统，也设计了相关的表概念。但是表只有对应的model，并没有编写实际的操作接口，所以今天我们来完善他们！
上次我们编写的「TestCase」表存在一点问题，首先是没有params字段，也没有body字段，当然可能还缺少很多字段，我们可以在后续的开发过程中陆续补充。
在dao/「init」.py加入新创建的表
修改TestCase表
编写项目和项目角色的增删改查接口
首先咱们明确一个要求，因为项目有公有和私有的概念，所以对于用户能看到的项目应该是这样的逻辑:
普通用户
能看到
公有项目和私有项目(用户在该成员中，成员或创建者)管理员
能看到所有项目
由于我们需要通过用户id拿到对应所属的project，所以我们先编写project_role相关接口
编写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或者说他们不是非数字，那么我们就给个默认值。
篇幅有限，这篇内容先到这里吧！再不去做饭要饿死了！