mach

   1 #!/usr/bin/env python
   2 # This Source Code Form is subject to the terms of the Mozilla Public
   3 # License, v. 2.0. If a copy of the MPL was not distributed with this
   4 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   5
   6 from __future__ import print_function, unicode_literals
   7
   8 import os
   9 import platform
  10 import sys
  11
  12 # Ensure we are running Python 2.7+. We put this check here so we generate a
  13 # user-friendly error message rather than a cryptic stack trace on module
  14 # import.
  15 if sys.version_info[0] == 2 and sys.version_info[1] < 7:
  16     print('Python 2.7 or above is required to run mach.')
  17     print('You are running', platform.python_version())
  18     sys.exit(1)
  19
  20 # TODO Bug 794506 Integrate with the in-tree virtualenv configuration.
  21 SEARCH_PATHS = [
  22     'python/mach',
  23     'python/mozbuild',
  24     'build',
  25     'build/pymake',
  26     'python/blessings',
  27     'python/psutil',
  28     'python/which',
  29     'other-licenses/ply',
  30     'xpcom/idl-parser',
  31     'testing',
  32     'testing/xpcshell',
  33     'testing/mozbase/mozprocess',
  34     'testing/mozbase/mozinfo',
  35 ]
  36
  37 # Individual files providing mach commands.
  38 MACH_MODULES = [
  39     'layout/tools/reftest/mach_commands.py',
  40     'testing/mochitest/mach_commands.py',
  41     'testing/xpcshell/mach_commands.py',
  42 ]
  43
  44 our_dir = os.path.dirname(os.path.abspath(__file__))
  45
  46 try:
  47     import mach.main
  48 except ImportError:
  49     sys.path[0:0] = [os.path.join(our_dir, path) for path in SEARCH_PATHS]
  50
  51     import mach.main
  52
  53 # All of the code is in a module because EVERYTHING IS A LIBRARY.
  54 mach = mach.main.Mach(our_dir)
  55 mach.load_commands_from_sys_path()
  56
  57 for path in MACH_MODULES:
  58     mach.load_commands_from_file(os.path.join(our_dir, path))
  59
  60 sys.exit(mach.run(sys.argv[1:]))