Fix python indentation

This commit is contained in:
Jackson Harper
2024-07-05 18:11:38 +08:00
parent eee38fcbb9
commit ff80c32406

View File

@ -4,7 +4,6 @@ from flask import request, jsonify
from functools import wraps from functools import wraps
from datetime import datetime, timedelta from datetime import datetime, timedelta
SECRET_KEY = os.getenv('JWT_SECRET') SECRET_KEY = os.getenv('JWT_SECRET')
ADMIN_SECRET_KEY = os.getenv('JWT_ADMIN_SECRET_KEY') ADMIN_SECRET_KEY = os.getenv('JWT_ADMIN_SECRET_KEY')
@ -20,39 +19,38 @@ def generate_admin_token():
def user_token_required(f): def user_token_required(f):
@wraps(f) @wraps(f)
def decorated(*args, **kwargs): def decorated(*args, **kwargs):
token = None token = None
if 'Authorization' in request.headers: if 'Authorization' in request.headers:
print("request.headers['Authorization'].split(" ")[1]", request.headers['Authorization'].split(" ")[1]) token = request.headers['Authorization'].split(" ")[1]
token = request.headers['Authorization'].split(" ")[1] if not token:
if not token: return jsonify({'message': 'Token is missing!'}), 401
return jsonify({'message': 'Token is missing!'}), 401 try:
try: data = jwt.decode(token, SECRET_KEY, algorithms=["HS256"])
data = jwt.decode(token, SECRET_KEY, algorithms=["HS256"]) request.user_id = data['uid']
request.user_id = data['uid'] except jwt.ExpiredSignatureError:
except jwt.ExpiredSignatureError: return jsonify({'message': 'Token has expired!'}), 401
return jsonify({'message': 'Token has expired!'}), 401 except jwt.InvalidTokenError:
except jwt.InvalidTokenError: return jsonify({'message': 'Token is invalid!'}), 401
return jsonify({'message': 'Token is invalid!'}), 401 return f(*args, **kwargs)
return f(*args, **kwargs) return decorated
return decorated
def admin_token_required(f): def admin_token_required(f):
@wraps(f) @wraps(f)
def decorated(*args, **kwargs): def decorated(*args, **kwargs):
token = None token = None
if 'Authorization' in request.headers: if 'Authorization' in request.headers:
token = request.headers['Authorization'].split(" ")[1] token = request.headers['Authorization'].split(" ")[1]
if not token: if not token:
return jsonify({'message': 'Token is missing!'}), 401 return jsonify({'message': 'Token is missing!'}), 401
try: try:
data = jwt.decode(token, ADMIN_SECRET_KEY, algorithms=["HS256"]) data = jwt.decode(token, ADMIN_SECRET_KEY, algorithms=["HS256"])
if data['role'] != 'admin': if data['role'] != 'admin':
return jsonify({'message': 'Admin token required!'}), 403 return jsonify({'message': 'Admin token required!'}), 403
except jwt.ExpiredSignatureError: except jwt.ExpiredSignatureError:
return jsonify({'message': 'Token has expired!'}), 401 return jsonify({'message': 'Token has expired!'}), 401
except jwt.InvalidTokenError: except jwt.InvalidTokenError:
return jsonify({'message': 'Token is invalid!'}), 401 return jsonify({'message': 'Token is invalid!'}), 401
return f(*args, **kwargs) return f(*args, **kwargs)
return decorated return decorated