回顾
上回说到，编写项目权限相关接口，但是我们只完成了核心方法的编写，这次我们就来定义相关的接口。
编写接口
先实现项目的查询和新增接口，接着实现项目角色的添加接口。
编写app/controllers/project/project.py
from flask import Blueprint, request, jsonify
from app import pity
from app.dao.project.ProjectDao import ProjectDao
from app.handler.fatcory import ResponseFactory
from app.handler.page import PageHandler
from app.utils.decorator import permission
pr = Blueprint("project", __name__, url_prefix="/project")
@pr.route("/list")
@permission()
def list_project(user_info):
"""
获取项目列表
:param user_info:
:return:
"""
page, size = PageHandler.page()
user_role, user_id = user_info["role"], user_info["id"]
name = request.args.get("name")
result, total, err = ProjectDao.list_project(user_id, user_role, page, size, name)
if err is not None:
return jsonify(dict(code=110, data=result, msg=err))
return jsonify(dict(code=0, data=ResponseFactory.model_to_list(result), msg="操作成功"))
@pr.route("/insert", methods=["POST"])
@permission(pity.config.get("MANAGER"))
def insert_project(user_info):
try:
user_id = user_info["id"]
data = request.get_json()
if not data.get("name") or not data.get("owner"):
return jsonify(dict(code=101, msg="项目名称/项目负责人不能为空"))
private = data.get("private", False)
err = ProjectDao.add_project(data.get("name"), data.get("owner"), user_id, private)
if err is not None:
return jsonify(dict(code=110, msg=err))
return jsonify(dict(code=0, msg="操作成功"))
except Exception as e:
return jsonify(dict(code=111, msg=str(e)))
这边基本上很简单，流程就是 获取参数->校验参数->调用方法，获取返回结果，成功失败返回不同的response.
注意到代码里面有一处bug:
- 把蓝图加入run.py
测试一下
这里我们先调用了登录接口拿到了token:
仔细查看原因，发现是这里出了问题:
可以看到在用Python写多个返回值的时候，一定要记得返回多少就用多少，不然IDE并不会给出提示。这也是Python的一个小缺点吧，当然它也可以指定返回类型，只不过我觉得复杂了。
由于err是我们预期的err，并且我们有做try/except处理，所以我们直接raise，抛出异常即可。
- 再次尝试
测试添加项目接口
除了token以外我们需要3个数据: 项目名字、项目管理员、项目权限。
这边我直接偷懒，去数据库查看有哪些用户:
可以看到user_id从1到4都有，我这边选4(李逍遥)当做我这个测试项目的owner。
可以看到没有创建成功，提示我权限不足。
还记得之前设定好的权限系统吗？因为我把这个接口的权限设置为manager也就是经理才能创建项目。所以在我们还没有完成用户管理页面的时候，我们先在数据库软件或用sql去手动修改:
接着我们需要重新登录，获取新token。
可能有的读者发现一个问题，这样如果改动了用户的权限，用户还需要重新登录，为什么不能在解析用户信息的时候，重新获取一遍用户信息呢？这是一个很好的问题，问就是两者都可以呀，都没毛病的。你甚至可以把用户信息缓存到redis，不用每次都读db。
- 重新测试
可以看到我们的代码写的太随意，导致if写错了，修正如下:
- 重启服务后测试
又报错了，commit是一个不接受参数的方法，所以我们直接去掉。而且还有一个问题，就是返回值我是复制的上一个方法的，导致多了2个0。我们一并处理掉:
- 再次测试
- 通过project/list接口去查看
所以我们编写代码需要仔细一点，就算是复制/粘贴也要更细心，不能像笔者一样马虎！
后续测试
其实我们这边的接口是分不同场景的，不同的角色需要不同的测试场景，我们这边设计用例的时候还需要考虑普通用户的项目权限问题，这些就留给读者课后去测试了。
今天的内容就到这里了，只写了2个接口，下一篇开始继续回到前端，写相关的页面！如果觉得每篇篇幅较短的话，可以给我提建议，我争取满足大家~