import argparse import pprint import logging from restic_volume_backup import log from restic_volume_backup.config import Config from restic_volume_backup.containers import RunningContainers from restic_volume_backup import backup_runner from restic_volume_backup import restic logger = logging.getLogger(__name__) def main(): """CLI entrypoint""" args = parse_args() config = Config() containers = RunningContainers() if args.action == 'status': status(config, containers) elif args.action == 'backup': backup(config, containers) elif args.action == 'start-backup-process': start_backup_process(config, containers) def status(config, containers): """Outputs the backup config for the compose setup""" logger.info("Backup config for compose project '%s'", containers.this_container.project_name) logger.info("Current service: %s", containers.this_container.name) # logger.info("Backup process: %s", containers.backup_process_container.name # if containers.backup_process_container else 'Not Running') logger.info("Backup running: %s", containers.backup_process_running) backup_containers = containers.containers_for_backup() for container in backup_containers: logger.info('service: %s', container.service_name) if container.volume_backup_enabled: for mount in container.filter_mounts(): logger.info(' - volume: %s', mount.source) if container.database_backup_enabled: instance = container.instance print(instance.container_type) print(instance.ping()) if len(backup_containers) == 0: logger.info("No containers in the project has 'restic-volume-backup.enabled' label") def backup(config, containers): """Request a backup to start""" # Make sure we don't spawn multiple backup processes if containers.backup_process_running: raise ValueError("Backup process already running") logger.info("Initializing repository") # TODO: Errors when repo already exists restic.init_repo(config.repository) logger.info("Starting backup container..") # Map all volumes from the backup container into the backup process container 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) backup_runner.run( image=containers.this_container.image, command='restic-volume-backup start-backup-process', volumes=volumes, environment=containers.this_container.environment, labels={ "restic-volume-backup.backup_process": 'True', "com.docker.compose.project": containers.this_container.project_name, }, ) def start_backup_process(config, containers): """The actual backup process running inside the spawned container""" if (not containers.backup_process_container or containers.this_container == containers.backup_process_container is False): 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) logger.info("start-backup-process") restic.backup_files(config.repository, source='/backup') def parse_args(): parser = argparse.ArgumentParser(prog='restic_volume_backup') parser.add_argument( 'action', choices=['status', 'backup', 'start-backup-process'], ) return parser.parse_args() if __name__ == '__main__': main()