def del_user_allrole_lib(self, userid):
"""13删除用户的所有角色"""
user = User.by_id(userid)
user.roles = []
self.db.add(user)
self.db.commit()
flash(self, '用户角色删除成功', 'success')
def del_user_role_lib(self, userid):
user = User.by_id(userid)
if user is None:
return
user.roles = []
self.db.add(user)
self.db.commit()
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_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 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 del_user_role_lib(self, userid):
'''12 删除用户角色'''
user = User.by_id(userid)
if user is None:
flash(self, '用户角色不存在，删除失败！', 'error')
return
self.db.delete(user)
self.db.commit()
flash(self, '用户删除成功！', 'success')
def del_user_lib(self, uuid):
"""12为用户删除角色"""
user = User.by_id(uuid)
if user is None:
return
user.roles = []
self.db.add(user)
self.db.commit()
flash(self, "为用户删除角色删除成功", "success")
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_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_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_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 del_user_dev_role_lib(self, userid, roleid):
"""14为用户删除角色"""
try:
user = User.by_id(userid)
role = Role.by_id(roleid)
if user is None or role is None:
flash(self, '用户或者角色不存在', 'error')
user.roles.remove(role) # 添加到中间表
self.db.add(user)
self.db.commit()
flash(self, '用户删除角色成功', 'success')
except Exception as e:
print e
flash(self, '用户删除角色失败', 'error')
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 del_user_role_lib(self, userid):
"""12删除用户角色"""
try:
user = User.by_id(userid)
if user is None:
flash(self, '用户不存在', 'error')
# []
user.roles = []
# user.roles.remove(obj) # 删除一个对象
# user.roles.pop() # 弹出最后一个
self.db.add(user)
self.db.commit()
flash(self, '用户删除角色成功', 'success')
except Exception as e:
print e
flash(self, '用户删除角色失败', 'error')
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')
def auth_email_libs(self, email, email_code, u):
"""验证邮箱验证码"""
redis_text = self.conn.get('email:%s' % email)
if redis_text:
text_dict = json.loads(redis_text)
if text_dict and text_dict['email_code'] == email_code:
user = self.current_user
if not user:
user = User.by_id(text_dict[u])
print user
user.email = email
user.update_time = datetime.now()
self.db.add(user)
self.db.commit()
return {'status': True, 'msg': "邮箱修改成功"}
return {'status': False, 'msg': "验证码错误"}
return {'status': False, 'msg': "验证码过期"}
def auth_email_libs(self, email_code, email, u):
"""验证邮箱验证码"""
redis_text = self.conn.get("email:%s" % email)
if redis_text:
text_dict = json.loads(redis_text)
if text_dict and text_dict['email_code'] == email_code:
user = self.current_user
if not user:
user = User.by_id(text_dict[u])
# 为什么需要by_id呢？
# 因为有可能用户永别的浏览器打开邮件，重新登陆，
# 所以我们就获取不到current_user，需要by_id来重新获取到current_user
user.email = email
user.update_time = datetime.now()
self.db.add(user)
self.db.commit()
return {'status': True, 'msg': '邮箱修改成功'}
return {'status': False, 'msg': '邮箱验证码不正确'}
return {'status': False, 'msg': '邮箱验证码已过期,请重新绑定'}
def auth_email_libs(self, email, email_code, u):
"""验证邮箱验证码"""
"""
获取redis中对应的值"email:[email protected]"
"{"5ec36183-4f12-43f4-81e9-a7ba871d1eb6": 2, "email_code": "FHFx"}"
"""
redis_text = self.conn.get('email:%s' % email)
if redis_text: # 判断如果没有值 则redis过期
text_dict = json.loads(redis_text) # dumps是将dict转化成str格式，loads是将str转化成dict格式。
if text_dict and text_dict['email_code'] == email_code: # 如果键值对存在 并且 验证码和url中传的值相同
user = self.current_user
if not user: # 可能打开了其他的浏览器 并不存在缓存 所以需要重新进行查询
user = User.by_id(text_dict[u])
# dbSession.query(cls).filter_by(id=id).first() # 如果model内没有用封装类方法 就得这么写
# print user
user.email = email
user.update_time = datetime.now()
self.db.add(user)
self.db.commit()
return {'status': True, 'msg': '邮箱修改成功'}
return {'status': False, 'msg': '验证码错误'}
return {'status': False, 'msg': '验证码过期'}
def del_user(self, user_id):
''' 删除用户 '''
user = User.by_id(user_id)
self.db.delete(user)
self.db.commit()
return {'status': True}