from functools import wraps
from flask import session, redirect, url_for, request


def current_user():
    return session.get('user')


def login_required(f):
    @wraps(f)
    def decorated(*args, **kwargs):
        if not current_user():
            session['next_url'] = request.url
            return redirect(url_for('auth.login'))
        return f(*args, **kwargs)
    return decorated