diff --git a/apps/power/views.py b/apps/power/views.py index 9a14048..f001336 100644 --- a/apps/power/views.py +++ b/apps/power/views.py @@ -54,7 +54,7 @@ def add_user_power_list(): BllModuleRelation().delete( and_(EntityModuleRelation.object_id == user_id, EntityModuleRelation.module_type != "1", - EntityModuleRelation.object_type != "1") + EntityModuleRelation.object_type == "2") ) inster_module_list = [] for k,v in inster_dic.items(): @@ -123,8 +123,20 @@ 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") + # sys_module_list = request.values.getlist("sys_module") + # client_module_list = request.values.getlist("client_module") + client_module = request.values.getlist("client_module_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 = { + "1": client_module if client_module else [], + "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 [], + } description = request.values.get("description") # 获取角色代码或者角色名的角色信息 obj = BllRole().findEntity( @@ -161,30 +173,43 @@ def add_role(): # 删除原有的权限,每次重新创建 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: + for k,v in inster_dic.items(): + for i in v: 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() + 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 ) ) + # 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())