def get_config_workload_id():
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_config_backup_id = ("select id from config_workloads;")
cursor.execute(get_config_backup_id)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_available_vms_of_workload(workload_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_count = ("select count(*) from workload_vms where workload_id='" +
str(workload_id) + "' and status <> 'deleted';")
cursor.execute(get_count)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_available_workload_types():
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_workload_types = (
"select count(*) from workload_types where deleted <> 1")
cursor.execute(get_workload_types)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
LOG.error(str(e))
finally:
cursor.close()
conn.close()
def get_available_workloads():
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_available_workloads = (
"select count(*) from workloads where status=\"available\"")
cursor.execute(get_available_workloads)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_snapshot_restore_id(snapshot_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_snapshot_restore_id = (
"select id from restores where snapshot_id='" + snapshot_id +
"' order by deleted_at asc limit 1")
cursor.execute(get_snapshot_restore_id)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_workload_status(workload_name):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_workload_snapshot = (
"select status from workloads where display_name='" +
workload_name + "' order by created_at desc limit 1")
cursor.execute(get_workload_snapshot)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_workload_display_description(workload_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_workload_display_description = (
"select display_description from workloads where id='" +
workload_id + "'")
cursor.execute(get_workload_display_description)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_inprogress_snapshot_id(workload_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_workload_vmid = (
"select id from snapshots where workload_id='" + workload_id +
"' and status<>'available' order by updated_at desc limit 1")
cursor.execute(get_workload_vmid)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_workload_vmid(workload_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_workload_vmid = (
"select vm_id from workload_vms where workload_id='" +
workload_id + "'")
cursor.execute(get_workload_vmid)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_workload_schedule(workload_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_workload_schedule = (
"select jobschedule from workloads where id='" + workload_id +
"' order by created_at desc limit 1")
cursor.execute(get_workload_schedule)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
LOG.error(str(e))
finally:
cursor.close()
conn.close()
def get_workload_count(workload_name):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_workload_count = (
"select count(*) from workloads where display_name='" +
workload_name + "' and status=\"available\"")
cursor.execute(get_workload_count)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_available_snapshots():
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_available_snapshots = (
"select count(*) from snapshots where deleted=0 and status = 'available' order by created_at desc limit 1"
)
cursor.execute(get_available_snapshots)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_deleted_workload(workload_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_deleted_workload = ("select status from workloads where id='" +
str(workload_id) +
"' order by updated_at desc limit 1")
cursor.execute(get_deleted_workload)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_workload_type_data(workload_type_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_workload_types = (
"select * from workload_types where deleted <> 1 and ID='" +
str(workload_type_id) + "'")
cursor.execute(get_workload_types)
rows = cursor.fetchall()
for row in rows:
return row
except Exception as e:
LOG.error(str(e))
finally:
cursor.close()
conn.close()
def get_available_snapshots_for_workload(snapshot_name, workload_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_available_snapshots = (
"select count(*) from snapshots where display_name='" +
snapshot_name + "' and workload_id='" + str(workload_id) +
"' and deleted=0 order by created_at desc limit 1")
cursor.execute(get_available_snapshots)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_vmids():
try:
vmlist = []
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_vmids = (
"select vm_id from workload_vms where status=\"available\"")
cursor.execute(get_vmids)
rows = cursor.fetchall()
for row in rows:
vmlist.append(str(row[0]))
return vmlist
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_snapshot_restore_delete_status(restore_name, restore_type):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_snapshot_restore_delete_status = (
"select deleted from restores where display_name='" +
restore_name + "' and restore_type='" + restore_type +
"' order by deleted_at desc limit 1")
cursor.execute(get_snapshot_restore_delete_status)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
def get_workload_snapshot_delete_status(snapshot_name, snapshot_type,
snapshot_id):
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_workload_snapshot_delete_status = (
"select deleted from snapshots where display_name='" +
snapshot_name + "' and snapshot_type='" + snapshot_type +
"' and id='" + str(snapshot_id) +
"' order by deleted_at desc limit 1")
cursor.execute(get_workload_snapshot_delete_status)
rows = cursor.fetchall()
for row in rows:
return row[0]
except Exception as e:
print(str(e))
finally:
cursor.close()
conn.close()
