测试平台系列(24) 编写成员管理功能

2021-04-12 本文已影响0人米洛丶

回顾

上一节我们编写了项目设置相关功能，还剩下2个tab没有编写，分别是: 用例列表和成员管理.

编写成员列表及部分接口

添加成员
搜索成员
编辑成员
删除成员

今天因为时间关系就不展示删除成员了，只编写编辑/新增成员接口和页面。

编写新增/编辑角色的方法

由于之前我们已经编辑过新增角色的方法了，但是需要修改一点:

image

这里角色的筛选条件不能加上，为什么呢？因为加上的话，我可以创建同一个user_id的不同角色，比如一个组长一个组员。所以我们只要user_id在数据表里出现过，就视为重复添加。

新增编辑方法

编辑方法流程会比较多，首先我们需要判断用户是否是超管，如不是，则判断他是不是owner，如果也不是，则判断他是不是组长，如果是则判断他要修改的用户是不是组长，用户是组长则不让他修改！

    @staticmethod
    def update_project_role(role_id, project_role, user, user_role):
        try:
            role = ProjectRole.query.filter_by(id=role_id, deleted_at=None).first()
            if role is None:
                return "该用户角色不存在"
            if user_role != pity.config.get("ADMIN"):
                project = Project.query.filter_by(id=role.project_id).first()
                if project is None:
                    return "该项目不存在"
                if project.owner != user and role.project_role == 1:
                    # 说明既不是owner也不是超管，无法修改阻止的权限
                    return "不能修改其他组长的权限"
            updater_role = ProjectRole.query.filter_by(user_id=user, project_id=role.project_id, deleted_at=None).first()
            if updater_role is None or updater_role.project_role == 1:
                return "对不起，你没有权限"
            role.project_role = project_role
            role.updated_at = datetime.now()
            role.update_user = user
            db.session.commit()
        except Exception as e:
            ProjectRoleDao.log.error(f"修改项目用户失败, {e}")
            return f"修改项目用户失败, {e}"
        return None

判断他要修改的角色id是否存在
判断是否是超管或者owner
判断要修改的是不是组长
判断他自身是否是组长

这些校验都通过以后，才允许他修改组员的权限。

补全具体的接口方法

@pr.route("/role/insert", methods=["POST"])
@permission()
def insert_project_role(user_info):
    try:
        data = request.get_json()
        if data.get("user_id") is None or data.get("project_role") is None or data.get("project_id") is None:
            return jsonify(dict(code=101, msg="请求参数有误"))
        err = ProjectRoleDao.add_project_role(data.get("user_id"), data.get("project_id"), data.get("project_role"),
                                              user_info["id"])
        if err is not None:
            return jsonify(dict(code=110, msg=err))
    except Exception as e:
        return jsonify(dict(code=110, msg=str(e)))
    return jsonify(dict(code=0, msg="操作成功"))


@pr.route("/role/update", methods=["POST"])
@permission()
def update_project_role(user_info):
    try:
        data = request.get_json()
        if data.get("user_id") is None or data.get("project_role") is None or data.get("project_id") is None \
                or data.get("id") is None:
            return jsonify(dict(code=101, msg="请求参数有误"))
        err = ProjectRoleDao.update_project_role(data.get("id"), data.get("project_role"),
                                                 user_info["id"], user_info["role"])
        if err is not None:
            return jsonify(dict(code=110, msg=err))
    except Exception as e:
        return jsonify(dict(code=110, msg=str(e)))
    return jsonify(dict(code=0, msg="操作成功"))

2段代码比较接近，基本就是校验参数+调用核心方法。