Commit bee39955 authored by Ahmad Anvari's avatar Ahmad Anvari

Add auth middleware

parent 6bb0e5e3
from functools import wraps from functools import wraps
from flask import g, request, redirect, url_for from flask import g, request
from utils.jwt import decode_jwt
from db.user import User
from flask import jsonify
def login_required(f): def login_required(f):
@wraps(f) @wraps(f)
def decorated_function(*args, **kwargs): def decorated_function(*args, **kwargs):
auth = request.headers.get('Authorization') auth = request.headers.get('Authorization')
print(auth) token = auth.split(' ')[1]
try:
payload = decode_jwt(token)
user = User.find_by_id(payload['user_id'])
if user is None:
return jsonify({"Error": "Not authorized"}), 401
g.user = user
except:
return jsonify({"Error": "Not authorized"}), 401
return f(*args, **kwargs) return f(*args, **kwargs)
return decorated_function return decorated_function
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment