Template for creating awesome REPLs with prompt-toolkit

litecli.py

from prompt_toolkit import prompt
from prompt_toolkit.history import FileHistory
from prompt_toolkit.contrib.completers import WordCompleter
from pygments.lexers.sql import SqlLexer
import sqlite3

history = FileHistory('history.txt')
completer = WordCompleter(['select', 'insert', 'delete', 'drop', 'from'])
connection = sqlite3.connect(':memory:')

while 1:
    try:
        inp = prompt('prompt> ', history=history, lexer=SqlLexer, completer=completer)
        with connection:
            print (connection.execute(inp).fetchall())
    except EOFError:
        print ('Good bye!')
        exit(0)
    except Exception as e:
        print (str(e))