140 lines
4.5 KiB
Python
140 lines
4.5 KiB
Python
import argparse
|
|
import pprint
|
|
import logging
|
|
|
|
from restic_compose_backup import (
|
|
backup_runner,
|
|
log,
|
|
restic,
|
|
)
|
|
from restic_compose_backup.config import Config
|
|
from restic_compose_backup.containers import RunningContainers
|
|
|
|
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
|
|
ping = instance.ping()
|
|
logger.info(' - %s (is_ready=%s)', instance.container_type, ping == 0)
|
|
|
|
if len(backup_containers) == 0:
|
|
logger.info("No containers in the project has 'restic-compose-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)
|
|
|
|
result = backup_runner.run(
|
|
image=containers.this_container.image,
|
|
command='restic-compose-backup start-backup-process',
|
|
volumes=volumes,
|
|
environment=containers.this_container.environment,
|
|
source_container_id=containers.this_container.id,
|
|
labels={
|
|
"restic-compose-backup.backup_process": 'True',
|
|
"com.docker.compose.project": containers.this_container.project_name,
|
|
},
|
|
)
|
|
logger.info('Backup container exit code: %s', result)
|
|
# TODO: Alert
|
|
|
|
|
|
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")
|
|
|
|
# Back up volumes
|
|
try:
|
|
vol_result = restic.backup_files(config.repository, source='/backup')
|
|
logger.info('Volume backup exit code: %s', vol_result)
|
|
# TODO: Alert
|
|
except Exception as ex:
|
|
logger.error(ex)
|
|
# TODO: Alert
|
|
|
|
# back up databases
|
|
for container in containers.containers_for_backup():
|
|
if container.database_backup_enabled:
|
|
try:
|
|
instance = container.instance
|
|
logger.info('Backing up %s in service %s', instance.container_type, instance.service_name)
|
|
result = instance.backup()
|
|
logger.info('Exit code: %s', result)
|
|
# TODO: Alert
|
|
except Exception as ex:
|
|
logger.error(ex)
|
|
# TODO: Alert
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(prog='restic_compose_backup')
|
|
parser.add_argument(
|
|
'action',
|
|
choices=['status', 'backup', 'start-backup-process'],
|
|
)
|
|
return parser.parse_args()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|