#!/usr/bin/env python

"""

ptpython: Interactive Python shell.

Usage:

ptpython [ --vi ] [ --history=<filename> ] [ --no-colors ]

[ --autocompletion=<type> ] [ --always-multiline ]

[ --interactive=<filename> ] [--] [ <file> <arg>... ]

ptpython -h | --help

Options:

--vi : Use Vi keybindings instead of Emacs bindings.

--history=<filename> : Path to history file.

--interactive=<filename> : Start interactive shell after executing this file.

Other environment variables:

PYTHONSTARTUP: file executed on interactive startup (no default)

"""

from __future__ import absolute_import, unicode_literals

import docopt

import os

import six

import sys

from ptpython.repl import embed

def run():

a = docopt.docopt(__doc__)

vi_mode = bool(a['--vi'])

no_colors = bool(a['--no-colors'])

# Create globals/locals dict.

globals_, locals_ = {}, {}

# Log history

if a['--history']:

history_filename = os.path.expanduser(a['--history'])

else:

history_filename = os.path.expanduser('~/.ptpython_history')

# Startup path

startup_paths = []

if 'PYTHONSTARTUP' in os.environ:

startup_paths.append(os.environ['PYTHONSTARTUP'])

# --interactive

if a['--interactive']:

startup_paths.append(a['--interactive'])

# Add the current directory to `sys.path`.

sys.path.append('.')

# When a file has been given, run that, otherwise start the shell.

if a['<file>']:

sys.argv = [a['<file>']] + a['<arg>']

six.exec_(compile(open(a['<file>'], "rb").read(), a['<file>'], 'exec'))

else:

# Run interactive shell.

embed(globals_, locals_, vi_mode=vi_mode, history_filename=history_filename,

no_colors=no_colors, startup_paths=startup_paths)

if __name__ == '__main__':

run()