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

40 lines
1.1 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
command = sys.argv[1:]
rc = subprocess.call(command)
sys.exit(rc)