# -*- mode: python -*-

# Add auto-completion and a stored history file of commands to your Python

# interactive interpreter. Requires Python 2.0+, readline. Autocomplete is

# bound to the Esc key by default (you can change it - see readline docs).

#

# Store the file in ~/.pystartup, and set an environment variable to point

# to it:  "export PYTHONSTARTUP=~/.pystartup" in bash.

import os

import sys

import atexit

import readline

import rlcompleter

historyPath = os.path.expanduser("~/.pyhistory")

def save_history(historyPath=historyPath):

    import readline

    readline.write_history_file(historyPath)

if os.path.exists(historyPath):

    readline.read_history_file(historyPath)

term_with_colors = ['xterm', 'xterm-color', 'xterm-256color', 'linux', 'screen', 'screen-256color', 'screen-bce']

if os.environ.get('TERM') in term_with_colors:

    green='\001\033[32m\002'

    red='\001\033[31m\002'

    reset='\001\033[0m\002'

    sys.ps1 = red + '>>> ' + reset

    sys.ps2 = green + '... ' + reset

del term_with_colors

atexit.register(save_history)

del os, sys, atexit, readline, rlcompleter, save_history, historyPath