.konchrc

# -*- coding: utf-8 -*-
# vi: set ft=python :
import os
import django

from django.conf import settings
from django.utils.termcolors import colorize
from django.apps import apps
import konch
import elasticsearch_dsl
from elasticsearch_dsl import connections, Search

from elasticsearch_metrics.registry import registry
from elasticsearch_metrics.management.color import color_style


# Available options:
#   'context', 'banner', 'shell', 'prompt', 'output',
#   'context_format', 'ipy_extensions', 'ipy_autoreload',
#   'ipy_colors', 'ipy_highlighting_style'
context = {
    "dsl": elasticsearch_dsl,
    "Search": Search,
    "registry": registry,
    "apps": apps,
    "settings": settings,
    "style": color_style(),
}
konch.config(
    {
        "banner": colorize(
            "Welcome to the django-elasticsearch-metrics shell. Happy hacking!",
            opts=("bold",),
        ),
        "ipy_colors": "linux",
    }
)


def setup():
    # Set up django and add the default elasticsearch-py client and Metric to the context
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "tests.settings")
    django.setup()
    from elasticsearch_metrics.metrics import Metric

    context["client"] = connections.get_connection()
    context["Metric"] = Metric
    konch.config({"context": context})


if os.path.exists(".konchrc.local"):
    konch.use_file(".konchrc.local", trust=True)