mirror of
https://github.com/Cian-H/invenio-theme-iform.git
synced 2026-08-14 23:53:22 +01:00
130 lines
4.2 KiB
Python
130 lines
4.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2020-2024 Graz University of Technology.
|
|
#
|
|
# invenio-theme-iform is free software; you can redistribute it and/or
|
|
# modify it under the terms of the MIT License; see LICENSE file for more
|
|
# details.
|
|
|
|
"""invenio module for I-Form theme."""
|
|
|
|
from flask import g, has_request_context, request
|
|
from flask_login import login_required
|
|
|
|
from . import config
|
|
from .views import index, locked, require_iform_authenticated
|
|
|
|
|
|
class InvenioThemeIform(object):
|
|
"""invenio-theme-iform extension."""
|
|
|
|
def __init__(self, app=None):
|
|
"""Extension initialization."""
|
|
if app:
|
|
self.init_app(app)
|
|
|
|
def init_app(self, app):
|
|
"""Flask application initialization."""
|
|
# add index route rule
|
|
# https://flask.palletsprojects.com/en/1.1.x/api/#flask.Flask.add_url_rule
|
|
app.add_url_rule("/", "index", index)
|
|
self.init_config(app)
|
|
|
|
app.register_error_handler(423, locked)
|
|
|
|
@app.before_request
|
|
def block_registration_if_disabled():
|
|
if request.endpoint == "security.register":
|
|
from flask import current_app, render_template
|
|
from .administration.views import get_site_settings
|
|
settings = get_site_settings()
|
|
if not settings.get("registration_enabled", True):
|
|
admin_email = (
|
|
current_app.config.get("APP_RDM_ADMIN_EMAIL_RECIPIENT")
|
|
or current_app.config.get("SUPPORT_EMAIL")
|
|
or current_app.config.get("THEME_SUPPORT_EMAIL")
|
|
or current_app.config.get("SECURITY_EMAIL_SENDER")
|
|
or "info@repo.i-form.ie"
|
|
)
|
|
return (
|
|
render_template(
|
|
"invenio_theme_iform/accounts/registration_disabled.html",
|
|
admin_email=admin_email,
|
|
),
|
|
403,
|
|
)
|
|
|
|
app.extensions["invenio-theme-iform"] = self
|
|
|
|
@app.context_processor
|
|
def inject_site_settings():
|
|
from .administration.views import get_site_settings
|
|
settings = get_site_settings()
|
|
return dict(registration_enabled=settings.get("registration_enabled", True))
|
|
|
|
def init_config(self, app):
|
|
"""Initialize configuration."""
|
|
for k in dir(config):
|
|
if k.startswith("INVENIO_THEME_IFORM_") or k.startswith("THEME_IFORM_"):
|
|
app.config.setdefault(k, getattr(config, k))
|
|
|
|
|
|
def finalize_app(app):
|
|
"""Finalize app."""
|
|
modify_user_dashboard(app)
|
|
guard_view_functions(app)
|
|
|
|
|
|
def create_quick_upload_submenu(app):
|
|
"""Create a submenu for the quick-upload menu."""
|
|
root_menu = app.extensions["menu"].root_node
|
|
|
|
quick_upload_submenu = root_menu.submenu("quick_upload")
|
|
quick_upload_submenu._text = "Quick upload"
|
|
quick_upload_submenu._order = 1
|
|
quick_upload_submenu.register("records.upload", "Upload Research Data")
|
|
|
|
|
|
def modify_user_dashboard(app):
|
|
"""Modify user dashboard."""
|
|
root_menu = app.extensions["menu"].root_node
|
|
|
|
user_dashboard_menu = root_menu.submenu("dashboard")
|
|
user_dashboard_menu.submenu("overview").register(
|
|
"invenio_theme_iform.overview",
|
|
text="Overview",
|
|
order=0,
|
|
)
|
|
|
|
root_menu.submenu("actions.deposit").register(
|
|
"invenio_theme_iform.overview",
|
|
"My dashboard",
|
|
order=1,
|
|
)
|
|
|
|
|
|
def guard_view_functions(app):
|
|
"""Guard view-functions against unauthenticated access."""
|
|
endpoints_to_guard = [
|
|
"invenio_app_rdm_users.communities",
|
|
"invenio_app_rdm_users.requests",
|
|
"invenio_app_rdm_users.uploads",
|
|
]
|
|
|
|
for endpoint in endpoints_to_guard:
|
|
view_func = app.view_functions.get(endpoint)
|
|
if not view_func:
|
|
continue
|
|
|
|
# decorate view-func
|
|
# same as if view-func were defined with:
|
|
# @login_required
|
|
# @require_iform_authenticated_user
|
|
view_func = login_required(
|
|
require_iform_authenticated(
|
|
view_func,
|
|
),
|
|
)
|
|
|
|
app.view_functions[endpoint] = view_func
|