Source code for invenio_indexer.ext

# -*- coding: utf-8 -*-
# This file is part of Invenio.
# Copyright (C) 2016-2018 CERN.
# Copyright (C) 2016 TIND.
# 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.

"""Flask exension for Invenio-Indexer."""

from __future__ import absolute_import, print_function

import six
from flask import current_app
from werkzeug.utils import cached_property, import_string

from . import config
from .cli import run  # noqa
from .signals import before_record_index

[docs]class InvenioIndexer(object): """Invenio-Indexer extension.""" def __init__(self, app=None): """Extension initialization. :param app: The Flask application. (Default: ``None``) """ if app: self.init_app(app)
[docs] def init_app(self, app): """Flask application initialization. :param app: The Flask application. """ self.init_config(app) app.extensions['invenio-indexer'] = self hooks = app.config.get('INDEXER_BEFORE_INDEX_HOOKS', []) for hook in hooks: if isinstance(hook, six.string_types): hook = import_string(hook) before_record_index.connect_via(app)(hook)
[docs] def init_config(self, app): """Initialize configuration. :param app: The Flask application. """ for k in dir(config): if k.startswith('INDEXER_'): app.config.setdefault(k, getattr(config, k))
[docs] def record_to_index(self): """Import the configurable 'record_to_index' function.""" return import_string(current_app.config.get('INDEXER_RECORD_TO_INDEX'))