My tools of the trade for python programming.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

176 lines
5.0 KiB

# $Id: interactive_tools.py,v 1.2 2011-07-18 15:48:52 wirawan Exp $
#
# wpylib.interactive_tools
# Created: 20091204
# Wirawan Purwanto
#
# Simple and dirty tools for interactive python mode
#
# FIXME TODO LIST:
# - needs to detect 'python -i' presence in case of python <= 2.5.
import atexit
import inspect
import os
import sys
MSGLOG = sys.stderr
def get_global_namespace_(global_ns):
"""Internal routine to get global namespace from another function
in this module.
BEWARE the limitation! This function is rather fragile and must be
called such that the stack order is like this (in descending
order):
* global module statement ...
* a function from this module
* this function
"""
import pprint
if global_ns == None:
# Use introspection to fetch the global namespace of the calling
# function:
from inspect import stack
caller_info = stack()[2] # by default, _two_ level up
caller_frame = caller_info[0] # get the frame record
g = caller_frame.f_globals
else:
g = global_ns
try:
builtin_ns = g['__builtins__']
except:
builtin_ns = None # this should not ever happen!
if False: # for debugging only
pprint.pprint(stack())
print "caller_info = ",
pprint.pprint(caller_info)
pprint (sorted(g.keys()))
pprint (g['__builtins__'])
return (g, builtin_ns)
def detect_interactive(global_ns=None):
"""Detects if we are in an interactive python session.
"""
"""The detection scheme is rather complex one because each mode has its
own way of flagging that it is ``interactive''.
See some discussions in:
http://stackoverflow.com/questions/1212779/detecting-when-a-python-script-is-being-run-interactively
These are some known cases:
* For ipython, see if "__IPYTHON__" is defined in the user's global
namespace.
* For "python -i" invocation, sys.flags.interactive will be set to True.
(python 2.6+ only)
* For vanilla "python" invocation with no argument, then sys.ps1
variable exists.
"""
def is_python_i():
# This approach only works for Python 2.6+
try:
return sys.flags.interactive
except:
return False
(g, b) = get_global_namespace_(global_ns)
if "__IPYTHON__" in g or hasattr(b, "__IPYTHON__"):
return {
'session': 'ipython',
}
elif is_python_i(): # sys.flags.interactive:
return {
'session': 'python -i',
}
elif hasattr(sys, 'ps1'):
return {
'session': 'python',
}
else:
return False
def init_interactive(use_readline=True, global_ns=None):
"""Perform standard initialization for my scripts.
Some options can be given to tweak the environment.
CAVEAT: this method still does not work universally yet.
We need to execute some of the statements in the global (top level)
namespace, not in this function's namespace.
So in general init_interactive *MUST* at this point be called from the
main program, not from any subroutine.
Otherwise, you must explicitly specify the global namespace as the
`global_ns` argument.
Works under ipython, `python -i' and vanilla python.
"""
(g, b) = get_global_namespace_(global_ns)
int_info = detect_interactive(g)
if use_readline:
# Assume interactive python, create a completer if it hasn't been
# loaded. This is needed for e.g. sciclone, jaguar with no ipython
# there.
if 'rlcompleter' not in sys.modules \
and 'IPython.ipy_completers' not in sys.modules:
try:
import readline
except ImportError:
print >>MSGLOG, "Module readline not available."
else:
print "Loading classic readline/rlcompleter."
import rlcompleter
readline.parse_and_bind("tab: complete")
_histfile = os.path.join(os.environ["HOME"], ".pyhistory")
# read history, if it exists
if os.path.exists(_histfile):
readline.read_history_file(_histfile)
# register saving handler
atexit.register(readline.write_history_file, _histfile)
del _histfile
# Interestingly, under ipython the sys.argv is only "correct" to extract
# the program name at the first loading of the script.
# So we must capture that:
if not int_info:
pass
#print >>MSGLOG, "init_interactive: Cannot detect python session type."
return False
elif int_info['session'] == 'ipython' and 'ARGV' not in g:
g['ARGV'] = sys.argv
try:
g['MYSELF'] = os.path.abspath(sys.argv[0])
except:
pass
return True
elif int_info['session'] in ('python', 'python -i'):
g.setdefault('MYSELF', sys.argv[0])
g.setdefault('ARGV', sys.argv)
return True
#print "_-helo"
class printstr:
"""A hack to record printout in a logged interactive python session
(I had logged ipython in mind, but other similar framework can use
this as well).
We intentionally use __repr__ to print the string.
Usage:
printstr(<python object>)
"""
def __init__(self, obj):
self.str = str(obj)
def __str__(self):
return self.str
def __repr__(self):
return self.str