bin/tahoe-script.template

   1 #!/bin/false # You must specify a python interpreter.
   2 u"Tahoe-LAFS does not run under Python 3. Please use a version of Python between 2.4.4 and 2.7.x inclusive."
   3
   4 import sys, os, subprocess
   5
   6 where = os.path.realpath(sys.argv[0])
   7 base = os.path.dirname(os.path.dirname(where))
   8
   9 if sys.platform == "win32":
  10     perhaps_installed_tahoe = os.path.join(os.path.dirname(sys.executable), 'Scripts', 'tahoe.pyscript')
  11 else:
  12     perhaps_installed_tahoe = "/usr/bin/tahoe"
  13
  14 whoami = '''\
  15 I am a "bin%stahoe" executable who is only for the convenience of running
  16 Tahoe from its source distribution -- I work only when invoked as the "tahoe"
  17 script that lives in the "bin" subdirectory of a Tahoe source code
  18 distribution, and only if you have already run "python setup.py build".
  19 ''' % (os.path.sep,)
  20
  21 # look for Tahoe.home .
  22 homemarker = os.path.join(base, "Tahoe.home")
  23 if not os.path.exists(homemarker):
  24     print whoami
  25     print '''\
  26 I just tried to run and found that I am not living in such a directory, so I
  27 am stopping now. To run Tahoe after it has been is installed, please execute
  28 my brother, who gets installed into the appropriate place for executables
  29 when you run "make install" (perhaps as "%s").
  30 ''' % (perhaps_installed_tahoe,)
  31     sys.exit(1)
  32
  33 # we've found our home. Put the tahoe support/lib etc. in our PYTHONPATH.
  34 if sys.platform == "win32":
  35     supportdir = os.path.join(base, "support", "Lib", "site-packages")
  36 else:
  37     supportdir = os.path.join(base, "support",
  38                               "lib",
  39                               "python%d.%d" % sys.version_info[:2],
  40                               "site-packages")
  41
  42 # update PYTHONPATH so that child processes (like twistd) will use this too
  43 pp = os.environ.get("PYTHONPATH")
  44 if pp:
  45     pp = os.pathsep.join([supportdir] + pp.split(os.pathsep))
  46 else:
  47     pp = supportdir
  48 os.environ["PYTHONPATH"] = pp
  49
  50 # find commandline args and the location of the tahoe executable.
  51 if sys.platform == "win32":
  52     import re
  53     from ctypes import WINFUNCTYPE, POINTER, byref, c_wchar_p, c_int, windll
  54
  55     GetCommandLineW = WINFUNCTYPE(c_wchar_p)(("GetCommandLineW", windll.kernel32))
  56     CommandLineToArgvW = WINFUNCTYPE(POINTER(c_wchar_p), c_wchar_p, POINTER(c_int)) \
  57                             (("CommandLineToArgvW", windll.shell32))
  58
  59     argc = c_int(0)
  60     argv_unicode = CommandLineToArgvW(GetCommandLineW(), byref(argc))
  61
  62     # See src/allmydata/scripts/runner.py for the corresponding unmangler.
  63     # Note that this doesn't escape \x7F. If it did, test_unicode_arguments_and_output
  64     # in test_runner.py wouldn't work.
  65     def mangle(s):
  66         return str(re.sub(ur'[^\x20-\x7F]', lambda m: u'\x7F%x;' % (ord(m.group(0)),), s))
  67
  68     argv = [mangle(argv_unicode[i]) for i in xrange(0, argc.value)]
  69
  70     # Take only the suffix with the same number of arguments as sys.argv.
  71     # This accounts for anything that can cause initial arguments to be stripped,
  72     # for example, the Python interpreter or any options passed to it, or runner
  73     # scripts such as 'coverage run'. It works even if there are no such arguments,
  74     # as in the case of a frozen executable created by bb-freeze or similar.
  75
  76     argv = argv[-len(sys.argv):]
  77
  78     # On Windows, the script is not directly executable and must be run via python.
  79     prefix = [sys.executable]
  80     script = os.path.join(base, "support", "Scripts", "tahoe.pyscript")
  81     args = argv[1:]
  82 else:
  83     # On non-Windows, invoke the script directly, so that 'top' for example shows 'tahoe'.
  84     prefix = []
  85     script = os.path.join(base, "support", "bin", "tahoe")
  86     args = sys.argv[1:]
  87
  88 if not os.path.exists(script):
  89     print whoami
  90     print '''\
  91 I just tried to run and could not find my brother at
  92 "%s". To run Tahoe when it is installed, please execute my
  93 brother, who gets installed into the appropriate place for executables
  94 when you run "python setup.py install" (perhaps as "%s").
  95 ''' % (script, perhaps_installed_tahoe)
  96     sys.exit(1)
  97
  98 # Support indirection via another "runner" script (e.g. coverage).
  99 # For example: bin/tahoe @RUNNER RUNNER_ARGS @tahoe TAHOE_ARGS
 100
 101 if len(args) >= 1 and args[0].startswith('@'):
 102     runner = args[0][1:]
 103     if runner.endswith('.py') or runner.endswith('.pyscript'):
 104         prefix = [sys.executable]
 105     else:
 106         prefix = []
 107
 108     def _subst(a):
 109         if a == '@tahoe': return script
 110         return a
 111     command = prefix + [runner] + map(_subst, args[1:])
 112 else:
 113     command = prefix + [script] + args
 114
 115 try:
 116     res = subprocess.call(command, env=os.environ)
 117 except Exception, le:
 118     print whoami
 119     print '''\
 120 I just tried to invoke my brother at "%s"
 121 and got an exception.
 122 ''' % (script,)
 123     raise
 124 else:
 125     sys.exit(res)