restic-compose-backup/restic_volume_backup/cli.py

115 lines
3.7 KiB
Python
Raw Normal View History

2019-04-15 16:31:08 +00:00
import argparse
import pprint
2019-11-15 13:23:56 +00:00
import logging
2019-04-15 16:31:08 +00:00
2019-11-15 15:47:40 +00:00
from restic_volume_backup import log
2019-04-15 15:25:42 +00:00
from restic_volume_backup.config import Config
from restic_volume_backup.containers import RunningContainers
2019-04-17 01:45:51 +00:00
from restic_volume_backup import backup_runner
from restic_volume_backup import restic
2019-04-13 17:04:54 +00:00
2019-11-15 13:23:56 +00:00
logger = logging.getLogger(__name__)
2019-04-13 17:04:54 +00:00
def main():
2019-11-15 13:23:56 +00:00
"""CLI entrypoint"""
2019-04-15 16:31:08 +00:00
args = parse_args()
config = Config()
2019-04-13 17:04:54 +00:00
containers = RunningContainers()
2019-04-15 16:31:08 +00:00
if args.action == 'status':
2019-04-18 03:01:02 +00:00
status(config, containers)
2019-04-16 22:08:24 +00:00
2019-04-18 03:01:02 +00:00
elif args.action == 'backup':
backup(config, containers)
2019-04-16 22:08:24 +00:00
2019-04-18 03:01:02 +00:00
elif args.action == 'start-backup-process':
start_backup_process(config, containers)
2019-04-13 17:04:54 +00:00
2019-04-17 02:38:15 +00:00
2019-04-18 03:01:02 +00:00
def status(config, containers):
2019-11-12 11:39:49 +00:00
"""Outputs the backup config for the compose setup"""
2019-11-15 13:23:56 +00:00
logger.info("Backup config for compose project '%s'", containers.this_container.project_name)
logger.info("Current service: %s", containers.this_container.name)
2019-11-15 15:47:40 +00:00
# logger.info("Backup process: %s", containers.backup_process_container.name
# if containers.backup_process_container else 'Not Running')
2019-11-15 13:23:56 +00:00
logger.info("Backup running: %s", containers.backup_process_running)
backup_containers = containers.containers_for_backup()
for container in backup_containers:
2019-11-29 00:32:09 +00:00
logger.info('service: %s', container.service_name)
2019-12-03 02:04:49 +00:00
if container.volume_backup_enabled:
for mount in container.filter_mounts():
2019-11-29 00:32:09 +00:00
logger.info(' - volume: %s', mount.source)
2019-12-03 00:47:58 +00:00
if container.database_backup_enabled:
instance = container.instance
2019-12-03 02:04:49 +00:00
ping = instance.ping()
logger.info(' - %s (is_ready=%s)', instance.container_type, ping == 0)
2019-12-02 21:53:00 +00:00
if len(backup_containers) == 0:
2019-11-15 13:23:56 +00:00
logger.info("No containers in the project has 'restic-volume-backup.enabled' label")
2019-04-18 03:01:02 +00:00
def backup(config, containers):
2019-11-15 13:23:56 +00:00
"""Request a backup to start"""
2019-04-18 03:01:02 +00:00
# Make sure we don't spawn multiple backup processes
if containers.backup_process_running:
2019-11-12 11:39:49 +00:00
raise ValueError("Backup process already running")
2019-04-18 03:01:02 +00:00
2019-11-15 13:23:56 +00:00
logger.info("Initializing repository")
2019-04-18 03:01:02 +00:00
# TODO: Errors when repo already exists
restic.init_repo(config.repository)
2019-11-15 13:23:56 +00:00
logger.info("Starting backup container..")
# Map all volumes from the backup container into the backup process container
2019-11-15 13:23:56 +00:00
volumes = containers.this_container.volumes
# Map volumes from other containers we are backing up
mounts = containers.generate_backup_mounts('/backup')
volumes.update(mounts)
pprint.pprint(volumes, indent=2)
2019-04-18 03:01:02 +00:00
backup_runner.run(
image=containers.this_container.image,
command='restic-volume-backup start-backup-process',
volumes=volumes,
2019-11-12 11:39:49 +00:00
environment=containers.this_container.environment,
labels={
"restic-volume-backup.backup_process": 'True',
"com.docker.compose.project": containers.this_container.project_name,
},
2019-04-18 03:01:02 +00:00
)
2019-04-13 23:35:14 +00:00
def start_backup_process(config, containers):
2019-11-15 15:47:40 +00:00
"""The actual backup process running inside the spawned container"""
2019-11-12 11:39:49 +00:00
if (not containers.backup_process_container
or containers.this_container == containers.backup_process_container is False):
2019-11-15 15:47:40 +00:00
logger.error(
"Cannot run backup process in this container. Use backup command instead. "
"This will spawn a new container with the necessary mounts."
)
return
status(config, containers)
2019-11-15 15:47:40 +00:00
logger.info("start-backup-process")
2019-11-15 13:23:56 +00:00
2019-11-29 00:32:09 +00:00
restic.backup_files(config.repository, source='/backup')
2019-11-12 11:39:49 +00:00
2019-04-15 16:31:08 +00:00
def parse_args():
parser = argparse.ArgumentParser(prog='restic_volume_backup')
parser.add_argument(
'action',
choices=['status', 'backup', 'start-backup-process'],
2019-04-15 16:31:08 +00:00
)
return parser.parse_args()
2019-04-13 17:04:54 +00:00
if __name__ == '__main__':
main()