def add_role_lib(self, name):
"""02添加角色函数"""
role= Role.by_name(name)
if role is not None:
flash(self, "角色已经存在，添加失败", "error")
return
role = Role()
role.name = name
self.db.add(role)
self.db.commit()
flash(self, "角色添加成功", "success")
def permission_manager_list_lib(self):
"""01权限管理页面函数"""
roles = Role.all()
permissions = Permission.all()
menus = Menu.all()
handlers = Handler.all()
users = User.all()
#研发员工
dev_role = Role.by_name('研发员工')
dev_users = dev_role.users if dev_role else []
return roles, permissions, menus, handlers, users, dev_users, dev_role.id
def save_role_name(self, role_name):
''' 存入新增角色名 '''
if role_name == '':
flash(self, "角色不能为空", "error")
return
role = Role.by_name(role_name)
if role is not None:
flash(self, "角色已经存在", "error")
return
role = Role()
role.name = role_name
self.db.add(role)
self.db.commit()
flash(self, "角色添加成功", "success")
def add_role_lib(self, name):
'''02添加一个角色'''
if not name:
flash(self, '请输入角色名称！', 'error')
return
print type(name), 'name:%s' % name
role = Role.by_name(name)
if role is not None:
flash(self, '角色已经存在！', 'error')
return
role = Role()
role.name = name
self.db.add(role)
self.db.commit()
flash(self, '角色添加成功！', 'success')
def permission_manager_list_lib(self):
''' 返回给权限管理页面 '''
roles = Role.all()
permissions = Permission.all()
menus = Menu.all()
handlers = Handler.all()
users = User.all()
return roles, permissions, menus, handlers, users
def get(self):
kw = {
'user_msg': self.get_redis_json_to_dict('user'),
'role_msg': self.get_redis_json_to_dict('role'),
'system_msg': self.get_redis_json_to_dict('system'),
'roles': Role.all(),
}
self.render('message/message_send_message.html', **kw)
def permission_manage_list_lib(self):
"""01权限管理页面函数"""
roles = Role.all()
permissions = Permission.all()
menus = Menu.all()
handlers = Handler.all()
users = User.all()
# 研发员工
dev_role = Role.by_name('研发部门')
dev_users = dev_role.users if dev_role else []
# print("*****")
# print(type(dev_role))
# print(dev_role)
if dev_role is None:
return roles, permissions, menus, handlers, users, dev_users, None
return roles, permissions, menus, handlers, users, dev_users, dev_role.id
def del_user_role_lib(self, userid,roleid):
"""12删除用户的角色"""
user = User.by_id(userid)
role = Role.by_id(roleid)
user.roles.remove(role)
self.db.add(user)
self.db.commit()
flash(self, '用户角色删除成功', 'success')
def del_role_lib(self, roleid):
"""03删除角色"""
role = Role.by_id(roleid)
if role is None: #如果role是空的话，我们就什么也不操作
flash(self, "角色不存在")
return
self.db.delete(role)
self.db.commit()
flash(self, "角色删除成功", "success")
def del_role_lib(self, roleid):
"""03删除角色"""
role= Role.by_id(roleid)
if role is None:
flash(self, "角色删除失败", "error")
return
self.db.delete(role)
self.db.commit()
flash(self, "角色删除成功", "success")
def add_role_permission_lib(self, permissionid, roleid):
permission = Permission.by_id(permissionid)
role = Role.by_id(roleid)
if permission is None or role is None:
return
permission.roles.append(role) #多对多关系添加
self.db.add(permission)
self.db.commit()
def del_user_dev_role_lib(self, userid, roleid):
"""14为用户删除角色"""
user = User.by_id(userid)
role = Role.by_id(roleid)
if user is None or role is None:
return
user.roles.remove(role)
self.db.add(user)
self.db.commit()
def del_role_name(self, role_id):
''' 删除角色名 '''
role = Role.by_id(role_id)
if role:
self.db.delete(role)
self.db.commit()
flash(self, "删除成功", "success")
return {'status': True}
return {'status': False, 'msg': '不存在的ID'}
def add_user_dev_role_lib(self, userid, roleid):
"""13为用户添加角色"""
user = User.by_id(userid)
role = Role.by_id(roleid)
if user is None or role is None:
return
user.roles.append(role)
self.db.add(user)
self.db.commit()
def add_permission_role_lib(self, permissionid , roleid):
"""11为角色添加权限"""
permission = Permission.by_id(permissionid)
role = Role.by_id(roleid)
if permission is None or role is None:
return
role.permissions.append(permission)
self.db.add(role)
self.db.commit()
def permission_manager_list_lib(self):
# 获取所有的角色/权限/菜单/处理器/用户
roles = Role.all()
permissions = Permission.all()
menus = Menu.all()
handlers = Handler.all()
users = User.all()
return roles, permissions, menus, handlers, users
def add_user_role_lib(self, userid, roleid):
"""10为用户添加角色"""
user = User.by_id(userid)
role = Role.by_id(roleid)
if user is None or role is None:
return
user.roles.append(role) #注意这里画龙点睛
self.db.add(user)
self.db.commit()
flash(self, "为用户添加角色添加成功", "success")
def add_user_dev_role_lib(self, userid, roleid):
"""13添加用户到研发部门"""
user = User.by_id(userid)
role = Role.by_id(roleid)
if user is None or role is None:
return
user.roles.append(role)
self.db.add(user)
self.db.commit()
flash(self, "为用户添加研发部门的角色添加成功", "success")
def add_role_permission_lib(self, roleid, permissionid):
"""11为角色添加权限"""
role = Role.by_id(roleid)
permission = Permission.by_id(permissionid)
if role is None or permission is None:
return
role.permissions.append(permission)
self.db.add(role)
self.db.commit()
flash(self, "为角色添加权限添加成功", "success")
def get(self):
self.conn.delete('massage:%s' %
self.current_user.name) # 到这个页面删除拥护邮件记录
kw = {
'roles': Role.all(),
'user_msg': self.get_redis_json_to_dict('user'),
'role_msg': self.get_redis_json_to_dict('role'),
'system_msg': self.get_redis_json_to_dict('system'),
}
self.render('message/message_send_message.html', **kw)
def add_user_role_lib(self, userid, roleid):
user = User.by_id(userid)
role = Role.by_id(roleid)
if user is None or role is None:
return
user.roles.append(role) #多对多关系添加
#role.users.append(user)
self.db.add(user)
self.db.commit()
def del_role_lib(self, roleid):
'''03删除角色'''
role = Role.by_id(roleid) # 获取用户信息
# 判断角色是否存在
if role is None:
flash(self, '角色删除失败！', 'error')
return
self.db.delete(role)
self.db.commit()
flash(self, '角色删除成功！', 'success')
def del_user_for_one_role_lib(self, userid, roleid):
"""15为用户删除某个角色"""
user = User.by_id(userid)
role = Role.by_id(roleid)
# print("********%s"%role)
if user is None or role is None:
return
user.roles.remove(role)
self.db.add(user)
self.db.commit()
flash(self, "为用户删除某个角色删除成功", "success")
def add_role_permission_lib(self, permissionid, roleid):
"""11为角色添加权限"""
permission = Permission.by_id(permissionid)
role = Role.by_id(roleid)
if permission is None or role is None:
flash(self, "为角色添加权限失败", "error")
return
permission.roles.append(role) #多对多关系添加
self.db.add(permission)
self.db.commit()
flash(self, "为角色添加权限成功", "success")
def send_role_message(cls, self, content, send_type, role_id):
role = Role.by_id(role_id)
redis_msg = cls.dict_to_json(self, content, send_type, role.name)
self.conn.rpush('message:%s' % send_type, redis_msg)
role_users = role.users
for user in role_users:
if MessageWebSocket.users.get(user.username, None) is not None:
MessageWebSocket.users[user.username].write_message(redis_msg)
else:
# self.conn.lpush("ws:role_not_online": message)
pass
def send_role_message(cls, self, content, send_type, roleid):
role = Role.by_id(roleid)
redis_msg = cls.dict_to_json(self, content, send_type, role.name)
self.conn.rpush('message:%s' % send_type, redis_msg)
role_users = role.users #[zhangsan, lishi , wangwu] [zhangsan, lishi]
for user in role_users:
if MessageWebSocket.users.get(user.name, None) is not None:
MessageWebSocket.users[user.name].write_message(redis_msg)
else:
#self.conn.lpush("ws:role_off_line",message)
pass
def add_user_role_lib(self, userid, roleid):
"""10为用户添加角色"""
user = User.by_id(userid)
role = Role.by_id(roleid)
if user is None or role is None:
flash(self, "为用户添加角色失败", "error")
return
user.roles.append(role) #这行的意思是：往多对多关系表中插入数据
#role.users.append(user)
self.db.add(user)
self.db.commit()
flash(self, "为用户添加角色成功", "success")
def user_add_role(self, user_id, role_id):
''' 给用户添加角色 '''
user = User.by_id(user_id)
if user is None:
return {'status': False, 'msg': '不存在该用户ID'}
role = Role.by_id(role_id)
if role is None:
return {'status': False, 'msg': '不存在该角色ID'}
user_role = UserToRole()
user_role.r_id = role_id
user_role.u_id = user_id
self.db.add(user_role)
self.db.commit()
return {'status': True}
def add_role_permission_lib(self, roleid, permissionid):
'''11 给角色添加权限'''
if not permissionid or not roleid:
flash(self, '请输入角色id!和权限id!', 'error')
return
permission = Permission.by_id(permissionid)
role = Role.by_id(roleid)
if permission is None or role is None:
flash(self, '角色id或权限id不存在！', 'error')
return
permission.roles.append(role) # 多对多关系添加,数组添加
print 'permission%s' % permission
self.db.add(permission)
self.db.commit()
flash(self, '给角色添加权限成功！', 'success')
def add_user_role_lib(self, userid, roleid):
'''10 给用户添加角色'''
if not userid or not roleid:
flash(self, '请输入用户id和角色id!', 'error')
return
user = User.by_id(userid)
role = Role.by_id(roleid)
if user is None or role is None:
flash(self, '用户id或角色id不存在！', 'error')
return
user.roles.append(role) # 多对多关系添加
# role.users.append(user)
self.db.add(user)
self.db.commit()
flash(self, '给用户添加角色成功！', 'success')