restic-compose-backup/restic-backup/backup.py

import os
import sys
from containers import RunningContainers
import restic

cmds = ['volumes', 'backup', 'snapshots', 'check']


class Config:
    repository = os.environ['RESTIC_REPOSITORY']
    password = os.environ['RESTIC_PASSWORD']

    @classmethod
    def check(cls):
        if not cls.repository:
            raise ValueError("CONTAINER env var not set")

        if not cls.password:
            raise ValueError("PASSWORD env var not set")


def main():
    if len(sys.argv) < 2:
        raise ValueError("Missing argument: {}".format(cmds))

    mode = sys.argv[1]
    if mode not in cmds:
        raise ValueError("Valid arguments: {}".format(cmds))

    Config.check()

    containers = RunningContainers()

    if mode == 'volumes':
        volumes = containers.volume_mounts()
        for vol in volumes:
            # print(vol)
            print(vol.mount_string())

        binds = containers.bind_mounts()
        for vol in binds:
            print(vol.mount_string())

    if mode == 'backup':
        restic.init_repo(Config.repository)

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

    if mode == 'snapshots':
        restic.snapshots(Config.repository)

if __name__ == '__main__':
    main()