coop-cloud/backup-bot-two
coop-cloud/backup-bot-two
6
0
Fork 4
Code Issues 9 Pull Requests Packages Projects Releases Wiki Activity

add restore-path argument for undeployed apps closes #59

Browse Source
This commit is contained in:
Moritz 2024-10-28 17:01:14 +01:00
parent 1214f59c79
commit 4b4371ed3f
1 changed files with 39 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
backupbot.py
View File

@ -116,7 +116,10 @@ def create(retries):
def restore(snapshot_id, target, noninteractive, volumes, container, no_commands): def restore(snapshot_id, target, noninteractive, volumes, container, no_commands):
app_settings = parse_backup_labels('restore', container) app_settings = parse_backup_labels('restore', container)
if SERVICE != 'ALL': if SERVICE != 'ALL':
app_settings = {SERVICE: app_settings[SERVICE]} if not app_settings.get(SERVICE):
logger.error(f"The app {SERVICE} is not running, use the restore-path argument to restore paths of undeployed apps")
exit(1)
app_settings = {SERVICE: app_settings.get(SERVICE)}
pre_commands, post_commands, backup_paths, apps_versions = get_backup_details(app_settings, volumes) pre_commands, post_commands, backup_paths, apps_versions = get_backup_details(app_settings, volumes)
snapshots = get_snapshots(snapshot_id) snapshots = get_snapshots(snapshot_id)
if not snapshots: if not snapshots:
@ -159,6 +162,41 @@ def restore(snapshot_id, target, noninteractive, volumes, container, no_commands
logger.debug(result) logger.debug(result)
@cli.command()
@click.option('snapshot_id', '--snapshot', '-s', envvar='SNAPSHOT', default='latest')
@click.option('target', '--target', '-t', envvar='TARGET', default='/')
@click.option('noninteractive', '--noninteractive', envvar='NONINTERACTIVE', is_flag=True)
@click.argument('paths', nargs=-1, required=True, envvar='INCLUDE_PATH')
def restore_path(snapshot_id, target, noninteractive, paths):
""" PATHS: list of paths to restore"""
snapshots = get_snapshots(snapshot_id)
if not snapshots:
logger.error(f"No Snapshots with ID {snapshot_id} for app {SERVICE} found.")
exit(1)
snapshot = snapshots[0]
snapshot_id = snapshot['short_id']
if not noninteractive:
print(f"Snapshot to restore: \t{snapshot_id}")
restore_app_versions = app_versions_from_tags(snapshot.get('tags'))
print("Apps:")
for app, version in restore_app_versions.items():
if SERVICE == 'ALL' or SERVICE == app:
print(f"\t{app} \t {version}")
print("The following paths will be restored:")
for p in paths:
print(f'\t{p}')
snapshot_date = datetime.fromisoformat(snapshot['time'])
delta = datetime.now(tz=timezone.utc) - snapshot_date
print(f"This snapshot is {delta} old")
print("\nTHIS COMMAND WILL IRREVERSIBLY OVERWRITES FILES")
prompt = input("Type YES (uppercase) to continue: ")
if prompt != 'YES':
logger.error("Restore aborted")
exit(1)
print(f"Restoring Snapshot {snapshot_id} at {target}")
result = restic_restore(snapshot_id=snapshot_id, include=paths, target_dir=target)
logger.debug(result)
def restic_restore(snapshot_id, include=[], target_dir=None): def restic_restore(snapshot_id, include=[], target_dir=None):
cmd = restic.cat.base_command() + ['restore', snapshot_id] cmd = restic.cat.base_command() + ['restore', snapshot_id]
for path in include: for path in include: