def change_info(message):
"""
修改个人信息
:return:
"""
if message['code'] != Code.SUCCESS.value:
return jsonify(message)
params = request.values.to_dict()
passwords = set_password(params['password'])
sys_user = SysUser(params['username'], passwords, params['email'],
params['avatar'])
result = SysUser.update(sys_user)
if result is None:
# 为空说明存入数据库没有报错
sys_user = SysUser.get_info(params['username'])
token = Verificate.encode_auth_token(
sys_user.id, sys_user.last_login.strftime("%Y-%m-%d %H:%M:%S"))
data = {
'info': {
'username': sys_user.username,
'password': sys_user.password,
'email': sys_user.email,
'avatar': sys_user.avatar
},
'token': token.decode()
}
return jsonify(
response.return_message(data, Message.SUCCESS.value,
Code.SUCCESS.value))
else:
return jsonify(
response.return_message(None, Message.BAD_REQUEST.value,
Code.BAD_REQUEST.value))
def reset_pwd():
"""
重置密码
:return:
"""
params = request.values.to_dict()
passwords = None if params['password'] == '' else set_password(
params['password'])
sys_user = SysUser(params['username'], passwords, params['email'],
params['avatar'])
result = SysUser.update(sys_user)
if result is None:
return jsonify(
response.return_message(None, Message.SUCCESS.value,
Code.SUCCESS.value))
else:
return jsonify(
response.return_message(None, Message.BAD_REQUEST.value,
Code.BAD_REQUEST.value))
def register():
"""
注册管理员
:return: 用户基本信息
"""
username = request.values.get('username')
password = set_password(request.values.get('password'))
email = request.values.get('email')
avatar = request.values.get('avatar')
sys_user = SysUser(username, password, email, avatar)
result = SysUser.save(sys_user)
if result is None:
sys_user = SysUser.get_info(username)
data = {"id": sys_user.id, "username": username}
return jsonify(
response.return_message(data, Message.REGISTER_SUCCESS.value,
Code.SUCCESS.value))
else:
return jsonify(
response.return_message(None, Message.REGISTER_FAILED.value,
Code.BAD_REQUEST.value))
def reset_pwd():
"""
重置密码
:return:
"""
params = request.values.to_dict()
password = params['password']
if password is None or len(password) < Constant.PASSWORD_LENGTH.value:
return jsonify(
response.return_message(None,
Message.PASSWORD_LENGTH_LESS_THAN.value,
Code.BAD_REQUEST.value))
passwords = set_password(password)
result = SysUser.reset_password(params['username'], passwords)
if result is None:
return jsonify(
response.return_message(None, Message.SUCCESS.value,
Code.SUCCESS.value))
else:
return jsonify(
response.return_message(None, Message.RESET_PASSWORD_FAILED.value,
Code.BAD_REQUEST.value))