chore(db): optimize database backup to once per day and reduce max backups to 4

- Change default IIB_DB_FILE_BACKUP_MAX from 8 to 4
- Add daily backup check to prevent multiple backups per day
- Skip backup if one already exists for the current date
chore/optimize-db-backup
zanllp 2026-02-22 13:44:30 +08:00
parent 230b394f4f
commit 0aa0321ee2
2 changed files with 18 additions and 4 deletions

View File

@ -12,7 +12,7 @@ IIB_SECRET_KEY=
IIB_SERVER_LANG=auto
# This configuration parameter specifies the maximum number of database file backups for the IIB .
IIB_DB_FILE_BACKUP_MAX=8
IIB_DB_FILE_BACKUP_MAX=4
# Set the cache directory for IIB, including image cache and video cover cache.
# The default is the system's temporary directory, but if you want to specify a custom directory, set it here.

View File

@ -83,14 +83,28 @@ def backup_db_file(db_file_path):
if not os.path.exists(db_file_path):
return
max_backup_count = int(os.environ.get('IIB_DB_FILE_BACKUP_MAX', '8'))
max_backup_count = int(os.environ.get('IIB_DB_FILE_BACKUP_MAX', '4'))
if max_backup_count < 1:
return
backup_folder = os.path.join(cwd,'iib_db_backup')
os.makedirs(backup_folder, exist_ok=True)
# Check if backup already exists today
current_date = datetime.now().date()
backup_files = os.listdir(backup_folder) if os.path.exists(backup_folder) else []
pattern = r"iib\.db_(\d{4}-\d{2}-\d{2})"
for f in backup_files:
match = re.search(pattern, f)
if match:
backup_date = datetime.strptime(match.group(1), '%Y-%m-%d').date()
if backup_date == current_date:
print(f"\033[93mIIB Database backup already exists for today ({current_date}). Skipping backup.\033[0m")
return
current_time = datetime.now()
timestamp = current_time.strftime('%Y-%m-%d %H-%M-%S')
backup_filename = f"iib.db_{timestamp}"
os.makedirs(backup_folder, exist_ok=True)
backup_filepath = os.path.join(backup_folder, backup_filename)
shutil.copy2(db_file_path, backup_filepath)
backup_files = os.listdir(backup_folder)
@ -103,7 +117,7 @@ def backup_db_file(db_file_path):
for i in range(files_to_remove_count):
file_to_remove = os.path.join(backup_folder, sorted_backup_files[i][0])
os.remove(file_to_remove)
print(f"\033[92mIIB Database file has been successfully backed up to the backup folder.\033[0m")
def get_sd_webui_conf(**kwargs):