flask-galatea/helpers.py

24 lines
804 B
Python

#This file is part galatea blueprint for Flask.
#The COPYRIGHT file at the top level of this repository contains
#the full copyright notices and license terms.
from flask import redirect, url_for, session, request
from functools import wraps
def secure(function):
@wraps(function)
def decorated_function(*args, **kwargs):
if not request.is_secure:
return redirect(request.url.replace('http://', 'https://'))
else:
return function(*args, **kwargs)
return decorated_function
def login_required(f):
@wraps(f)
def decorated_function(*args, **kwargs):
logged = session.get('logged_in', None)
if not logged:
return redirect(url_for('.login', lang='es'))
return f(*args, **kwargs)
return decorated_function