You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
3.8 KiB
Python
102 lines
3.8 KiB
Python
#!/usr/bin/env python
|
|
# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
"""Wrapper for chromite tools.
|
|
|
|
The script is intend to be symlinked to any number of chromite tools, attempts
|
|
to find the path for chromite, and runs the right tool based on argv[0].
|
|
|
|
It is intended to run in several ways:
|
|
- Outside the chroot, it should be _copied_ to someplace that is in the
|
|
path (like depot_tools). It will search for the right chromite by looking
|
|
for a file 'chromite/shell/main.py' upward based on the CWD.
|
|
- Inside the chroot, it might be _either_ copied to someplace in the path (since
|
|
depot_tools is in the path in the chroot) or it might run from chromite/bin
|
|
directly, which should be in the PATH. In any case, we'll look for the
|
|
real 'chromite/shell/main.py' based on the environment variable
|
|
CROS_WORKON_SRCROOT, so it doesn't matter what the CWD is.
|
|
|
|
If you're looking at a copy and want to know where the original looks at, look
|
|
here:
|
|
http://git.chromium.org/gitweb/?p=chromite.git;a=blob;f=bin/chromite
|
|
|
|
Since this script is _copied_, it should remain small and not use internal libs.
|
|
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
|
|
|
|
def Search(path):
|
|
"""Return an iterator of lists of places to look for chromite."""
|
|
|
|
if os.path.exists('/etc/debian_chroot'):
|
|
# We're in the chroot. Chromite should be in the python path inside the
|
|
# chroot, so we don't do any searching. NOTE that we purposely don't want
|
|
# CROS_WORKON_SRCROOT in the python path.
|
|
yield []
|
|
else:
|
|
# Look in $CROS_WORKON_SRCROOT first. The idea is that a user would set
|
|
# this manually if they wanted to specify a particular version of chromite.
|
|
if 'CROS_WORKON_SRCROOT' in os.environ:
|
|
yield [os.environ['CROS_WORKON_SRCROOT']]
|
|
|
|
# Search upward until we either end up with a blank dir or the "parent" dir
|
|
# doesn't change.
|
|
prev_path = None
|
|
while path and path != prev_path:
|
|
yield [path]
|
|
path, prev_path = os.path.dirname(path), path
|
|
|
|
|
|
def main():
|
|
# Decide what the name of the called binary will be.
|
|
our_name = os.path.basename(sys.argv[0])
|
|
# Some tools map to an arbitrary path, rather than to a .py file of the same
|
|
# exact name. By default, we pick the tool from chromite/bin/
|
|
TOOLS_MAP = {
|
|
'chromite' : 'shell.main',
|
|
'cbuildbot' : 'buildbot.cbuildbot',
|
|
}
|
|
our_name = TOOLS_MAP.get(our_name, 'bin.' + our_name)
|
|
|
|
for path in Search(os.getcwd()):
|
|
sys.path = path + sys.path
|
|
try:
|
|
exec 'import chromite.' + our_name + ' as our_tool'
|
|
break
|
|
except ImportError, e:
|
|
# Just in case there is actually something wrong with Chromite, print
|
|
# a sensible error. We match only the end of the string so that we can
|
|
# handle an error within the chromite directory.
|
|
# The full error is 'No module named (chromite.)shell.main'
|
|
# Note: If you hit the directory containing chromite on the way up, then
|
|
# the error will be 'No module named shell.main' so we must check only the
|
|
# shell.main part.
|
|
if not str(e).endswith(our_name):
|
|
raise
|
|
|
|
# We've got different modules named chromite in the tree, pulling in the
|
|
# wrong one will break the right one. So unload it.
|
|
if 'chromite' in sys.modules:
|
|
del sys.modules['chromite']
|
|
sys.path = sys.path[len(path):]
|
|
else:
|
|
# TODO(dianders): Should we actually print out the 'repo init' call that
|
|
# the user should use?
|
|
sys.stderr.write(
|
|
"ERROR: Couldn't find the chromite tool.\n"
|
|
"\n"
|
|
"Please change to a directory inside your Chromium OS source tree\n"
|
|
"and retry. If you need to setup a Chromium OS source tree, see:\n"
|
|
" http://www.chromium.org/chromium-os/developer-guide\n")
|
|
return 1
|
|
return our_tool.main()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|