bootstrap-vz/base/main.py

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')