一、注册接口
class Register(Resource):
def post(self):
data = request.get_json() #request.json() 获取到请求的参数，得到一个字典对象
#print(f'请求入参：{data}')
username = data.get("username")
password = data.get("password")
if not username or not password:
return jsonify({"code": 111, "msg": "用户名或密码不能为空"})
if Users.query.filter_by(username=username).first() is not None:
print('existing user')
return jsonify({"code": 222, "msg": "账号已存在"})
with current_app.app_context():
user = Users(username=username,password=password)
user.hash_password(password)
db.session.add(user)
db.session.commit()
return {'status':200,'msg':'注册成功！',"data": {"username": username}}
二、登录接口
class Login(Resource):
def post(self):
args = reqparse.RequestParser()\
.add_argument('username', type=str, location='json', required=True, help="用户名不能为空")\
.add_argument("password", type=str, location='json', required=True, help="密码不能为空")\
.parse_args()
print(f"args: {args}")
user = Users.query.filter_by(username=args.get('username')).first()
if not user:
return {"code": 222, "msg": f"用户名或密码不正确"}
else:
if not user.is_active:
return {"code": 333, "msg": f"{user.username} not active"}
else:
# 验证密码
if user.verify_password(args.get('password')):
access_token = create_access_token(identity=user.username)
print(f"access_token:{access_token}")
return jsonify({
"code": "0",
"message": "success",
"data": {
"access_token": access_token,
"userid": user.id
}
})
else:
return {"code": 222, "msg": f"用户名或密码不正确"}