pybackive/backive/config/config.py

import logging
import os
import pwd
from ruamel.yaml import YAML
import logging
import jsonschema

from backive.core.backup import Backup
from backive.core.device import Device


class Config:
    __shared_state = dict()
    _config = dict()

    def __init__(self):
        self.__dict__ = self.__shared_state
        if not self._config:
            logging.info("Loading configuration...")
            self._schema = dict()
            self._backups = list()
            self._devices = list()
            file_path = os.path.realpath(__file__)
            schema_path = os.path.join(
                    os.path.dirname(
                        file_path
                    ),
                    "schema.yml"
                )
            with open(schema_path, "r") as stream:
                self._schema = YAML().load(stream)

    def find_config(self):
        # who are we?
        uid = os.getuid()
        # name?
        user = pwd.getpwuid(uid).pw_name
        try:
            if uid == 0:
                config_file = "/etc/backive.yml"
            else:
                config_file = os.path.join(
                    os.path.expanduser("~"),
                    ".config",
                    "backive",
                    "backive.yml"
                )

            with open(config_file, "r") as cfg:
                self._config = YAML().load(cfg)
            jsonschema.validate(self._config, self._schema)
        except Exception as e:
            logging.error(e)

    def get_devices(self):
        if self._config.get("devices", None) and not self._devices:
            data = self._config.get("devices")
            for device in data:
                self._devices.append(
                    Device.instance(
                        device,
                        data.get(device)
                        )
                    )
        return self._devices

    def get_backups(self):
        if self._config.get("backups", None) and not self._backups:
            data = self._config.get("backups")
            for name in data:
                self._backups.append(
                    Backup.instance(
                        name,
                        data.get(name)
                        )
                    )
        return self._backups

    def get_backups_by_device(self, uuid):
        name = None
        for k, v in self._config.get("devices").items():
            if v.get("uuid") == uuid:
                name = k
        if name:
            return self.get_device_backups(name)
        return None

    def get_device_backups(self, name):
        backups = list()
        for backup in self.get_backups():
            if backup.target == name:
                backups.append(backup)
        return backups

    def get_preferences(self):
        if self._config.get("preferences", None):
            return self._config.get("preferences")
        return {}