def cleanup(plugins, events, cloud, target, dump=None): env = reset_tasks.Environment(cloud, plugins) runner = tasks_base.TaskflowRunner(env) cleanup_workload = runner.get_resource(reset_tasks.CleanupWorkload, {"id": cloud.name}) runner.add(cleanup_workload.delete) if dump: with open(dump, "w") as f: utils.dump_flow(runner.create_flow(), f, True) else: runner.run()
def setup(plugins, events, cloud, target, num_tenants, num_servers, num_volumes, workloads, dump=None): env = reset_tasks.Environment(cloud, plugins) runner = tasks_base.TaskflowRunner(env) setup_workload = runner.get_resource(reset_tasks.SetupWorkload, { "id": cloud.name, "populate": { "num_tenants": num_tenants, "num_servers": num_servers, "num_volumes": num_volumes, }, "workloads": workloads, }) runner.add(setup_workload.create) if dump: with open(dump, "w") as f: utils.dump_flow(runner.create_flow(), f, True) else: runner.run()
def main(): args = get_parser().parse_args() logging.basicConfig(level=logging.INFO) events = Events() Cloud, Identity = load_cloud_driver(is_fake=args.fake) clouds_config = args.config["CLOUDS"] plugins_config = args.config["PLUGINS"] if args.action == "migrate": flow = graph_flow.Flow("migrate-resources") store = {} src_config = clouds_config["source"] src = init_client(src_config, "source", Cloud, Identity) if args.setup: workloads = clouds_config["source"].get("workloads", {}) management.setup(events, src, "source", args.num_tenants, args.num_servers, args.num_volumes, workloads) dst_config = clouds_config["destination"] dst = init_client(dst_config, "destination", Cloud, Identity) migrate_function = RESOURCES_MIGRATIONS[args.resource] if args.ids: ids = args.ids elif args.tenant: ids = get_ids_by_tenant(src, args.resource, args.tenant) elif args.host: ids = get_ids_by_host(src, args.resource, args.host) else: raise exceptions.UsageError("Missing tenant ID") ctx = context.Context(plugins_config, src, dst) resources_flow = migrate_function(ctx, flow, ids) if (args.dump): with open(args.dump, "w") as f: utils.dump_flow(resources_flow, f, True) return 0 flows.run_flow(resources_flow, ctx.store) elif args.action == "cleanup": cloud_config = clouds_config[args.target] cloud = init_client(cloud_config, args.target, Cloud, Identity) management.cleanup(events, cloud, args.target) elif args.action == "setup": src_config = clouds_config["source"] src = init_client(src_config, "source", Cloud, Identity) workloads = clouds_config["source"].get("workloads", {}) management.setup(plugins_config, events, src, "source", args.num_tenants, args.num_servers, args.num_volumes, workloads) elif args.action == "evacuate": src = init_client(clouds_config["source"], "source", Cloud, Identity) dst = init_client(clouds_config["destination"], "destination", Cloud, Identity) ctx = context.Context(plugins_config, src, dst) flow = evacuation_tasks.evacuate_servers(ctx, args.hostname) if (args.dump): with open(args.dump, "w") as f: utils.dump_flow(flow, f, True) return flows.run_flow(flow, ctx.store) elif args.action == "reassign": fuel_config = clouds_config["fuel"]["endpoint"] os.environ["SERVER_ADDRESS"] = fuel_config["host"] os.environ["LISTEN_PORT"] = str(fuel_config["port"]) os.environ["KEYSTONE_USER"] = fuel_config["username"] os.environ["KEYSTONE_PASS"] = fuel_config["password"] src_config = clouds_config["source"] dst_config = clouds_config["destination"] config = { "source": src_config["environment"], "destination": dst_config["environment"], } src = init_client(src_config, "source", Cloud, Identity) dst = init_client(dst_config, "destination", Cloud, Identity) ctx = context.Context(config, src, dst) flow = reassignment_tasks.reassign_node(ctx, args.hostname) if (args.dump): with open(args.dump, "w") as f: utils.dump_flow(flow, f, True) return flows.run_flow(flow, ctx.store)