Source code for invenio_iiif.ext

# -*- coding: utf-8 -*-
#
# This file is part of Invenio.
# Copyright (C) 2018 CERN.
#
# Invenio is free software; you can redistribute it and/or modify it
# under the terms of the MIT License; see LICENSE file for more details.

"""IIIF API for Invenio."""

from __future__ import absolute_import, print_function

from flask_iiif import IIIF
from flask_restful import Api

from . import config
from .handlers import image_opener, protect_api


[docs]class InvenioIIIF(object): """Invenio-IIIF extension.""" def __init__(self, app=None): """Extension initialization.""" if app: self.init_app(app)
[docs] def init_app(self, app): """Flask application initialization.""" self.init_config(app) self.iiif_ext = IIIF(app=app) self.iiif_ext.api_decorator_handler(protect_api) self.iiif_ext.uuid_to_image_opener_handler(image_opener) app.extensions['invenio-iiif'] = self
[docs] def init_config(self, app): """Initialize configuration.""" for k in dir(config): if k.startswith('IIIF_'): app.config.setdefault(k, getattr(config, k))
[docs]class InvenioIIIFAPI(InvenioIIIF): """Invenio-IIIF extension."""
[docs] def init_app(self, app): """Flask application initialization.""" super(InvenioIIIFAPI, self).init_app(app) api = Api(app=app) self.iiif_ext.init_restful(api, prefix=app.config['IIIF_API_PREFIX'])