def admin_backup():
""" Load the backup management page """
if not utils_general.user_has_permission('edit_settings'):
return redirect(url_for('routes_general.home'))
form_backup = forms_misc.Backup()
backup_dirs_tmp = []
if not os.path.isdir('/var/Mycodo-backups'):
flash("Error: Backup directory doesn't exist.", "error")
else:
backup_dirs_tmp = sorted(next(os.walk(BACKUP_PATH))[1])
backup_dirs_tmp.reverse()
backup_dirs = []
full_paths = []
for each_dir in backup_dirs_tmp:
if each_dir.startswith("Mycodo-backup-"):
full_path = os.path.join(BACKUP_PATH, each_dir)
backup_dirs.append((each_dir, get_directory_size(full_path) / 1000000.0))
full_paths.append(full_path)
if request.method == 'POST':
if form_backup.backup.data:
backup_size, free_before, free_after = can_perform_backup()
if free_after / 1000000 > 50:
cmd = "{pth}/mycodo/scripts/mycodo_wrapper backup-create" \
" | ts '[%Y-%m-%d %H:%M:%S]'" \
" >> {log} 2>&1".format(pth=INSTALL_DIRECTORY,
log=BACKUP_LOG_FILE)
subprocess.Popen(cmd, shell=True)
flash(gettext("Backup in progress"), "success")
else:
flash(
"Not enough free space to perform a backup. A backup "
"requires {size_bu:.1f} MB but there is only "
"{size_free:.1f} MB available, which would leave "
"{size_after:.1f} MB after the backup. If the free space "
"after a backup is less than 50 MB, the backup cannot "
"proceed. Free up space by deleting current "
"backups.".format(size_bu=backup_size / 1000000,
size_free=free_before / 1000000,
size_after=free_after / 1000000),
'error')
elif form_backup.download.data:
def get_all_file_paths(directory):
file_paths = []
for root, directories, files in os.walk(directory):
for filename in files:
# join the two strings in order to form the full filepath
filepath = os.path.join(root, filename)
file_paths.append(filepath)
return file_paths
try:
backup_date_version = form_backup.selected_dir.data
download_dir = os.path.join(BACKUP_PATH, 'Mycodo-backup-{}'.format(backup_date_version))
save_file = "Mycodo_Backup_{dv}_{host}_.zip".format(
dv=backup_date_version, host=socket.gethostname().replace(' ', ''))
file_paths = get_all_file_paths(download_dir)
string_remove = "{}".format(os.path.join(BACKUP_PATH, download_dir))
if not os.path.isdir(download_dir):
flash("Directory not found: {}".format(download_dir), "error")
else:
# Zip all files in the_backup directory
data = io.BytesIO()
with zipfile.ZipFile(data, 'w') as zip:
# writing each file one by one
for file in file_paths:
# Remove first two directory names from zip file, so the Mycodo root is the zip root
zip.write(file, file.replace(string_remove, ""))
data.seek(0)
# Send zip file to user
return send_file(
data,
mimetype='application/zip',
as_attachment=True,
attachment_filename=save_file
)
except Exception as err:
flash("Error: {}".format(err), "error")
elif form_backup.delete.data:
cmd = "{pth}/mycodo/scripts/mycodo_wrapper backup-delete {dir}" \
" 2>&1".format(pth=INSTALL_DIRECTORY,
dir=form_backup.selected_dir.data)
subprocess.Popen(cmd, shell=True)
flash(gettext("Deletion of backup in progress"),
"success")
elif form_backup.restore.data:
cmd = "{pth}/mycodo/scripts/mycodo_wrapper backup-restore {backup}" \
" | ts '[%Y-%m-%d %H:%M:%S]'" \
" >> {log} 2>&1".format(pth=INSTALL_DIRECTORY,
backup=form_backup.full_path.data,
log=RESTORE_LOG_FILE)
subprocess.Popen(cmd, shell=True)
flash(gettext("Restore in progress"),
"success")
return render_template('admin/backup.html',
form_backup=form_backup,
backup_dirs=backup_dirs,
full_paths=full_paths)
def admin_backup():
""" Load the backup management page """
if not utils_general.user_has_permission('edit_settings'):
return redirect(url_for('routes_general.home'))
form_backup = forms_misc.Backup()
backup_dirs_tmp = []
if not os.path.isdir('/var/Mycodo-backups'):
flash("Error: Backup directory doesn't exist.", "error")
else:
backup_dirs_tmp = sorted(next(os.walk(BACKUP_PATH))[1])
backup_dirs_tmp.reverse()
backup_dirs = []
full_paths = []
for each_dir in backup_dirs_tmp:
if each_dir.startswith("Mycodo-backup-"):
full_path = os.path.join(BACKUP_PATH, each_dir)
backup_dirs.append(
(each_dir, get_directory_size(full_path) / 1000000.0))
full_paths.append(full_path)
if request.method == 'POST':
if form_backup.backup.data:
backup_size, free_before, free_after = can_perform_backup()
if free_after / 1000000 > 50:
cmd = "{pth}/mycodo/scripts/mycodo_wrapper backup-create" \
" | ts '[%Y-%m-%d %H:%M:%S]'" \
" >> {log} 2>&1".format(pth=INSTALL_DIRECTORY,
log=BACKUP_LOG_FILE)
subprocess.Popen(cmd, shell=True)
flash(gettext("Backup in progress"), "success")
else:
flash(
"Not enough free space to perform a backup. A backup "
"requires {size_bu:.1f} MB but there is only "
"{size_free:.1f} MB available, which would leave "
"{size_after:.1f} MB after the backup. If the free space "
"after a backup is less than 50 MB, the backup cannot "
"proceed. Free up space by deleting current "
"backups.".format(size_bu=backup_size / 1000000,
size_free=free_before / 1000000,
size_after=free_after / 1000000),
'error')
elif form_backup.delete.data:
cmd = "{pth}/mycodo/scripts/mycodo_wrapper backup-delete {dir}" \
" 2>&1".format(pth=INSTALL_DIRECTORY,
dir=form_backup.selected_dir.data)
subprocess.Popen(cmd, shell=True)
flash(gettext("Deletion of backup in progress"), "success")
elif form_backup.restore.data:
cmd = "{pth}/mycodo/scripts/mycodo_wrapper backup-restore {backup}" \
" | ts '[%Y-%m-%d %H:%M:%S]'" \
" >> {log} 2>&1".format(pth=INSTALL_DIRECTORY,
backup=form_backup.full_path.data,
log=RESTORE_LOG_FILE)
subprocess.Popen(cmd, shell=True)
flash(gettext("Restore in progress"), "success")
return render_template('admin/backup.html',
form_backup=form_backup,
backup_dirs=backup_dirs,
full_paths=full_paths)
def admin_backup():
""" Load the backup management page """
if not utils_general.user_has_permission('edit_settings'):
return redirect(url_for('routes_general.home'))
form_backup = forms_misc.Backup()
backup_dirs_tmp = []
if not os.path.isdir('/var/Mycodo-backups'):
flash("Error: Backup directory doesn't exist.", "error")
else:
backup_dirs_tmp = sorted(next(os.walk(BACKUP_PATH))[1])
backup_dirs_tmp.reverse()
backup_dirs = []
full_paths = []
for each_dir in backup_dirs_tmp:
if each_dir.startswith("Mycodo-backup-"):
full_path = os.path.join(BACKUP_PATH, each_dir)
backup_dirs.append((each_dir, get_directory_size(full_path) / 1000000.0))
full_paths.append(full_path)
if request.method == 'POST':
if form_backup.backup.data:
backup_size, free_before, free_after = can_perform_backup()
if free_after / 1000000 > 50:
cmd = "{pth}/mycodo/scripts/mycodo_wrapper backup-create" \
" | ts '[%Y-%m-%d %H:%M:%S]'" \
" >> {log} 2>&1".format(pth=INSTALL_DIRECTORY,
log=BACKUP_LOG_FILE)
subprocess.Popen(cmd, shell=True)
flash(gettext("Backup in progress"), "success")
else:
flash(
"Not enough free space to perform a backup. A backup "
"requires {size_bu:.1f} MB but there is only "
"{size_free:.1f} MB available, which would leave "
"{size_after:.1f} MB after the backup. If the free space "
"after a backup is less than 50 MB, the backup cannot "
"proceed. Free up space by deleting current "
"backups.".format(size_bu=backup_size / 1000000,
size_free=free_before / 1000000,
size_after=free_after / 1000000),
'error')
elif form_backup.delete.data:
cmd = "{pth}/mycodo/scripts/mycodo_wrapper backup-delete {dir}" \
" 2>&1".format(pth=INSTALL_DIRECTORY,
dir=form_backup.selected_dir.data)
subprocess.Popen(cmd, shell=True)
flash(gettext("Deletion of backup in progress"),
"success")
elif form_backup.restore.data:
cmd = "{pth}/mycodo/scripts/mycodo_wrapper backup-restore {backup}" \
" | ts '[%Y-%m-%d %H:%M:%S]'" \
" >> {log} 2>&1".format(pth=INSTALL_DIRECTORY,
backup=form_backup.full_path.data,
log=RESTORE_LOG_FILE)
subprocess.Popen(cmd, shell=True)
flash(gettext("Restore in progress"),
"success")
return render_template('admin/backup.html',
form_backup=form_backup,
backup_dirs=backup_dirs,
full_paths=full_paths)
