17 lines
518 B
Python
17 lines
518 B
Python
from functools import wraps
|
|
from flask import request
|
|
from flask_jwt_extended import get_jwt_identity
|
|
from werkzeug.exceptions import Unauthorized
|
|
|
|
def logged_user_id(f):
|
|
@wraps(f)
|
|
def decorated_function(*args, **kwargs):
|
|
logged_user_id = get_jwt_identity()
|
|
if not logged_user_id:
|
|
raise Unauthorized('Token inválido ou expirado.')
|
|
|
|
# Passe o logged_user_id como argumento para a função
|
|
return f(logged_user_id, *args, **kwargs)
|
|
|
|
return decorated_function
|