zeroinstall/cmd/__init__.py

   1 """
   2 The B{0install} command-line interface.
   3 """
   4
   5 # Copyright (C) 2011, Thomas Leonard
   6 # See the README file for details, or visit http://0install.net.
   7
   8 from __future__ import print_function
   9
  10 from zeroinstall import _, logger
  11 import os, sys
  12 from optparse import OptionParser
  13 import logging
  14
  15 from zeroinstall import SafeException
  16
  17 valid_commands = ['add', 'select', 'download', 'run', 'update', 'whatchanged', 'apps', 'destroy',
  18                   'config', 'import', 'list', 'add-feed', 'remove-feed', 'list-feeds',
  19                   'digest']
  20
  21 class UsageError(Exception): pass
  22
  23 def _ensure_standard_fds():
  24         "Ensure stdin, stdout and stderr FDs exist, to avoid confusion."
  25         for std in (0, 1, 2):
  26                 try:
  27                         os.fstat(std)
  28                 except OSError:
  29                         fd = os.open(os.devnull, os.O_RDONLY)
  30                         if fd != std:
  31                                 os.dup2(fd, std)
  32                                 os.close(fd)
  33
  34 def _no_command(command_args):
  35         """Handle --help and --version"""
  36         parser = OptionParser(usage=_("usage: %prog COMMAND\n\nTry --help with one of these:") +
  37                         "\n\n0install " + '\n0install '.join(valid_commands))
  38         parser.add_option("-V", "--version", help=_("display version information"), action='store_true')
  39
  40         (options, args) = parser.parse_args(command_args)
  41         if options.version:
  42                 import zeroinstall
  43                 print("0install (zero-install) " + zeroinstall.version)
  44                 print("Copyright (C) 2011 Thomas Leonard")
  45                 print(_("This program comes with ABSOLUTELY NO WARRANTY,"
  46                                 "\nto the extent permitted by law."
  47                                 "\nYou may redistribute copies of this program"
  48                                 "\nunder the terms of the GNU Lesser General Public License."
  49                                 "\nFor more information about these matters, see the file named COPYING."))
  50                 sys.exit(0)
  51         parser.print_help()
  52         sys.exit(2)
  53
  54 def main(command_args, config = None):
  55         """Act as if 0install was run with the given arguments.
  56         @arg command_args: array of arguments (e.g. C{sys.argv[1:]})
  57         @type command_args: [str]
  58         """
  59         _ensure_standard_fds()
  60
  61         if config is None:
  62                 from zeroinstall.injector.config import load_config
  63                 config = load_config()
  64
  65         # The first non-option argument is the command name (or "help" if none is found).
  66         command = None
  67         for i, arg in enumerate(command_args):
  68                 if not arg.startswith('-'):
  69                         command = arg
  70                         del command_args[i]
  71                         break
  72                 elif arg == '--':
  73                         break
  74
  75         verbose = False
  76         try:
  77                 if command is None:
  78                         return _no_command(command_args)
  79
  80                 if command not in valid_commands:
  81                         raise SafeException(_("Unknown sub-command '%s': try --help") % command)
  82
  83                 # Configure a parser for the given command
  84                 module_name = command.replace('-', '_')
  85                 cmd = __import__('zeroinstall.cmd.' + module_name, globals(), locals(), [module_name], 0)
  86                 parser = OptionParser(usage=_("usage: %%prog %s [OPTIONS] %s") % (command, cmd.syntax))
  87
  88                 parser.add_option("-c", "--console", help=_("never use GUI"), action='store_false', dest='gui')
  89                 parser.add_option("", "--dry-run", help=_("just print what would be executed"), action='store_true')
  90                 parser.add_option("-g", "--gui", help=_("show graphical policy editor"), action='store_true')
  91                 parser.add_option("-v", "--verbose", help=_("more verbose output"), action='count')
  92                 parser.add_option("", "--with-store", help=_("add an implementation cache"), action='append', metavar='DIR')
  93
  94                 cmd.add_options(parser)
  95                 (options, args) = parser.parse_args(command_args)
  96                 verbose = options.verbose
  97
  98                 if options.verbose:
  99                         if options.verbose == 1:
 100                                 logger.setLevel(logging.INFO)
 101                         else:
 102                                 logger.setLevel(logging.DEBUG)
 103                         import zeroinstall
 104                         logger.info(_("Running 0install %(version)s %(args)s; Python %(python_version)s"), {'version': zeroinstall.version, 'args': repr(command_args), 'python_version': sys.version})
 105
 106                 if options.with_store:
 107                         from zeroinstall import zerostore
 108                         for x in options.with_store:
 109                                 config.stores.stores.append(zerostore.Store(os.path.abspath(x)))
 110                         logger.info(_("Stores search path is now %s"), config.stores.stores)
 111
 112                 config.handler.dry_run = bool(options.dry_run)
 113
 114                 cmd.handle(config, options, args)
 115         except KeyboardInterrupt:
 116                 logger.info("KeyboardInterrupt")
 117                 sys.exit(1)
 118         except UsageError:
 119                 parser.print_help()
 120                 sys.exit(1)
 121         except SafeException as ex:
 122                 if verbose: raise
 123                 try:
 124                         from zeroinstall.support import unicode
 125                         print(unicode(ex), file=sys.stderr)
 126                 except:
 127                         print(repr(ex), file=sys.stderr)
 128                 sys.exit(1)
 129         return