def _get_imaging_task(orig_managerCls, orig_creds, dest_managerCls, dest_creds,
imaging_args):
# NOTE: destManagerCls may == origManagerCls,
# but creds MUST be different for a migration.
if dest_managerCls and dest_creds != orig_creds:
migrate_task = migrate_instance_task.si(orig_managerCls, orig_creds,
dest_managerCls, dest_creds,
**imaging_args)
return migrate_task
else:
image_task = machine_imaging_task.si(orig_managerCls, orig_creds,
imaging_args)
return image_task
def _get_imaging_task(orig_managerCls, orig_creds,
dest_managerCls, dest_creds, imaging_args):
# NOTE: destManagerCls may == origManagerCls,
# but creds MUST be different for a migration.
if dest_managerCls and dest_creds != orig_creds:
migrate_task = migrate_instance_task.si(
orig_managerCls, orig_creds, dest_managerCls, dest_creds,
**imaging_args)
return migrate_task
else:
image_task = machine_imaging_task.si(
orig_managerCls, orig_creds, imaging_args)
return image_task
def start_machine_imaging(machine_request, delay=False):
"""
Builds up a machine imaging task using core.models.machine_request
delay - If true, wait until task is completed before returning
"""
machine_request.status = 'imaging'
machine_request.save()
instance_id = machine_request.instance.provider_alias
(orig_managerCls, orig_creds, dest_managerCls,
dest_creds) = machine_request.prepare_manager()
imaging_args = machine_request.get_imaging_args()
#Step 1 - On OpenStack, sync/freeze BEFORE starting migration/imaging
init_task = None
#if orig_managerCls == OSImageManager: # TODO:AND if instance still running
# freeze_task = freeze_instance_task.si(
# machine_request.instance.created_by_identity_id, instance_id)
# init_task = freeze_task
if dest_managerCls and dest_creds != orig_creds:
#Will run machine imaging task..
migrate_task = migrate_instance_task.si(orig_managerCls, orig_creds,
dest_managerCls, dest_creds,
**imaging_args)
if not init_task:
init_task = migrate_task
else:
init_task.link(migrate_task)
else:
image_task = machine_imaging_task.si(orig_managerCls, orig_creds,
imaging_args)
if not init_task:
init_task = image_task
else:
init_task.link(image_task)
#The new image ID will be the first argument in process_request
process_task = process_request.s(machine_request.id)
if dest_managerCls and dest_creds != orig_creds:
migrate_task.link(process_task)
else:
image_task.link(process_task)
async = init_task.apply_async(link_error=machine_request_error.s(
machine_request.id), )
if delay:
async .get()
return async
def start_machine_imaging(machine_request, delay=False):
"""
Builds up a machine imaging task using core.models.machine_request
delay - If true, wait until task is completed before returning
"""
machine_request.status = "processing"
machine_request.save()
instance_id = machine_request.instance.provider_alias
(orig_managerCls, orig_creds, dest_managerCls, dest_creds) = machine_request.prepare_manager()
imaging_args = machine_request.get_imaging_args()
# Step 1 - On OpenStack, sync/freeze BEFORE starting migration/imaging
init_task = None
if orig_managerCls == OSImageManager: # TODO:AND if instance still running
freeze_task = freeze_instance_task.si(machine_request.instance.created_by_identity_id, instance_id)
init_task = freeze_task
if dest_managerCls and dest_creds != orig_creds:
# Will run machine imaging task..
migrate_task = migrate_instance_task.si(
orig_managerCls, orig_creds, dest_managerCls, dest_creds, **imaging_args
)
if not init_task:
init_task = migrate_task
else:
init_task.link(migrate_task)
else:
image_task = machine_imaging_task.si(orig_managerCls, orig_creds, imaging_args)
if not init_task:
init_task = image_task
else:
init_task.link(image_task)
# The new image ID will be the first argument in process_request
process_task = process_request.s(machine_request.id)
if dest_managerCls and dest_creds != orig_creds:
migrate_task.link(process_task)
else:
image_task.link(process_task)
async = init_task.apply_async(link_error=machine_request_error.s(machine_request.id))
if delay:
async.get()
return async
def start_machine_imaging(machine_request, delay=False):
"""
Builds up a machine imaging task using core.models.machine_request
delay - If true, wait until task is completed before returning
"""
machine_request.status = "imaging"
machine_request.save()
instance_id = machine_request.instance.provider_alias
(orig_managerCls, orig_creds, dest_managerCls, dest_creds) = machine_request.prepare_manager()
imaging_args = machine_request.get_imaging_args()
admin_driver = machine_request.new_admin_driver()
admin_ident = machine_request.new_admin_identity()
imaging_error_task = machine_request_error.s(machine_request.id)
init_task = None
if dest_managerCls and dest_creds != orig_creds:
# Task 1 = Migrate Task
migrate_task = migrate_instance_task.si(
orig_managerCls, orig_creds, dest_managerCls, dest_creds, **imaging_args
)
if not init_task:
init_task = migrate_task
else:
init_task.link(migrate_task)
else:
# Task 1 = Imaging Task
image_task = machine_imaging_task.si(orig_managerCls, orig_creds, imaging_args)
if not init_task:
init_task = image_task
else:
init_task.link(image_task)
# Task 2 = Process the machine request
# (Save tags, name, description, metadata, etc.)
process_task = process_request.s(machine_request.id)
process_task.link_error(imaging_error_task)
if dest_managerCls and dest_creds != orig_creds:
migrate_task.link(process_task)
migrate_task.link_error(imaging_error_task)
else:
image_task.link(process_task)
image_task.link_error(imaging_error_task)
# Task 3 = Validate the new image by launching an instance
validate_task = validate_new_image.s(machine_request.id)
process_task.link(validate_task)
# Task 4 = Wait for new instance to be 'active'
wait_for_task = wait_for.s(
admin_driver.__class__, admin_driver.provider, admin_driver.identity, "active", return_id=True
)
validate_task.link(wait_for_task)
validate_task.link_error(imaging_error_task)
# Task 5 = Terminate the new instance on completion
destroy_task = destroy_instance.s(admin_ident.id)
wait_for_task.link(destroy_task)
wait_for_task.link_error(imaging_error_task)
# Task 6 - Finally, email the user that their image is ready!
email_task = imaging_complete.s(machine_request.id)
destroy_task.link_error(imaging_error_task)
destroy_task.link(email_task)
# Start the task.
async = init_task.apply_async()
if delay:
async.get()
return async
def start_machine_imaging(machine_request, delay=False):
"""
Builds up a machine imaging task using core.models.machine_request
delay - If true, wait until task is completed before returning
"""
machine_request.status = 'imaging'
machine_request.save()
instance_id = machine_request.instance.provider_alias
(orig_managerCls, orig_creds, dest_managerCls,
dest_creds) = machine_request.prepare_manager()
imaging_args = machine_request.get_imaging_args()
admin_driver = machine_request.new_admin_driver()
admin_ident = machine_request.new_admin_identity()
imaging_error_task = machine_request_error.s(machine_request.id)
init_task = None
if dest_managerCls and dest_creds != orig_creds:
#Task 1 = Migrate Task
migrate_task = migrate_instance_task.si(orig_managerCls, orig_creds,
dest_managerCls, dest_creds,
**imaging_args)
if not init_task:
init_task = migrate_task
else:
init_task.link(migrate_task)
else:
#Task 1 = Imaging Task
image_task = machine_imaging_task.si(orig_managerCls, orig_creds,
imaging_args)
if not init_task:
init_task = image_task
else:
init_task.link(image_task)
#Task 2 = Process the machine request
# (Save tags, name, description, metadata, etc.)
process_task = process_request.s(machine_request.id)
process_task.link_error(imaging_error_task)
if dest_managerCls and dest_creds != orig_creds:
migrate_task.link(process_task)
migrate_task.link_error(imaging_error_task)
else:
image_task.link(process_task)
image_task.link_error(imaging_error_task)
#Task 3 = Validate the new image by launching an instance
validate_task = validate_new_image.s(machine_request.id)
process_task.link(validate_task)
#Task 4 = Wait for new instance to be 'active'
wait_for_task = wait_for.s(admin_driver.__class__,
admin_driver.provider,
admin_driver.identity,
"active",
return_id=True)
validate_task.link(wait_for_task)
validate_task.link_error(imaging_error_task)
#Task 5 = Terminate the new instance on completion
destroy_task = destroy_instance.s(admin_ident.id)
wait_for_task.link(destroy_task)
wait_for_task.link_error(imaging_error_task)
#Task 6 - Finally, email the user that their image is ready!
email_task = imaging_complete.s(machine_request.id)
destroy_task.link_error(imaging_error_task)
destroy_task.link(email_task)
# Start the task.
async = init_task.apply_async()
if delay:
async .get()
return async