restic-compose-backup/restic_volume_backup/cli.py

import argparse
import sys

from restic_volume_backup.config import Config
from restic_volume_backup.containers import RunningContainers


def main():
    args = parse_args()
    config = Config()
    containers = RunningContainers()

    if args.action == 'status':
        print()
        print("Backup config for compose project '{}'".format(containers.this_container.project_name))
        print()

        for container in containers.containers:
            print('service: {}'.format(container.service_name))
            for mount in container.filter_mounts():
                print(' - {}'.format(mount.source))

        print()

    elif args.mode == 'backup':
        print("Starting backup ..")
        # TODO: Errors when repo already exists
        # restic.init_repo(Config.repository)

        # for vol in containers.backup_volumes():
        #     restic.backup_volume(Config.repository, vol)

    elif args.mode == 'snapshots':
        restic.snapshots(config.repository)


def parse_args():
    parser = argparse.ArgumentParser(prog='restic_volume_backup')
    parser.add_argument(
        'action',
        choices=['status', 'backup'],
    )
    return parser.parse_args()


if __name__ == '__main__':
    main()