tahoe-lafs/misc/run-with-pythonpath.py

51 lines
1.4 KiB
Python

# -*- python -*-
# you must invoke this with an explicit python, from the tree root
"""Run an arbitrary command with a PYTHONPATH that will include the Tahoe
code, including dependent libraries. Run this like:
python misc/run-with-pythonpath.py python foo.py
or
python misc/run-with-pythonpath.py trial -r poll allmydata.test.test_util
"""
import os, sys, subprocess
# figure out where support/lib/pythonX.X/site-packages is
# add it to os.environ["PYTHONPATH"]
# spawn the child process
def pylibdir(prefixdir):
pyver = "python%d.%d" % (sys.version_info[:2])
if sys.platform == "win32":
return os.path.join(prefixdir, "Lib", "site-packages")
else:
return os.path.join(prefixdir, "lib", pyver, "site-packages")
basedir = os.path.dirname(os.path.abspath(__file__))
supportlib = pylibdir(os.path.abspath("support"))
oldpp = os.environ.get("PYTHONPATH", "").split(os.pathsep)
if oldpp == [""]:
# grr silly split() behavior
oldpp = []
newpp = os.pathsep.join(oldpp + [supportlib,])
os.environ['PYTHONPATH'] = newpp
if sys.platform == "win32":
command = sys.argv[1:]
rc = subprocess.call(command)
sys.exit(rc)
else:
from twisted.python.procutils import which
cmd = sys.argv[1]
if cmd and cmd[0] not in "/~.":
cmds = which(cmd)
if not cmds:
print >>sys.stderr, "'%s' not found on PATH" % (cmd,)
sys.exit(-1)
cmd = cmds[0]
os.execve(cmd, sys.argv[1:], os.environ)