Reduce concurrency; update TODO.

[wizard.git] / bin / wizard

diff --git a/bin/wizard b/bin/wizard
index c27919b73fdd4996d44b69bad360ec3bd0acfc2c..daf50c303655c8513ad7b1b00e07c553b35ce05f 100755 (executable)
--- a/bin/wizard
+++ b/bin/wizard
@@ -3,28 +3,44 @@
 import os
 import optparse
 import sys
+import logging
+import traceback
 
-sys.path.insert(0,os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+# import some non-standard modules to make it fail out early
+import decorator
+
+sys.path.insert(0,os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
 
 import wizard
-from wizard import command
+from wizard import command, prompt
 
 def main():
     usage = """usage: %prog COMMAND [ARGS]
 
 Wizard is a Git-based autoinstall management system for scripts.
 
-Its commands are:
-    configure       Configures an autoinstall (database, etc) to work
-    errors          Lists all broken autoinstall metadata
+User commands:
+    backup          Backup data not on filesystem (database, etc)
     install         Installs an application
+    migrate         Migrate autoinstalls from old format to Git-based format
+    remove          Removes an autoinstall, databases and other files
+    restore         Restores files and database to previous version
+    upgrade         Upgrades an autoinstall to the latest version
+
+Administrative commands:
+    blacklist       Marks an autoinstall to not try upgrades
+    errors          Lists all broken autoinstall metadata
     list            Lists autoinstalls, with optional filtering
     mass-migrate    Performs mass migration of autoinstalls of an application
-    migrate         Migrate autoinstalls from old format to Git-based format
-    prepare-config  Prepares configuration files for versioning
+    mass-upgrade    Performs mass upgrade of autoinstalls of an application
     research        Print statistics about a possible upgrade
     summary         Generate statistics (see help for subcommands)
-    upgrade         Upgrades an autoinstall to the latest version
+
+Utility commands:
+    prepare-pristine    Downloads and extracts pristine upstream files
+    prepare-new     Prepares a new repository
+    prepare-config  Prepares configuration files for versioning
+    quota           Prints the usage and available quota of a directory
 
 See '%prog help COMMAND' for more information on a specific command."""
 
@@ -43,19 +59,21 @@ See '%prog help COMMAND' for more information on a specific command."""
             default=False, help="Performs the operation without actually modifying any files.  Use in combination with --verbose to see commands that will be run.")
     # common variables for mass commands
     baton.add("--seen", dest="seen",
-            default=None, help="File to read/write paths of already processed installs."
-            "These will be skipped.")
+            default=None, help="File to read/write paths of successfully modified installs;"
+            "these will be skipped on re-runs.  If --log-dir is specified, this is automatically enabled.")
     baton.add("--no-parallelize", dest="no_parallelize", action="store_true",
             default=False, help="Turn off parallelization")
     baton.add("--max-processes", dest="max_processes", type="int", metavar="N",
-            default=10, help="Maximum subprocesses to run concurrently")
+            default=5, help="Maximum subprocesses to run concurrently")
     baton.add("--limit", dest="limit", type="int",
             default=None, help="Limit the number of autoinstalls to look at.")
+    baton.add("--user", "-u", dest="user",
+            default=None, help="Only mass migrate a certain user's installs.  No effect if versions_path is a file.")
     try:
         command_name = args[0]
     except IndexError:
         parser.print_help()
-        raise SystemExit(1)
+        sys.exit(1)
     baton.add("--log-dir", dest="log_dir",
         default=getenvpath("WIZARD_LOG_DIR") or "/tmp/wizard-%s" % command_name,
         help="Log files for Wizard children processes are placed here.")
@@ -66,14 +84,32 @@ See '%prog help COMMAND' for more information on a specific command."""
             parser.error("invalid action")
         except IndexError:
             parser.print_help()
-            raise SystemExit(1)
+            sys.exit(1)
         help_module.main(['--help'], baton)
     # Dispatch commands
+    command_module = get_command(command_name)
     try:
-        command_module = get_command(command_name)
-    except ImportError:
-        parser.error("invalid action")
-    command_module.main(rest_argv, baton)
+        command_module.main(rest_argv, baton)
+    except prompt.UserCancel as e:
+        print str(e)
+        sys.exit(1)
+    except Exception as e:
+        # log the exception
+        msg = traceback.format_exc()
+        if command.logging_setup:
+            outfun = logging.error
+        else:
+            outfun = sys.stderr.write
+        if isinstance(e, wizard.Error):
+            if e.quiet and not command.debug:
+                msg = str(e)
+                if command.logging_setup:
+                    msg = msg.replace("ERROR: ", "")
+            outfun(msg)
+            sys.exit(e.exitcode)
+        else:
+            outfun(msg)
+            sys.exit(1)
 
 def get_command(name):
     name = name.replace("-", "_")