diff --git a/server.py b/server.py index 5966b61..82c786b 100644 --- a/server.py +++ b/server.py @@ -89,12 +89,15 @@ elif AUTH_TYPE == "basic": basic_auth = BasicAuth(app) def check_auth_type(f): - LOG.error("check_oidc(arg): "+str(f)) - if AUTH_TYPE == "oidc": - flask_routes = ['%s' % rule for rule in app.url_map.iter_rules()] - for route in flask_routes: - LOG.error("Applying OIDC Require_Login to route: "+str(f)) - oidc.require_login(app.route(f)) + @wraps(f) + def decorated_function(*args, **kwargs): + if AUTH_TYPE == "oidc": + flask_routes = ['%s' % rule for rule in app.url_map.iter_rules()] + for route in flask_routes: + LOG.error("Applying OIDC Require_Login to route: "+str(f)) + oidc.require_login(f) + return f(*args, **kwargs) + return decorated_function ######################################################################################## # / pages - User-facing pages