restic-compose-backup/restic_volume_backup/containers.py

import os
import docker
import json
import pprint

from restic_volume_backup import utils

VOLUME_TYPE_BIND = "bind"
VOLUME_TYPE_VOLUME = "volume"


class Container:

    def __init__(self, data):
        self.id = data.get('Id')
        self.state = data.get('State')
        self.labels = data.get('Labels', {})
        self.names = data.get('Names', [])
        self.mounts = [Mount(mnt, container=self) for mnt in data.get('Mounts')]

        self.include = self.labels.get('restic-volume-backup.enabled', '').split(',')
        self.exlude = self.labels.get('restic-volume-backup.exclude', '').split(',')

    @property
    def backup_enabled(self):
        return self.labels.get('restic-volume-backup.enabled') == 'True'

    @property
    def is_running(self):
        return self.state == 'running'

    @property
    def service_name(self):
        return self.labels['com.docker.compose.service']

    @property
    def project_name(self):
        return self.labels['com.docker.compose.project']

    @property
    def is_oneoff(self):
        return self.labels['com.docker.compose.oneoff'] == 'True'

    def filter_mounts(self):
        """Get all mounts for this container matching filters"""
        for mount in self.mounts:
            if self.include:
                for pattern in self.include:
                    if pattern in mount.source:
                        yield mount
                        continue
            elif self.exlude:
                for pattern in self.exlude:
                    if pattern in mount.source:
                        continue
                    yield mount
            else:
                yield mount

    def to_dict(self):
        return {
            'Id': self.id,
            'Names': self.names,
            'State': self.state,
            'Labels': self.labels,
            'Mounts': [mnt.data for mnt in self.mounts],
            'include': self.include,
            'exclude': self.exlude,
        }


class Mount:
    """Mount wrapper"""
    def __init__(self, data, container=None):
        self._data = data
        self._container = container

    @property
    def container(self):
        return self._container

    @property
    def type(self):
        return self._data.get('Type')

    @property
    def name(self):
        return self._data.get('Name')

    @property
    def source(self):
        return self._data.get('Source')

    @property
    def destination(self):
        return self._data.get('Destination')

    def mount_string(self):
        if self.type == VOLUME_TYPE_VOLUME:
            return "- {}:{}:ro".format(self.name.split('_')[-1], self.destination)
        elif self.type == VOLUME_TYPE_BIND:
            return "- {}:{}:ro".format(self.source, self.destination)
        else:
            raise ValueError("Uknown volume type: {}".format(self.type))

    def __repr__(self):
        return str(self)

    def __str__(self):
        return str(self._data)

    def __hash__(self):
        """Uniquness for a volume"""
        if self.type == VOLUME_TYPE_VOLUME:
            return hash(self.name)
        elif self.type == VOLUME_TYPE_BIND:
            return hash(self.source)
        else:
            raise ValueError("Uknown volume type: {}".format(self.type))


class RunningContainers:

    def __init__(self):
        all_containers = utils.list_containers()
        self.containers = []
        self.this_container = None

        # Find the container we are running in.
        # If we don't have this information we cannot continue
        for container_data in all_containers:
            if container_data.get('Id').startswith(os.environ['HOSTNAME']):
                self.this_container = Container(container_data)

        if not self.this_container:
            raise ValueError("Cannot find metadata for backup container")

        # Gather all containers in the current compose setup
        for container_data in all_containers:
            # pprint.pprint(container_data, indent=2)
            container = Container(container_data)
            if container.project_name == self.this_container.project_name:
                if container.id != self.this_container.id:
                    self.containers.append(container)

        # for container in self.all_containers:
        #     # Weed out containers not beloging to this project.
        #     if container.project_name != self.backup_container.project_name:
        #         continue

        #     # Keep only containers with backup enabled
        #     if not container.backup_enabled:
        #         continue

        #     # and not oneoffs (started manually with run or similar)
        #     if container.is_oneoff:
        #         continue

        #     self.containers.append(container)

    # def gen_volumes(self, volume_type):
    #     """Generator yielding volumes of a specific type"""
    #     for cont in self.containers:
    #         for mnt in cont.mounts:
    #             if mnt.type == volume_type:
    #                 yield mnt

    # def volume_mounts(self):
    #     """Docker volumes"""
    #     return set(mnt for mnt in self.gen_volumes(VOLUME_TYPE_VOLUME))

    # def bind_mounts(self):
    #     """Host mapped volumes"""
    #     return set(mnt for mnt in self.gen_volumes(VOLUME_TYPE_BIND))

    def get_service(self, name):
        for container in self.containers:
            if container.service_name == name:
                return container

        return None

    def print_services(self):
        print()
        print("Backup config for compose project '{}'".format(self.this_container.project_name))
        print()

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

        print()