mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 07:26:29 +00:00
a4ead02a9b
Manifest and module loading has been refactored Provider modules now must implement validate_manifest like plugins do Simplified loading of manifests
56 lines
2 KiB
Python
56 lines
2 KiB
Python
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def main():
|
|
import log
|
|
args = get_args()
|
|
logfile = log.get_logfile_path(args.manifest)
|
|
log.setup_logger(logfile=logfile, debug=args.debug)
|
|
run(args)
|
|
|
|
|
|
def get_args():
|
|
from argparse import ArgumentParser
|
|
parser = ArgumentParser(description='Bootstrap Debian for the cloud.')
|
|
parser.add_argument('--debug', action='store_true',
|
|
help='Print debugging information')
|
|
parser.add_argument('--pause-on-error', action='store_true',
|
|
help='Pause on error, before rollback')
|
|
parser.add_argument('--dry-run', action='store_true',
|
|
help='Dont\'t actually run the tasks')
|
|
parser.add_argument('manifest', help='Manifest file to use for bootstrapping', metavar='MANIFEST')
|
|
return parser.parse_args()
|
|
|
|
|
|
def run(args):
|
|
from manifest import Manifest
|
|
manifest = Manifest(args.manifest)
|
|
provider = manifest.modules['provider']
|
|
plugins = manifest.modules['plugins']
|
|
|
|
from tasklist import TaskList
|
|
tasklist = TaskList()
|
|
provider.resolve_tasks(tasklist, manifest)
|
|
for plugin in plugins:
|
|
plugin.resolve_tasks(tasklist, manifest)
|
|
|
|
from bootstrapinfo import BootstrapInformation
|
|
bootstrap_info = BootstrapInformation(manifest=manifest, debug=args.debug)
|
|
|
|
try:
|
|
tasklist.run(info=bootstrap_info, dry_run=args.dry_run)
|
|
log.info('Successfully completed bootstrapping')
|
|
except (Exception, KeyboardInterrupt) as e:
|
|
log.exception(e)
|
|
if args.pause_on_error:
|
|
raw_input("Press Enter to commence rollback")
|
|
log.error('Rolling back')
|
|
rollback_tasklist = TaskList()
|
|
provider.resolve_rollback_tasks(rollback_tasklist, tasklist.tasks_completed, manifest)
|
|
for plugin in plugins:
|
|
resolve_rollback_tasks = getattr(plugin, 'resolve_rollback_tasks', None)
|
|
if callable(resolve_rollback_tasks):
|
|
resolve_rollback_tasks(rollback_tasklist, tasklist.tasks_completed, manifest)
|
|
rollback_tasklist.run(info=bootstrap_info, dry_run=args.dry_run)
|
|
log.info('Successfully completed rollback')
|