|
|
#!/usr/bin/env python
|
|
|
# -*- encoding: utf-8 -*-
|
|
|
'''
|
|
|
@Date:2022/08/01 17:04:40
|
|
|
'''
|
|
|
import sys
|
|
|
sys.path.append('.')
|
|
|
|
|
|
from sqlalchemy import and_, or_
|
|
|
from db_logic.module import BllModule
|
|
|
from db_logic.role import BllRole
|
|
|
from Common.auth import token_auth
|
|
|
from flask import Blueprint, jsonify, request, g
|
|
|
from db_logic.module_relation import BllModuleRelation
|
|
|
|
|
|
from db_logic.user import BllUser
|
|
|
|
|
|
from models.power_models import EntityModuleRelation
|
|
|
|
|
|
from models.user_models import EntityRole, EntityUser
|
|
|
from Common.Utils import PageParam, Utils
|
|
|
|
|
|
|
|
|
power_router = Blueprint('power', __name__)
|
|
|
|
|
|
# 获取用户权限
|
|
|
@power_router.route("/get_user_power_list", methods=["GET", "POST"])
|
|
|
@token_auth.login_required
|
|
|
def get_user_power_list():
|
|
|
user_id = request.values.get("user_id")
|
|
|
if user_id:
|
|
|
user_info = BllUser().findEntity(EntityUser.user_id == user_id)
|
|
|
data_list = BllModuleRelation().get_user_module_list(user_info)
|
|
|
return jsonify(Utils.true_return(data=data_list))
|
|
|
return jsonify(Utils.false_return(msg="用户id不能为空"))
|
|
|
|
|
|
|
|
|
# 分配用户权限
|
|
|
@power_router.route("/add_user_power", methods=["GET", "POST"])
|
|
|
@token_auth.login_required
|
|
|
def add_user_power_list():
|
|
|
user_id = request.values.get("user_id")
|
|
|
drug_module = request.values.getlist("drug_module_id")
|
|
|
standard_module = request.values.getlist("standard_module_id")
|
|
|
consumables_module = request.values.getlist("consumables_module_id")
|
|
|
instrument_modulet = request.values.getlist("instrument_module_id")
|
|
|
inster_dic = {
|
|
|
"2": drug_module if drug_module else [],
|
|
|
"3": standard_module if standard_module else [],
|
|
|
"4": consumables_module if consumables_module else [],
|
|
|
"5": instrument_modulet if instrument_modulet else [],
|
|
|
}
|
|
|
# 删除除去柜子权限外的所有权限
|
|
|
BllModuleRelation().delete(
|
|
|
and_(EntityModuleRelation.object_id == user_id,
|
|
|
EntityModuleRelation.module_type != "1",
|
|
|
EntityModuleRelation.object_type != "1")
|
|
|
)
|
|
|
inster_module_list = []
|
|
|
for k,v in inster_dic.items():
|
|
|
for i in v:
|
|
|
inster_module_list.append(
|
|
|
EntityModuleRelation(
|
|
|
object_type=2,
|
|
|
object_id=user_id,
|
|
|
module_id=i,
|
|
|
module_type=k,
|
|
|
create_date=Utils.get_str_datetime(),
|
|
|
create_user_id=g.current_user.user_id,
|
|
|
create_user_name=g.current_user.real_name
|
|
|
)
|
|
|
)
|
|
|
BllModuleRelation().insert_many(inster_module_list)
|
|
|
return jsonify(Utils.true_return())
|
|
|
|
|
|
|
|
|
@power_router.route("/user_power", methods=["GET", "POST"])
|
|
|
@token_auth.login_required
|
|
|
def get_login_user_power_list():
|
|
|
user_info = g.current_user
|
|
|
# user_id = '4cea74ee-0d8b-11ed-943e-f47b094925e1'
|
|
|
# user_info = BllUser().findEntity(user_id)
|
|
|
module_list = BllModuleRelation().get_login_user_module_list(
|
|
|
user_id=user_info.user_id, role_id=user_info.role_id)
|
|
|
module_list = Utils.msyql_table_model(module_list)
|
|
|
return jsonify(Utils.true_return(data=module_list))
|
|
|
|
|
|
# 获取角色列表
|
|
|
@power_router.route("/get_list", methods=["GET", "POST"])
|
|
|
@token_auth.login_required
|
|
|
def get_power_list():
|
|
|
# 根据角色表进行关联查询，
|
|
|
# 思路：角色表角色id->模块权限表（object_id）-> 终端权限表（）
|
|
|
# | |
|
|
|
# role_id=object_id module_id=module_id
|
|
|
page = int(request.values.get("page"))
|
|
|
page_size = int(request.values.get("page_size"))
|
|
|
page_param = PageParam(page, page_size)
|
|
|
data_list = BllModuleRelation().get_role_module(page_param)
|
|
|
return jsonify(Utils.true_return(data=Utils.msyql_table_model(data_list)))
|
|
|
|
|
|
# 通过角色id查询角色列表
|
|
|
@power_router.route("/get_role_module", methods=["GET", "POST"])
|
|
|
@token_auth.login_required
|
|
|
def get_role_module_list():
|
|
|
role_id = request.values.get("role_id")
|
|
|
data_list = BllModuleRelation().get_role_module_list(role_id)
|
|
|
return jsonify(Utils.true_return(data=Utils.msyql_table_model(data_list)))
|
|
|
|
|
|
|
|
|
# 获取模块列表
|
|
|
@power_router.route("/get_module_list", methods=["GET", "POST"])
|
|
|
# @token_auth.login_required
|
|
|
def get_module_list():
|
|
|
data_list = BllModule().get_module_list()
|
|
|
return jsonify(Utils.true_return(data=data_list))
|
|
|
|
|
|
# 添加角色
|
|
|
@power_router.route("/add_update_role", methods=["GET", "POST"])
|
|
|
@token_auth.login_required
|
|
|
def add_role():
|
|
|
role_id = request.values.get("role_id")
|
|
|
role_code = request.values.get("role_code")
|
|
|
role_name = request.values.get("role_name")
|
|
|
sys_module_list = request.values.getlist("sys_module")
|
|
|
client_module_list = request.values.getlist("client_module")
|
|
|
description = request.values.get("description")
|
|
|
# 获取角色代码或者角色名的角色信息
|
|
|
obj = BllRole().findEntity(
|
|
|
or_(EntityRole.role_code == role_code, EntityRole.role_name == role_name))
|
|
|
# 判断是创建角色
|
|
|
if not role_id:
|
|
|
# 判断信息重复
|
|
|
if obj:
|
|
|
return jsonify(Utils.false_return(msg="角色代码或角色名重复，创建失败"))
|
|
|
else:
|
|
|
# 创建角色
|
|
|
role_max_obj = BllRole().execute("select max(sort_index) as sort_index from rms_role").fetchone()
|
|
|
role_id=Utils.UUID()
|
|
|
role_obj = EntityRole(
|
|
|
role_id=role_id,
|
|
|
role_code=role_code,
|
|
|
role_name=role_name,
|
|
|
role_level=1,
|
|
|
description=description,
|
|
|
sort_index=int(role_max_obj.sort_index or '0') + 1
|
|
|
)
|
|
|
BllRole().insert(role_obj)
|
|
|
else:
|
|
|
# 修改角色信息
|
|
|
role_obj = BllRole().findEntity(role_id)
|
|
|
# 判断修改的信息是否重复
|
|
|
if not obj or role_obj.role_id == role_id:
|
|
|
role_obj.role_code=role_code,
|
|
|
role_obj.role_name=role_name,
|
|
|
role_obj.description=description,
|
|
|
BllRole().update(role_obj)
|
|
|
else:
|
|
|
return jsonify(Utils.false_return(msg="角色代码或角色名重复，修改失败"))
|
|
|
# 删除原有的权限，每次重新创建
|
|
|
BllModuleRelation().delete(EntityModuleRelation.object_id==role_id)
|
|
|
inster_module_list = []
|
|
|
if sys_module_list:
|
|
|
sys_module_list = eval(sys_module_list)
|
|
|
for module_id in sys_module_list:
|
|
|
inster_module_list.append(
|
|
|
EntityModuleRelation(
|
|
|
object_type=1,
|
|
|
object_id=role_id,
|
|
|
module_id=module_id,
|
|
|
module_type=2,
|
|
|
create_date=Utils.get_str_datetime()
|
|
|
)
|
|
|
)
|
|
|
if client_module_list:
|
|
|
client_module_list = eval(client_module_list)
|
|
|
for module_id in client_module_list:
|
|
|
inster_module_list.append(
|
|
|
EntityModuleRelation(
|
|
|
object_type=1,
|
|
|
object_id=role_id,
|
|
|
module_id=module_id,
|
|
|
module_type=1,
|
|
|
create_date=Utils.get_str_datetime()
|
|
|
)
|
|
|
)
|
|
|
BllModuleRelation().insert_many(inster_module_list)
|
|
|
return jsonify(Utils.true_return())
|
|
|
|
|
|
# 删除角色
|
|
|
@power_router.route("/del_role", methods=["GET", "POST"])
|
|
|
@token_auth.login_required
|
|
|
def remove_role():
|
|
|
role_id = request.values.get("role_id")
|
|
|
if role_id:
|
|
|
BllRole().delete(EntityRole.role_id==role_id)
|
|
|
BllModuleRelation().delete(EntityModuleRelation.object_id==role_id)
|
|
|
return jsonify(Utils.true_return(msg="删除成功"))
|
|
|
return jsonify(Utils.false_return(msg="请选择角色")) |
