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