app.py

from onegov.agency.api import AgencyApiEndpoint
from onegov.agency.api import MembershipApiEndpoint
from onegov.agency.api import PersonApiEndpoint
from onegov.agency.custom import get_global_tools
from onegov.agency.custom import get_top_navigation
from onegov.agency.forms import UserGroupForm
from onegov.agency.initial_content import create_new_organisation
from onegov.agency.pdf import AgencyPdfAr, AgencyPdfBs
from onegov.agency.pdf import AgencyPdfDefault
from onegov.agency.pdf import AgencyPdfZg
from onegov.agency.request import AgencyRequest
from onegov.agency.theme import AgencyTheme
from onegov.api import ApiApp
from onegov.core import utils
from onegov.org import OrgApp
from onegov.org.app import get_editor_asset as editor_assets
from onegov.org.app import get_i18n_localedirs as get_org_i18n_localedirs
from onegov.org.app import get_redactor_asset as redactor_assets


from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from collections.abc import Callable
    from webob import Response


class AgencyApp(OrgApp, ApiApp):

    request_class = AgencyRequest

    @property
    def root_pdf_exists(self):
        return self.filestorage.exists('root.pdf')

    @property
    def people_xlsx_exists(self):
        return self.filestorage.exists('people.xlsx')

    @property
    def root_pdf_modified(self):
        if self.root_pdf_exists:
            return self.filestorage.getdetails('root.pdf').modified

    @property
    def people_xlsx_modified(self):
        if self.people_xlsx:
            return self.filestorage.getdetails('people.xlsx').modified

    @property
    def root_pdf(self):
        result = None
        if self.filestorage.exists('root.pdf'):
            with self.filestorage.open('root.pdf', 'rb') as file:
                result = file.read()
        return result

    @root_pdf.setter
    def root_pdf(self, value):
        with self.filestorage.open('root.pdf', 'wb') as file:
            file.write(value.read())

    @property
    def people_xlsx(self):
        result = None
        if self.filestorage.exists('people.xlsx'):
            with self.filestorage.open('people.xlsx', 'rb') as file:
                result = file.read()
        return result

    @people_xlsx.setter
    def people_xlsx(self, value):
        with self.filestorage.open('people.xlsx', 'wb') as file:
            file.write(value.read())

    @property
    def pdf_class(self):
        pdf_layout = self.org.meta.get('pdf_layout')
        if pdf_layout == 'ar':
            return AgencyPdfAr
        if pdf_layout == 'zg':
            return AgencyPdfZg
        if pdf_layout == 'bs':
            return AgencyPdfBs
        return AgencyPdfDefault

    @property
    def enable_yubikey(self):
        return self.org.meta.get('enable_yubikey', self._enable_yubikey)

    @enable_yubikey.setter
    def enable_yubikey(self, value):
        self._enable_yubikey = value


@AgencyApp.setting(section='org', name='create_new_organisation')
def get_create_new_organisation_factory():
    return create_new_organisation


@AgencyApp.template_directory()
def get_template_directory():
    return 'templates'


@AgencyApp.template_variables()
def get_template_variables(request):
    return {
        'global_tools': tuple(get_global_tools(request)),
        'top_navigation': tuple(get_top_navigation(request)),
    }


@AgencyApp.setting(section='core', name='theme')
def get_theme():
    return AgencyTheme()


@AgencyApp.setting(section='org', name='usergroup_form_class')
def get_usergroup_form_class():
    return UserGroupForm


@AgencyApp.setting(section='i18n', name='localedirs')
def get_i18n_localedirs():
    mine = utils.module_path('onegov.agency', 'locale')
    return [mine] + get_org_i18n_localedirs()


@AgencyApp.setting(section='org', name='ticket_manager_roles')
def get_ticket_manager_roles():
    return ('admin', 'editor', 'member')


@AgencyApp.setting(section='org', name='disabled_extensions')
def get_disabled_extensions():
    return ('PersonLinkExtension', )


@AgencyApp.webasset_output()
def get_webasset_output():
    return 'assets/bundles'


@AgencyApp.webasset_path()
def get_js_path():
    return 'assets/js'


@AgencyApp.webasset('people-select')
def get_people_select_asset():
    yield 'people-select.js'


@AgencyApp.webasset('sortable-multi-checkbox')
def get_sortable_multi_checkbox_asset():
    yield 'jquery.js'
    yield 'sortable.js'
    yield 'sortable-multi-checkbox.js'


@AgencyApp.webasset('redactor', filters={'js': None})
def get_redactor_asserts():
    yield from redactor_assets()


@AgencyApp.webasset('editor')
def get_editor_assets():
    yield from editor_assets()


@AgencyApp.setting(section='api', name='endpoints')
def get_api_endpoints():
    return [
        AgencyApiEndpoint,
        PersonApiEndpoint,
        MembershipApiEndpoint
    ]