X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=lib%2Fpython%2Finit.py;h=7377a26912ec83ab2d87abc833b36a211460f244;hb=f9b42a2b3f0fa5719d13bc54179bb9b5d8315c11;hp=8b86300a6f1b21bef95548bde8ba959fded62f43;hpb=f46bd4d41cc7f243bc8a321effee5200aa69e709;p=dotfiles.git diff --git a/lib/python/init.py b/lib/python/init.py index 8b86300..7377a26 100644 --- a/lib/python/init.py +++ b/lib/python/init.py @@ -4,52 +4,64 @@ # into your .profile (use whatever syntax and initialization file # is appropriate for your shell): # -# PYTHONSTARTUP=$HOME/init.py # or where you really put it +# PYTHONSTARTUP=$HOME/init.py # or where you really put it # export PYTHONSTARTUP + def init(): - import sys, os - import __builtin__ + try: + import __builtin__ as builtins + except ImportError: + import builtins + import os + import sys - pyreadlinew32_startup = os.path.join(sys.prefix, - 'lib', 'site-packages', 'pyreadline', 'configuration', 'startup.py') + # readline/pyreadline + + pyreadlinew32_startup = os.path.join( + sys.prefix, 'lib', 'site-packages', + 'pyreadline', 'configuration', 'startup.py') if os.path.exists(pyreadlinew32_startup): execfile(pyreadlinew32_startup) else: - # From Bruce Edge + # From Bruce Edge: + # https://mail.python.org/pipermail/python-list/2001-March/062888.html try: - import rlcompleter, readline - initfile = os.environ.get('INPUTRC') or os.path.expanduser('~/.inputrc') + import rlcompleter # noqa: need for completion + import readline + initfile = os.environ.get('INPUTRC') \ + or os.path.expanduser('~/.inputrc') readline.read_init_file(initfile) - histfile = os.path.expanduser('~/.python-history') + histfile = os.path.expanduser('~/.python_history') try: readline.read_history_file(histfile) except IOError: - pass # No such file + pass # No such file def savehist(): - histfilesize = os.environ.get('HISTFILESIZE') or \ - os.environ.get('HISTSIZE') - if histfilesize: + histsize = os.environ.get('HISTSIZE') + if histsize: try: - histfilesize = int(histfilesize) + histsize = int(histsize) except ValueError: pass else: - readline.set_history_length(histfilesize) + readline.set_history_length(histsize) readline.write_history_file(histfile) import atexit atexit.register(savehist) except (ImportError, AttributeError): - pass # no readline or atexit, or readline doesn't have - # {read,write}_history_file - ignore the error + # no readline or atexit, or readline doesn't have + # {read,write}_history_file - ignore the error + pass + # terminal term = os.environ.get('TERM', '') if 'linux' in term: @@ -57,65 +69,139 @@ def init(): else: background = os.environ.get('BACKGROUND', 'light').lower() - # From Randall Hopper + # From Randall Hopper: + # https://mail.python.org/pipermail/python-list/2001-March/112696.html - for _term in ['linux', 'term', 'rxvt', 'vt100', 'screen']: - if _term in term: - if background == 'dark': - ps1_color = '3' # yellow - stdout_color = '7' # bold white - else: - ps1_color = '4' # blue - stdout_color = '0' # bold black + for _term in ['linux', 'rxvt', 'screen', 'term', 'vt100']: + if _term not in term: + continue - sys.ps1 = '\001\033[3%sm\002>>>\001\033[0m\002 ' % ps1_color - sys.ps2 = '\001\033[1;32m\002...\001\033[0m\002 ' # bold green + if background == 'dark': + ps1_color = '3' # yellow + stdout_color = '7' # bold white + else: + ps1_color = '4' # blue + stdout_color = '0' # bold black + sys.ps1 = '\001\033[3%sm\002>>>\001\033[0m\002 ' % ps1_color + sys.ps2 = '\001\033[1;32m\002...\001\033[0m\002 ' # bold green - # From Denis Otkidach + # From Denis Otkidach - class ColoredFile: - def __init__(self, fp, begin, end='\033[0m'): # reset all attributes - self.__fp = fp - self.__begin = begin - self.__end = end + class ColoredFile: + def __init__(self, fp, begin, + end='\033[0m'): # reset all attributes + self.__fp = fp + self.__begin = begin + self.__end = end - def write(self, s): - self.__fp.write(self.__begin+s+self.__end) + def write(self, s): + self.__fp.write(self.__begin+s+self.__end) - def writelines(self, lines): - map(self.write, lines) + def writelines(self, lines): + map(self.write, lines) - def __getattr__(self, attr): - return getattr(self.__fp, attr) + def __getattr__(self, attr): + return getattr(self.__fp, attr) - sys.stdout = ColoredFile(sys.stdout, '\033[1;3%sm' % stdout_color) - sys.stderr = ColoredFile(sys.stderr, '\033[31m') # red + sys.stdout = ColoredFile(sys.stdout, '\033[1;3%sm' % stdout_color) + sys.stderr = ColoredFile(sys.stderr, '\033[31m') # red - break + break try: import locale except ImportError: - pass # locale was not compiled + pass # locale was not compiled else: try: locale.setlocale(locale.LC_ALL, '') + except (ImportError, locale.Error): + pass # no locale support or unsupported locale - from pprint import pprint + # set displayhook and excepthook - def displayhook(value): - if value is not None: - __builtin__._ = value - pprint(value) + from pprint import pprint + pager = os.environ.get("PAGER") or 'more' - sys.displayhook = displayhook + # if your pager is 'less', options '-F' and '-R' must be passed to it, + # and option '-X' is very much recommended + if pager == 'less': + less = os.environ.get("LESS") or '' + for opt in 'X', 'R', 'F': + if opt not in less: + less = opt + less + os.environ["LESS"] = less - except (ImportError, locale.Error): - pass # no locale support or unsupported locale + class BasePager: + def write(self, value): + self.stdin.write(value) + def pprint(self, value): + pprint(value, stream=self.stdin) - # From: Paul Magwene with a lot of my fixes + def close(self): + self.stdin.close() + + try: + from subprocess import Popen, PIPE + except ImportError: + class Pager(BasePager): + def __init__(self): + self.pipe = Popen(pager, shell=True, stdin=PIPE) + self.stdin = self.pipe.stdin + + def close(self): + BasePager.close(self) + self.pipe.wait() + else: + class Pager(BasePager): + def __init__(self): + self.stdin = os.popen(pager, 'w') + + def displayhook(value): + if value is not None: + builtins._ = value + pager = Pager() + pager.pprint(value) + pager.close() + + sys.displayhook = displayhook + + from traceback import format_exception + + def excepthook(etype, evalue, etraceback): + lines = format_exception(etype, evalue, etraceback) + pager = Pager() + for line in lines: + pager.write( + '\033[31m' + line.rstrip('\n') + '\033[0m\n') # red, reset + pager.close() + + sys.excepthook = excepthook + + #try: + # import cgitb + #except ImportError: + # pass + #else: + # # cgitb.enable() overrides sys.excepthook + # cgitb.enable(format='text') + + # From Thomas Heller: + # https://mail.python.org/pipermail/python-list/2001-April/099020.html + + # import pdb + # + # def info(*args): + # pdb.pm() + # sys.excepthook = info + + # utilities + + # From: Paul Magwene: + # https://mail.python.org/pipermail/python-list/2001-March/086191.html + # With a lot of my fixes: class DirLister: def __getitem__(self, key): @@ -145,18 +231,10 @@ def init(): path = os.path.expanduser(os.path.expandvars(path or '~')) os.chdir(path) - __builtin__.ls = DirLister() - __builtin__.cd = DirChanger() - - - # From Thomas Heller - # - #import pdb - # - #def info(*args): - # pdb.pm() - #sys.excepthook = info + builtins.ls = DirLister() + builtins.cd = DirChanger() + # print working directory class Pwd: def __repr__(self): @@ -165,8 +243,9 @@ def init(): def __call__(self): return repr(self) - __builtin__.pwd = Pwd() + builtins.pwd = Pwd() + # exit REPL with 'exit', 'quit' or simple 'x' class _Exit: def __repr__(self): @@ -175,10 +254,61 @@ def init(): def __call__(self, msg=None): sys.exit(msg) - __builtin__.x = _Exit() + builtins.x = _Exit() + + # In Python 2.5+ exit and quit are objects + if isinstance(builtins.exit, str): + builtins.exit = builtins.quit = x # noqa: x is defined as _Exit + + # print conten of a file + + class _Cat: + def __repr__(self): + return "Usage: cat('filename')" + + def __call__(self, filename): + fp = open(filename, 'rU') + text = fp.read() + fp.close() + print(text) + + builtins.cat = _Cat() + + # call shell + + class _Sh: + def __repr__(self): + os.system(os.environ["SHELL"]) + return '' + + def __call__(self, cmdline): + os.system(cmdline) + + builtins.sh = _Sh() + + # paginate a file + + class _Pager: + def __repr__(self): + return "Usage: pager('filename')" + + def __call__(self, filename): + os.system("%s '%s'" % (pager, filename.replace("'", '"\'"'))) + + builtins.pager = _Pager() + + # edit a file + + class _Editor: + def __repr__(self): + return "Usage: edit('filename')" + + def __call__(self, filename): + editor = os.environ.get("VISUAL") \ + or os.environ.get("EDITOR") or 'vi' + os.system("%s '%s'" % (editor, filename.replace("'", '"\'"'))) - if isinstance(__builtin__.exit, str): # In Python 2.5+ exit and quit are objects - __builtin__.exit = __builtin__.quit = x + builtins.edit = builtins.editor = _Editor() init()