craftbeerpi4-pione/core/controller/config_controller.py

import logging
import os

from cbpi_api.config import ConfigType

from core.database.model import ConfigModel
from utils import load_config


class ConfigController():
    '''
    The main actor controller
    '''
    model = ConfigModel

    def __init__(self, cbpi):
        self.cache = {}
        self.logger = logging.getLogger(__name__)
        self.cbpi = cbpi
        self.cbpi.register(self)

    async def init(self):
        this_directory = os.path.dirname(__file__)
        self.static = load_config(os.path.join(this_directory, '../../config/config.yaml'))
        items = await self.model.get_all()
        for key, value in items.items():
            self.cache[value.name] = value

    def get(self, name, default=None):
        self.logger.info("GET CONFIG VALUE %s (default %s)" % (name, default))
        if name in self.cache and self.cache[name].value is not None:
            return self.cache[name].value
        else:
            return default

    async def set(self, name, value):
        self.logger.debug("SET %s = %s" % (name, value))
        if name in self.cache:
            self.cache[name].value = value
            await self.model.update(**self.cache[name].__dict__)
            await self.cbpi.bus.fire(topic="config/%s/update" % name, name=name, value=value)

    async def add(self, name, value, type: ConfigType, description, options=None):
        await self.model.insert(name=name, value=value, type=type.value, description=description, options=options)
        await self.cbpi.bus.fire(topic="config/%s/add" % name, name=name, value=value)