测试平台系列(20) 编写项目的增删改查接口和页面(2)

原创

丶米洛丶 2021-09-07 10:32:18 ©著作权

文章标签 json 用户信息数据库 python 创建项目 文章分类 代码人生

©著作权归作者所有：来自51CTO博客作者丶米洛丶的原创作品，请联系作者获取转载授权，否则将追究法律责任

回顾

上回说到，编写项目权限相关接口，但是我们只完成了核心方法的编写，这次我们就来定义相关的接口。

编写接口

先实现「项目」的查询和新增接口，接着实现项目角色的添加接口。

编写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: