59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
import os
|
|
import frappe
|
|
from frappe.utils import get_site_path, cint
|
|
from frappe.utils.data import convert_utc_to_user_timezone
|
|
import datetime
|
|
|
|
def get_context(context):
|
|
def get_time(path):
|
|
dt = os.path.getmtime(path)
|
|
return convert_utc_to_user_timezone(datetime.datetime.utcfromtimestamp(dt)).strftime('%Y-%m-%d %H:%M')
|
|
|
|
def get_size(path):
|
|
size = os.path.getsize(path)
|
|
if size > 1048576:
|
|
return "{0:.1f}M".format(float(size) / 1048576)
|
|
else:
|
|
return "{0:.1f}K".format(float(size) / 1024)
|
|
|
|
path = get_site_path('private', 'backups')
|
|
files = [x for x in os.listdir(path) if os.path.isfile(os.path.join(path, x))]
|
|
backup_limit = get_scheduled_backup_limit()
|
|
|
|
if len(files) > backup_limit:
|
|
cleanup_old_backups(path, files, backup_limit)
|
|
|
|
files = [('/backups/' + _file,
|
|
get_time(os.path.join(path, _file)),
|
|
get_size(os.path.join(path, _file))) for _file in files if _file.endswith('sql.gz')]
|
|
files.sort(key=lambda x: x[1], reverse=True)
|
|
|
|
return {"files": files}
|
|
|
|
def get_scheduled_backup_limit():
|
|
backup_limit = frappe.db.get_singles_value('System Settings', 'backup_limit')
|
|
return cint(backup_limit)
|
|
|
|
def cleanup_old_backups(site_path, files, limit):
|
|
backup_paths = []
|
|
for f in files:
|
|
if f.endswith('sql.gz'):
|
|
_path = os.path.abspath(os.path.join(site_path, f))
|
|
backup_paths.append(_path)
|
|
|
|
backup_paths = sorted(backup_paths, key=os.path.getctime)
|
|
files_to_delete = len(backup_paths) - limit
|
|
|
|
for idx in range(0, files_to_delete):
|
|
f = os.path.basename(backup_paths[idx])
|
|
files.remove(f)
|
|
|
|
os.remove(backup_paths[idx])
|
|
|
|
def delete_downloadable_backups():
|
|
path = get_site_path('private', 'backups')
|
|
files = [x for x in os.listdir(path) if os.path.isfile(os.path.join(path, x))]
|
|
backup_limit = get_scheduled_backup_limit()
|
|
|
|
if len(files) > backup_limit:
|
|
cleanup_old_backups(path, files, backup_limit)
|