49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from flask import Blueprint, make_response, redirect, request
|
|
from flask_jwt_extended import (
|
|
create_access_token,
|
|
create_refresh_token,
|
|
jwt_required,
|
|
set_access_cookies,
|
|
set_refresh_cookies,
|
|
unset_access_cookies,
|
|
unset_refresh_cookies,
|
|
)
|
|
|
|
from pear_admin.extensions import db
|
|
from pear_admin.orms import UserORM
|
|
|
|
passport_api = Blueprint("passport", __name__)
|
|
|
|
|
|
@passport_api.post("/login")
|
|
def login_in():
|
|
data = request.get_json()
|
|
|
|
user: UserORM = db.session.execute(
|
|
db.select(UserORM).where(UserORM.username == data["username"])
|
|
).scalar()
|
|
|
|
if not user:
|
|
return {"message": "用户不存在", "code": -1}, 401
|
|
if user.check_password(data["password"]):
|
|
return {"message": "用户密码错误", "code": -1}, 401
|
|
|
|
access_token = create_access_token(user)
|
|
refresh_token = create_refresh_token(user)
|
|
|
|
response = make_response({"code": 0, "message": "登录成功"})
|
|
|
|
set_access_cookies(response, access_token)
|
|
set_refresh_cookies(response, refresh_token)
|
|
|
|
return response
|
|
|
|
|
|
@passport_api.route("/logout", methods=["GET", "POST"])
|
|
@jwt_required()
|
|
def logout():
|
|
response = make_response(redirect("/login"))
|
|
unset_access_cookies(response)
|
|
unset_refresh_cookies(response)
|
|
return response
|