pcli.py

#!/usr/bin/env python

from __future__ import unicode_literals
from __future__ import print_function

from os.path import expanduser

from docopt import docopt
from prompt_toolkit import HTML
from prompt_toolkit import prompt
from prompt_toolkit import PromptSession
from prompt_toolkit import print_formatted_text as printc

from engine import Engine
from interpreter import PythonInterpreter
from interactive import user_functions

from prompt_toolkit.lexers import PygmentsLexer
from prompt_toolkit.styles import Style
from pygments.lexers.python import PythonLexer
from prompt_toolkit.history import FileHistory
from prompt_toolkit.auto_suggest import AutoSuggestFromHistory


banner = """
POETS Client (pcli) v0.1
Type "help" for a list of available functions.
"""

usage="""POETS Client (pcli) v0.1

Usage:
  pcli.py [options]

Options:
  -n --nocolor   Disable terminal colors.
  -q --quiet     Suppress printing banner.

"""

styles = {
    "prompt": "ansiblue bold",
    "error": "gray"
}

ps1 = '<prompt>pcli> </prompt>'

syntactic_sugar = {
    "pp": "pp(_)",
    "ps": "ps()",
    "top": "top()",
    "help": "help()",
    "flush": "flush()",
    "engines": "engines()",
    "time on": "_timer = True",
    "time off": "_timer = False",
}


def create_prompt(history_file, nocolor=False):
    """Create a function that prompts the user for a command."""
    style = Style.from_dict({} if nocolor else styles)
    history = FileHistory(history_file)
    session = PromptSession(style=style, history=history)
    ps1_html = HTML(ps1)
    auto_suggest = AutoSuggestFromHistory()
    lexer = None if nocolor else PygmentsLexer(PythonLexer)
    def prompt():
        return session.prompt(ps1_html, auto_suggest=auto_suggest, lexer=lexer)
    return prompt


def main():

    args = docopt(usage, version="pcli v0.1")

    context_file = expanduser("~/.pcli_context")
    history_file = expanduser("~/.pcli_history")

    prompt = create_prompt(history_file, args["--nocolor"])
    interpreter = PythonInterpreter(user_functions, context_file)

    if not args["--quiet"]:
        printc(banner.strip())

    while True:
        try:
            command = prompt()
        except KeyboardInterrupt:
            continue
        except EOFError:
            break
        command = syntactic_sugar.get(command, command)
        result = interpreter.eval(command)
        print(result, end="")


if __name__ == '__main__':
    main()
else:
    # Import user_functions into global scope. This is just to make the
    # various functions available from the pcli shell available for importing
    # from this module directory.
    pcli_locals = locals()
    for func in user_functions:
        pcli_locals[func.func_name] = func