shell.py

#!/usr/bin/env python
# coding: utf-8

import os
import sys
import atexit
import IPython

from citadel.app import create_app


def hook_readline_hist():
    try:
        # Try to set up command history completion/saving/reloading
        import readline
    except ImportError:
        return

    # The place to store your command history between sessions
    histfile = os.environ["HOME"] + "/.citadel_history"
    readline.parse_and_bind('tab: complete')
    try:
        readline.read_history_file(histfile)
    except IOError:
        pass  # It doesn't exist yet.

    def save_hist():
        try:
            readline.write_history_file(histfile)
        except:
            print('Unable to save Python command history')
    atexit.register(save_hist)


def get_banner():
    return 'In Citadel shell now\n'


def pre_imports():
    from citadel.models import User, App, Release, AppUserRelation, Container, ELBInstance, OPLog
    from citadel.tasks import remove_container
    from citadel.ext import db, rds, get_etcd
    from citadel.rpc.client import get_core
    return locals()


def ipython_shell(user_ns):
    if getattr(IPython, 'version_info', None) and IPython.version_info[0] >= 1:
        from IPython.terminal.ipapp import TerminalIPythonApp
        from IPython.terminal.interactiveshell import TerminalInteractiveShell
    else:
        from IPython.frontend.terminal.ipapp import TerminalIPythonApp
        from IPython.frontend.terminal.interactiveshell import TerminalInteractiveShell

    class ShireIPythonApp(TerminalIPythonApp):
        def init_shell(self):
            self.shell = TerminalInteractiveShell.instance(
                config=self.config,
                display_banner=False,
                profile_dir=self.profile_dir,
                ipython_dir=self.ipython_dir,
                banner1=get_banner(),
                banner2=''
            )
            self.shell.configurables.append(self)

    app = ShireIPythonApp.instance()
    app.initialize()
    app.shell.user_ns.update(user_ns)

    citadel_app = create_app()
    with citadel_app.app_context():
        sys.exit(app.start())


def main():
    hook_readline_hist()
    ipython_shell(pre_imports())


if __name__ == '__main__':
    main()