def post(self):
parse = reqparse.RequestParser()
parse.add_argument('mobile',
location='json',
required=True,
type=check_mobile)
parse.add_argument('smscode', location='json', required=True)
parse.add_argument('password', location='json', required=True)
args = parse.parse_args()
mobile = args.get('mobile')
smscode = args.get('smscode')
password = args.get('password')
try:
server_smscode = current_app.redis_store.get('sms_%s' % mobile)
except Exception as e:
current_app.logger.error(e)
return jsonify(errno=RET.DBERR, errmsg="获取本地验证码失败")
if not server_smscode:
# 短信验证码过期
return jsonify(errno=RET.NODATA, errmsg="短信验证码过期")
if smscode != server_smscode.decode():
return jsonify(errno=RET.DATAERR, errmsg="短信验证码错误")
try:
current_app.redis_store.delete('sms_%s' % mobile)
except Exception as e:
current_app.logger.error(e)
user = User()
user.nick_name = mobile
user.mobile = mobile
user.password = password
try:
db.session.add(user)
db.session.commit()
except Exception as e:
db.session.rollback()
current_app.logger.error(e)
return jsonify(errno=RET.DATAERR, errmsg="数据保存错误")
# 状态保持
session["user_id"] = user.id
session["nick_name"] = user.nick_name
session["mobile"] = user.mobile
return jsonify(errno=RET.OK, errmsg="OK")
def createsuperuser(name, password):
"""创建管理员用户"""
if not all([name, password]):
print('参数不足')
return
user = User()
user.mobile = name
user.nick_name = name
user.password = password
user.is_admin = True
try:
db.session.add(user)
db.session.commit()
print("创建成功")
except Exception as e:
print(e)
db.session.rollback()
