Fix python indentation
This commit is contained in:
@ -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
|
||||||
Reference in New Issue
Block a user