Source code for pacifica.policy.search_render

#!/usr/bin/python
# -*- coding: utf-8 -*-
"""This is the render object for the search interface."""
import importlib
from six import text_type
from .config import get_config

ELASTIC_INDEX = get_config().get('elasticsearch', 'index')


[docs]class SearchRender(object): """Search render class to contain methods."""
[docs] @staticmethod def get_render_class(obj_cls): """Get the render class dynamically.""" obj_mod = importlib.import_module('.search.{}'.format(obj_cls), 'pacifica.policy') parts = [obj_cls_part.capitalize() for obj_cls_part in obj_cls.split('_')] return getattr(obj_mod, '{}Render'.format(''.join(parts)))
[docs] @classmethod def generate(cls, obj_cls, objs): """generate the institution object.""" render_cls = cls.get_render_class(obj_cls) for obj in objs: yield { '_op_type': 'update', '_index': ELASTIC_INDEX, '_type': 'doc', '_id': render_cls.obj_id(**obj), 'doc': render_cls.render(obj, True, obj_cls != 'transactions'), 'doc_as_upsert': True } if obj_cls == 'projects': st_render_cls = cls.get_render_class('science_themes') yield { '_op_type': 'update', '_index': ELASTIC_INDEX, '_type': 'doc', '_id': text_type('science_themes_{}').format(obj['science_theme']), 'doc': st_render_cls.render(obj, True, True), 'doc_as_upsert': True }