def rollback_create(maintenance, task, user=None, instances=None):
topology_path = maintenance.plan.replication_topology.class_path
steps = get_deploy_settings(topology_path)
if instances is None:
instances = get_instances_for(maintenance.infra, topology_path)
maintenance.id = None
maintenance.user = user.username if user else task.user
maintenance.task = task
maintenance.save()
if rollback_for_instances_full(
steps,
instances,
task,
maintenance.get_current_step,
maintenance.update_step,
):
maintenance.set_rollback()
task.set_status_success('Rollback executed with success')
infra = maintenance.infra
infra.delete()
else:
maintenance.set_error()
task.set_status_error('Could not do rollback\n'
'Please check error message and do retry')
def create_database(name,
plan,
environment,
team,
project,
description,
task,
backup_hour,
maintenance_window,
maintenance_day,
subscribe_to_email_events=True,
is_protected=False,
user=None,
retry_from=None):
topology_path = plan.replication_topology.class_path
name = slugify(name)
base_name = gen_infra_names(name, 0)
infra = get_or_create_infra(base_name, plan, environment, backup_hour,
maintenance_window, maintenance_day,
retry_from)
instances = get_instances_for(infra, topology_path)
database_create = DatabaseCreate()
database_create.task = task
database_create.name = name
database_create.plan = plan
database_create.environment = environment
database_create.team = team
database_create.project = project
database_create.description = description
database_create.subscribe_to_email_events = subscribe_to_email_events
database_create.is_protected = is_protected
database_create.user = user.username if user else task.user
database_create.infra = infra
database_create.database = infra.databases.first()
database_create.save()
steps = get_deploy_settings(topology_path)
since_step = None
if retry_from:
since_step = retry_from.current_step
if steps_for_instances(steps,
instances,
task,
database_create.update_step,
since_step=since_step):
database_create.set_success()
task.set_status_success('Database created')
database_create.database.finish_task()
else:
database_create.set_error()
task.set_status_error('Could not create database\n'
'Please check error message and do retry')
def create_database(
name, plan, environment, team, project, description, task,
subscribe_to_email_events=True, is_protected=False, user=None,
retry_from=None
):
topology_path = plan.replication_topology.class_path
number_of_vms = get_deploy_instances_size(topology_path)
name = slugify(name)
base_name = gen_infra_names(name, number_of_vms)
infra = get_or_create_infra(base_name, plan, environment, retry_from)
instances = []
for i in range(number_of_vms):
try:
instance_name = '{}-0{}-{}'.format(
base_name['name_prefix'], i+1, base_name['name_stamp']
)
instance = infra.instances.get(
Q(hostname__hostname__startswith=instance_name) |
Q(dns__startswith=instance_name)
)
except Instance.DoesNotExist:
instance = Instance()
instance.dns = base_name['vms'][i]
instance.databaseinfra = infra
driver = infra.get_driver()
instance.port = driver.get_default_database_port()
instance.instance_type = driver.get_default_instance_type()
instance.vm_name = instance.dns
instances.append(instance)
database_create = DatabaseCreate()
database_create.task = task
database_create.name = name
database_create.plan = plan
database_create.environment = environment
database_create.team = team
database_create.project = project
database_create.description = description
database_create.subscribe_to_email_events = subscribe_to_email_events
database_create.is_protected = is_protected
database_create.user = '******'
database_create.infra = infra
database_create.database = infra.databases.first()
database_create.save()
steps = get_deploy_settings(topology_path)
since_step = None
if retry_from:
since_step = retry_from.current_step
if steps_for_instances(
steps, instances, task, database_create.update_step,
since_step=since_step
):
database_create.set_success()
task.set_status_success('Database created')
else:
database_create.set_error()
task.set_status_error(
'Could not create database\n'
'Please check error message and do retry'
)
