def reassign():
# NOTE(akscram): Initialization of fuelclient.
fuel_config = flask.current_app.config["CLOUDS"]["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 = hooks.source.config()
dst_config = hooks.destination.config()
config = {
"source": src_config["environment"],
"destination": dst_config["environment"],
}
try:
src = hooks.source.connect()
dst = hooks.destination.connect()
ctx = context.Context(config, src, dst)
flow = node_tasks.reassign_node(ctx, host_id)
LOG.debug("Reassigning flow: %s", flow)
result = flows.run_flow(flow, ctx.store)
LOG.debug("Result of migration: %s", result)
except Exception:
msg = ("Error is occured during reassigning host {}"
.format(host_id))
LOG.exception(msg)
events.emit("error", {
"message": msg,
}, namespace="/events")
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)
