Skip to content

Commit 8d0fe70

Browse files
committed
Update flask_jwt_extended decorator
1 parent 863d50b commit 8d0fe70

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

emmaa_service/api.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from flask_restx import Api, Resource, fields, inputs, abort as restx_abort
1414
from flask_cors import CORS
1515

16-
from flask_jwt_extended import jwt_optional
16+
from flask_jwt_extended import jwt_required
1717
from urllib import parse
1818
from collections import defaultdict, Counter
1919
from copy import deepcopy
@@ -1146,14 +1146,14 @@ def session_expiration_check():
11461146

11471147

11481148
@app.route('/health')
1149-
@jwt_optional
1149+
@jwt_required(optional=True)
11501150
def health():
11511151
return jsonify({'status': 'pass'})
11521152

11531153

11541154
@app.route('/')
11551155
@app.route('/home')
1156-
@jwt_optional
1156+
@jwt_required(optional=True)
11571157
def get_home():
11581158
loaded = request.args.get('loaded')
11591159
loaded = (loaded == 'true')
@@ -1169,7 +1169,7 @@ def get_home():
11691169

11701170

11711171
@app.route('/dashboard/<model>')
1172-
@jwt_optional
1172+
@jwt_required(optional=True)
11731173
def get_model_dashboard(model):
11741174
"""Render model dashboard page."""
11751175
loaded = request.args.get('loaded')
@@ -1691,7 +1691,7 @@ def get_model_tests_page(model):
16911691

16921692

16931693
@app.route('/query')
1694-
@jwt_optional
1694+
@jwt_required(optional=True)
16951695
def get_query_page():
16961696
"""Render queries page."""
16971697
loaded = request.args.get('loaded')
@@ -1963,7 +1963,7 @@ def get_all_statements_page(model):
19631963

19641964

19651965
@app.route('/demos')
1966-
@jwt_optional
1966+
@jwt_required(optional=True)
19671967
def get_demos_page():
19681968
user, roles = resolve_auth(dict(request.args))
19691969
user_email = user.email if user else ""
@@ -1972,7 +1972,7 @@ def get_demos_page():
19721972

19731973

19741974
@app.route('/chat')
1975-
@jwt_optional
1975+
@jwt_required(optional=True)
19761976
def chat_with_the_model():
19771977
model = request.args.get('model', '')
19781978
user, roles = resolve_auth(dict(request.args))
@@ -2057,7 +2057,7 @@ def get_query_tests_page(model):
20572057

20582058

20592059
@app.route('/query/submit', methods=['POST'])
2060-
@jwt_optional
2060+
@jwt_required(optional=True)
20612061
def process_query():
20622062
"""Get result for a query."""
20632063
# Print inputs.
@@ -2241,7 +2241,7 @@ def email_unsubscribe_post():
22412241

22422242

22432243
@app.route('/subscribe/<model>', methods=['POST'])
2244-
@jwt_optional
2244+
@jwt_required(optional=True)
22452245
def model_subscription(model):
22462246
user, roles = resolve_auth(dict(request.args))
22472247
if not roles and not user:
@@ -2323,7 +2323,7 @@ def get_statement_by_paper(model, paper_id, paper_id_type, date, hash_val):
23232323

23242324

23252325
@app.route('/curation/submit/<hash_val>', methods=['POST'])
2326-
@jwt_optional
2326+
@jwt_required(optional=True)
23272327
def submit_curation_endpoint(hash_val, **kwargs):
23282328
user, roles = resolve_auth(dict(request.args))
23292329
if not roles and not user:

0 commit comments

Comments
 (0)