def delete_azure_virtual_machine(cloud_service_name, deployment_name, virtual_machine_name):
cs = db_adapter.find_first_object_by(AzureCloudService, name=cloud_service_name)
dm = db_adapter.find_first_object_by(AzureDeployment, name=deployment_name, cloud_service=cs)
db_adapter.delete_all_objects_by(AzureVirtualMachine,
name=virtual_machine_name,
deployment_id=dm.id)
db_adapter.commit()
def commit_azure_virtual_machine(
name,
label,
status,
dns,
public_ip,
private_ip,
cloud_service_name,
deployment_name,
experiment_id,
virtual_environment,
):
cs = db_adapter.find_first_object_by(AzureCloudService, name=cloud_service_name)
dm = db_adapter.find_first_object_by(AzureDeployment, name=deployment_name, cloud_service=cs)
vm = db_adapter.add_object_kwargs(
AzureVirtualMachine,
name=name,
label=label,
status=status,
dns=dns,
public_ip=public_ip,
private_ip=private_ip,
deployment=dm,
experiment_id=experiment_id,
virtual_environment=virtual_environment,
)
db_adapter.commit()
return vm
def update_azure_virtual_machine_status(cloud_service_name, deployment_name, virtual_machine_name, status):
cs = db_adapter.find_first_object_by(AzureCloudService, name=cloud_service_name)
dm = db_adapter.find_first_object_by(AzureDeployment, name=deployment_name, cloud_service=cs)
vm = db_adapter.find_first_object_by(AzureVirtualMachine, name=virtual_machine_name, deployment=dm)
vm.status = status
db_adapter.commit()
return vm
def commit_azure_endpoint(name, protocol, public_port, private_port, virtual_machine):
db_adapter.add_object_kwargs(AzureEndpoint,
name=name,
protocol=protocol,
public_port=public_port,
private_port=private_port,
virtual_machine=virtual_machine)
db_adapter.commit()
def commit_azure_cloud_service(name, label, location, status, experiment_id):
db_adapter.add_object_kwargs(AzureCloudService,
name=name,
label=label,
location=location,
status=status,
experiment_id=experiment_id)
db_adapter.commit()
def commit_azure_deployment(name, slot, status, cloud_service_name, experiment_id):
cs = db_adapter.find_first_object_by(AzureCloudService, name=cloud_service_name)
db_adapter.add_object_kwargs(AzureDeployment,
name=name,
slot=slot,
status=status,
cloud_service=cs,
experiment_id=experiment_id)
db_adapter.commit()
def commit_azure_storage_account(name, description, label, location, status, experiment_id):
db_adapter.add_object_kwargs(AzureStorageAccount,
name=name,
description=description,
label=label,
location=location,
status=status,
experiment_id=experiment_id)
db_adapter.commit()
def commit_azure_deployment(name, slot, status, cloud_service_name, experiment_id):
cs = db_adapter.find_first_object_by(AzureCloudService, name=cloud_service_name)
db_adapter.add_object_kwargs(AzureDeployment,
name=name,
slot=slot,
status=status,
cloud_service=cs,
experiment_id=experiment_id)
db_adapter.commit()
def commit_azure_storage_account(name, description, label, location, status,
experiment_id):
db_adapter.add_object_kwargs(AzureStorageAccount,
name=name,
description=description,
label=label,
location=location,
status=status,
experiment_id=experiment_id)
db_adapter.commit()
def commit_virtual_environment(provider, name, image, status, remote_provider, remote_paras, experiment_id):
ve = db_adapter.add_object_kwargs(VirtualEnvironment,
provider=provider,
name=name,
image=image,
status=status,
remote_provider=remote_provider,
remote_paras=remote_paras,
experiment_id=experiment_id)
db_adapter.commit()
return ve
def commit_virtual_environment(provider, name, image, status, remote_provider, remote_paras, experiment_id):
ve = db_adapter.add_object_kwargs(VirtualEnvironment,
provider=provider,
name=name,
image=image,
status=status,
remote_provider=remote_provider,
remote_paras=remote_paras,
experiment_id=experiment_id)
db_adapter.commit()
return ve
def delete_azure_virtual_machine(cloud_service_name, deployment_name,
virtual_machine_name):
cs = db_adapter.find_first_object_by(AzureCloudService,
name=cloud_service_name)
dm = db_adapter.find_first_object_by(AzureDeployment,
name=deployment_name,
cloud_service=cs)
db_adapter.delete_all_objects_by(AzureVirtualMachine,
name=virtual_machine_name,
deployment_id=dm.id)
db_adapter.commit()
def commit_azure_log(experiment_id, operation, status, note=None, code=None):
db_adapter.add_object_kwargs(
AzureLog, experiment_id=experiment_id, operation=operation, status=status, note=note, code=code
)
db_adapter.commit()
if status == ALStatus.FAIL:
update_experiment_status(experiment_id, EStatus.FAILED)
elif status == ALStatus.END:
need_status = EStatus.RUNNING
if operation == ALOperation.STOP_VIRTUAL_MACHINE:
need_status = EStatus.STOPPED
check_experiment_done(experiment_id, need_status)
def update_azure_virtual_machine_status(cloud_service_name, deployment_name,
virtual_machine_name, status):
cs = db_adapter.find_first_object_by(AzureCloudService,
name=cloud_service_name)
dm = db_adapter.find_first_object_by(AzureDeployment,
name=deployment_name,
cloud_service=cs)
vm = db_adapter.find_first_object_by(AzureVirtualMachine,
name=virtual_machine_name,
deployment=dm)
vm.status = status
db_adapter.commit()
return vm
def commit_azure_log(experiment_id, operation, status, note=None, code=None):
db_adapter.add_object_kwargs(AzureLog,
experiment_id=experiment_id,
operation=operation,
status=status,
note=note,
code=code)
db_adapter.commit()
if status == ALStatus.FAIL:
update_experiment_status(experiment_id, EStatus.FAILED)
elif status == ALStatus.END:
need_status = EStatus.RUNNING
if operation == ALOperation.STOP_VIRTUAL_MACHINE:
need_status = EStatus.STOPPED
check_experiment_done(experiment_id, need_status)
def commit_azure_virtual_machine(name, label, status, dns, public_ip, private_ip,
cloud_service_name, deployment_name, experiment_id, virtual_environment):
cs = db_adapter.find_first_object_by(AzureCloudService, name=cloud_service_name)
dm = db_adapter.find_first_object_by(AzureDeployment, name=deployment_name, cloud_service=cs)
vm = db_adapter.add_object_kwargs(AzureVirtualMachine,
name=name,
label=label,
status=status,
dns=dns,
public_ip=public_ip,
private_ip=private_ip,
deployment=dm,
experiment_id=experiment_id,
virtual_environment=virtual_environment)
db_adapter.commit()
return vm
def setup_db():
# initialize db tables
# make sure database and user correctly created in mysql
# in case upgrade the table structure, the origin table need be dropped firstly
Base.metadata.create_all(bind=engine)
# init db : add a super admin account
superadmin = db_adapter.find_first_object_by(AdminUserHackathonRel,
admin_email=ADMIN.DEFAULT_SUPER_ADMIN_EMAIL,
hackathon_id=ADMIN.SUPER_ADMIN_GROUP_ID)
if superadmin is None:
db_adapter.add_object_kwargs(AdminUserHackathonRel,
admin_email=ADMIN.DEFAULT_SUPER_ADMIN_EMAIL,
hackathon_id=ADMIN.SUPER_ADMIN_GROUP_ID,
state=AdminUserHackathonRelStates.Actived,
remarks='super admins',
create_time=datetime.utcnow())
db_adapter.commit()
def setup_db():
# initialize db tables
# make sure database and user correctly created in mysql
# in case upgrade the table structure, the origin table need be dropped firstly
Base.metadata.create_all(bind=engine)
# init db : add a super admin account
superadmin = db_adapter.find_first_object_by(
AdminUserHackathonRel,
admin_email=ADMIN.DEFAULT_SUPER_ADMIN_EMAIL,
hackathon_id=ADMIN.SUPER_ADMIN_GROUP_ID)
if superadmin is None:
db_adapter.add_object_kwargs(
AdminUserHackathonRel,
admin_email=ADMIN.DEFAULT_SUPER_ADMIN_EMAIL,
hackathon_id=ADMIN.SUPER_ADMIN_GROUP_ID,
state=AdminUserHackathonRelStates.Actived,
remarks='super admins',
create_time=datetime.utcnow())
db_adapter.commit()
def set_template_virtual_environment_count(experiment_id, count):
e = db_adapter.get_object(Experiment, experiment_id)
t = db_adapter.get_object(Template, e.template_id)
t.virtual_environment_count = count
db_adapter.commit()
def update_virtual_environment_remote_paras(virtual_machine, remote_paras):
ve = virtual_machine.virtual_environment
ve.remote_paras = remote_paras
db_adapter.commit()
def update_virtual_environment_status(virtual_machine, status):
ve = virtual_machine.virtual_environment
ve.status = status
db_adapter.commit()
def update_azure_virtual_machine_public_ip(virtual_machine, public_ip):
virtual_machine.public_ip = public_ip
db_adapter.commit()
def update_azure_virtual_machine_private_ip(virtual_machine, private_ip):
virtual_machine.private_ip = private_ip
db_adapter.commit()
def delete_azure_deployment(cloud_service_name, deployment_slot):
cs = db_adapter.find_first_object_by(AzureCloudService, name=cloud_service_name)
db_adapter.delete_all_objects_by(AzureDeployment,
slot=deployment_slot,
cloud_service_id=cs.id)
db_adapter.commit()
def delete_azure_storage_account(name):
db_adapter.delete_all_objects_by(AzureStorageAccount, name=name)
db_adapter.commit()
def delete_azure_deployment(cloud_service_name, deployment_slot):
cs = db_adapter.find_first_object_by(AzureCloudService, name=cloud_service_name)
db_adapter.delete_all_objects_by(AzureDeployment, slot=deployment_slot, cloud_service_id=cs.id)
db_adapter.commit()
def delete_azure_cloud_service(name):
db_adapter.delete_all_objects_by(AzureCloudService, name=name)
db_adapter.commit()
def update_experiment_status(experiment_id, status):
e = db_adapter.get_object(Experiment, experiment_id)
e.status = status
db_adapter.commit()
def update_virtual_environment_remote_paras(virtual_machine, remote_paras):
ve = virtual_machine.virtual_environment
ve.remote_paras = remote_paras
db_adapter.commit()
def update_virtual_environment_status(virtual_machine, status):
ve = virtual_machine.virtual_environment
ve.status = status
db_adapter.commit()
def update_azure_virtual_machine_public_ip(virtual_machine, public_ip):
virtual_machine.public_ip = public_ip
db_adapter.commit()
def update_experiment_status(experiment_id, status):
e = db_adapter.get_object(Experiment, experiment_id)
e.status = status
db_adapter.commit()
def set_template_virtual_environment_count(experiment_id, count):
e = db_adapter.get_object(Experiment, experiment_id)
t = db_adapter.get_object(Template, e.template_id)
t.virtual_environment_count = count
db_adapter.commit()
def delete_azure_storage_account(name):
db_adapter.delete_all_objects_by(AzureStorageAccount, name=name)
db_adapter.commit()
def update_azure_virtual_machine_private_ip(virtual_machine, private_ip):
virtual_machine.private_ip = private_ip
db_adapter.commit()
def delete_azure_cloud_service(name):
db_adapter.delete_all_objects_by(AzureCloudService, name=name)
db_adapter.commit()
