realms-wiki/realms/modules/search/commands.py

import click
from flask import current_app
from realms import search, cli_group
from realms.modules.wiki.models import Wiki


@cli_group(short_help="Search Module")
def cli():
    pass


@cli.command()
def rebuild_index():
    """ Rebuild search index
    """
    if current_app.config.get('SEARCH_TYPE') == 'simple':
        click.echo("Search type is simple, try using elasticsearch.")
        return

    # Wiki
    search.delete_index('wiki')
    wiki = Wiki(current_app.config['WIKI_PATH'])
    for entry in wiki.get_index():
        page = wiki.get_page(entry['name'])
        if not page:
            # Some non-markdown files may have issues
            continue
        # TODO add email?
        body = dict(name=page.name,
                    content=page.data,
                    message=page.info['message'],
                    username=page.info['author'],
                    updated_on=entry['mtime'],
                    created_on=entry['ctime'])
        search.index_wiki(name, body)