69ffcee98f
This commit undertakes a major architectural refactoring to improve project organization, scalability, and maintainability
35 lines
No EOL
1.2 KiB
Python
35 lines
No EOL
1.2 KiB
Python
from functools import wraps
|
|
from flask import request, jsonify, current_app
|
|
import jwt
|
|
from datetime import datetime, timedelta, timezone
|
|
|
|
def _getconfig():
|
|
jwt_key = current_app.config['JWT_KEY']
|
|
jwt_expire = current_app.config['JWT_EXPIRE'] # In days
|
|
return jwt_key, jwt_expire
|
|
|
|
def jwt_required(f):
|
|
@wraps(f)
|
|
def decorated_function(*args, **kwargs):
|
|
jwt_key, _ = _getconfig()
|
|
token = request.cookies.get("token")
|
|
if not token:
|
|
return jsonify({"error": "Authentication token missing"}), 400
|
|
try:
|
|
payload = jwt.decode(token, jwt_key, algorithms=["HS256"])
|
|
request.bid = payload["bid"]
|
|
except jwt.ExpiredSignatureError:
|
|
return jsonify({"error": "Token expired"}), 401
|
|
except jwt.InvalidTokenError:
|
|
return jsonify({"error": "Invalid token"}), 401
|
|
return f(*args, **kwargs)
|
|
return decorated_function
|
|
|
|
def token_gen(bid):
|
|
jwt_key, jwt_expire = _getconfig()
|
|
exptime = datetime.now(timezone.utc) + timedelta(days=jwt_expire)
|
|
token = jwt.encode(
|
|
{"bid": bid, "exp": exptime},
|
|
jwt_key,
|
|
algorithm="HS256")
|
|
return token |