105 lines
3.1 KiB
Python
105 lines
3.1 KiB
Python
import logging
|
|
import os
|
|
import pwd
|
|
import json
|
|
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)
|
|
self.find_config()
|
|
|
|
def find_config(self):
|
|
# who are we?
|
|
uid = os.getuid()
|
|
# name?
|
|
user = pwd.getpwuid(uid).pw_name
|
|
logging.debug("Trying to find the configuration")
|
|
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)
|
|
logging.debug("Found config: %s", json.dumps(self._config, indent=4))
|
|
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
|
|
|
|
async def get_backups_by_device(self, uuid):
|
|
name = None
|
|
if not self._config.get("devices"):
|
|
return 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.config.get("target_device") == name:
|
|
backups.append(backup)
|
|
return backups
|
|
|
|
def get_preferences(self):
|
|
if self._config.get("preferences", None):
|
|
return self._config.get("preferences")
|
|
return {}
|