todo.sr.ht/todosrht/decorators.py

16 lines
381 B
Python

from flask import redirect, request, abort
from flask_login import current_user
from functools import wraps
from todosrht.app import oauth_url
import urllib
def loginrequired(f):
@wraps(f)
def wrapper(*args, **kwargs):
if not current_user:
return redirect(oauth_url(request.url))
else:
return f(*args, **kwargs)
return wrapper