Skip to content

Commit df25a77

Browse files
committed
fix: 为获取refresh token接口添加自定义异常,修复前端循环请求接口的bug
1 parent 26c630e commit df25a77

1 file changed

Lines changed: 11 additions & 4 deletions

File tree

app/api/cms/user.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,11 @@
77
from operator import and_
88

99
from flask import jsonify
10-
from flask_jwt_extended import create_access_token, jwt_refresh_token_required, get_jwt_identity, get_current_user, \
11-
create_refresh_token
10+
from flask_jwt_extended import create_access_token, get_jwt_identity, get_current_user, \
11+
create_refresh_token, verify_jwt_refresh_token_in_request
1212
from lin.core import manager, route_meta, Log
1313
from lin.db import db
14-
from lin.exception import NotFound, Success, Failed, RepeatException, ParameterException
14+
from lin.exception import NotFound, Success, Failed, RepeatException, ParameterException, RefreshException
1515
from lin.jwt import login_required, admin_required, get_tokens
1616
from lin.log import Logger
1717
from lin.redprint import Redprint
@@ -100,16 +100,23 @@ def get_information():
100100

101101
@user_api.route('/refresh', methods=['GET'])
102102
@route_meta(auth='刷新令牌', module='用户', mount=False)
103-
@jwt_refresh_token_required
104103
def refresh():
104+
105+
try:
106+
verify_jwt_refresh_token_in_request()
107+
except Exception:
108+
return RefreshException()
109+
105110
identity = get_jwt_identity()
111+
106112
if identity:
107113
access_token = create_access_token(identity=identity)
108114
refresh_token = create_refresh_token(identity=identity)
109115
return jsonify({
110116
'access_token': access_token,
111117
'refresh_token': refresh_token
112118
})
119+
113120
return NotFound(msg='refresh_token未被识别')
114121

115122

0 commit comments

Comments
 (0)