def __init__(self, config_path, config_private_path):
with open(config_path, encoding = 'utf-8') as json_file:
config = json.load(json_file)
with open(config_private_path, encoding = 'utf-8') as json_file:
config_private = json.load(json_file)
self.config_path = config_path
self.week_index = config.get("week_index")
self.folder = config.get("folder")
self.filename = config.get("filename")
self.lesson_end_time = config.get("lesson_end_time")
self.greetings = config.get("greetings")
self.day_of_week = config.get("day_of_week")
self.lessons_schedule = config.get("lessons_schedule")
self.bot_token = config_private.get("bot_token")
self.db_url = config_private.get("db_url")
self.timetable = dict()
self.bot = aiogram.Bot(self.bot_token)
self.dp = aiogram.Dispatcher(self.bot)
self.xls = XlsHandler("../{}/{}".format(
self.folder,
self.filename
),
self.day_of_week
)
self.db = dbHandler(self.db_url)
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_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_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()
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_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_restores():
try:
conn = db_handler.dbHandler()
cursor = conn.cursor()
get_available_restores = ("select count(*) from restores where status=\"available\"")
cursor.execute(get_available_restores)
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_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_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_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_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_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_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_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()
app = Flask(__name__)
#app.permanent_session_lifetime = False
app.secret_key = "3a4ds3fdfad3s2fas5d"
mail_settings = {
"MAIL_SERVER": 'smtp.gmail.com',
"MAIL_PORT": 465,
"MAIL_USE_TLS": False,
"MAIL_USE_SSL": True,
"MAIL_USERNAME": '******',
"MAIL_PASSWORD": '******'
}
app.config.update(mail_settings)
mail = Mail(app)
serializer = URLSafeTimedSerializer("3a4ds3fdfad3s2fas5d")
handler = dbHandler()
@app.route('/')
def test():
return render_template('manage_plans.html', plans=handler.get_all_plans())
@app.route('/login_admin', methods=['POST', 'GET'])
def admin_login():
if request.method == 'POST':
user_name = request.form.get('user_name')
password = request.form.get('pswd')
auth = handler.authenticate_admin(user_name, password)
if auth:
session['admin'] = True
