craftbeerpi4-pione/cbpi/controller/actor_controller.py

import asyncio
import logging

from voluptuous import Schema

from cbpi.api import *
from cbpi.controller.crud_controller import CRUDController
from cbpi.database.model import ActorModel


class ActorController(CRUDController):
    '''
    The main actor controller
    '''
    model = ActorModel

    def __init__(self, cbpi):
        super(ActorController, self).__init__(cbpi)
        self.cbpi = cbpi
        self.state = False;
        self.logger = logging.getLogger(__name__)
        self.cbpi.register(self)
        self.types = {}
        self.actors = {}

    async def init(self):
        """
        This method initializes all actors during startup. It creates actor instances
        
        :return: 
        """
        await super(ActorController, self).init()
        for id, value in self.cache.items():
            await self._init_actor(value)

    def get_state(self):
        return dict(items=self.cache,types=self.types)

    async def _init_actor(self, actor):

        try:
            if actor.type in self.types:
                cfg = actor.config.copy()
                cfg.update(dict(cbpi=self.cbpi, id=id, name=actor.name))
                clazz = self.types[actor.type]["class"];
                self.cache[actor.id].instance = clazz(**cfg)
                self.cache[actor.id].instance.init()
                await self.cbpi.bus.fire(topic="actor/%s/initialized" % actor.id, id=actor.id)
            else:
                self.logger.error("Actor type '%s' not found (Available Actor Types: %s)" % (actor.type, ', '.join(self.types.keys())))
        except Exception as e:

            self.logger.error("Failed to init actor %s - Reason %s" % (actor.id, str(e)))

    async def _stop_actor(self, actor):
        actor.instance.stop()
        await self.cbpi.bus.fire(topic="actor/%s/stopped" % actor.id, id=actor.id)

    @on_event(topic="actor/+/switch/on")
    async def on(self, actor_id, power=100, **kwargs) -> None:
        """
        Method to switch an actor on.
        Supporting Event Topic "actor/+/on"
        
        :param actor_id: the actor id
        :param future
        :param power: as integer value between 1 and 100
        :param kwargs: 
        :return: 
        """

        actor_id = int(actor_id)
        if actor_id in self.cache:

            self.logger.debug("ON %s" % actor_id)
            actor = self.cache[actor_id].instance
            actor.on(power)
            await self.cbpi.bus.fire("actor/%s/on/ok" % actor_id)


    @on_event(topic="actor/+/toggle")
    async def toggle(self, actor_id, power=100, time=None, **kwargs) -> None:
        """
        Method to toggle an actor on or off
        Supporting Event Topic "actor/+/toggle"

        :param actor_id: the actor actor_id
        :param power: the power as integer between 0 and 100
        :return:
        """



        self.logger.debug("TOGGLE %s" % actor_id)
        actor_id = int(actor_id)
        if actor_id in self.cache:
            actor = self.cache[actor_id].instance

            if actor.state is True:
                await self.off(actor_id=actor_id)
            else:
                await self.on(actor_id=actor_id)

            if time is not None:
                async def time_toggle(cbpi, actor_id, time):
                    await asyncio.sleep(time)
                    await cbpi.bus.fire("actor/%s/off" % actor_id, actor_id = actor_id)
                await self.cbpi.job.start_job(time_toggle(self.cbpi, actor_id, time), "actor_%s_time_toggle" % actor_id, "actor_toggle")


    @on_event(topic="actor/+/off")
    async def off(self, actor_id, **kwargs) -> None:
        """
        Method to switch and actor off
        Supporting Event Topic "actor/+/off"
        
        :param actor_id: the actor actor_id
        :param kwargs: 
        """
        self.logger.debug("OFF %s" % actor_id)
        actor_id = int(actor_id)

        if actor_id in self.cache:
            actor = self.cache[actor_id].instance
            actor.off()
            await self.cbpi.bus.fire("actor/%s/off/ok" % actor_id)

    @on_event(topic="actor/+/action")
    async def call_action(self, actor_id, data, **kwargs) -> None:

        schema = Schema({"name":str, "parameter":dict})
        schema(data)
        name = data.get("name")
        parameter = data.get("parameter")
        actor = self.cache[actor_id].instance.__getattribute__(name)(**parameter)

    async def _post_add_callback(self, m):
        '''
    
        :param m: 
        :return: 
        '''

        await self._init_actor(m)
        pass

    async def _pre_delete_callback(self, actor_id):
        if hasattr(self.cache[int(actor_id)], "instance")  and self.cache[int(actor_id)].instance is not None:
            await self._stop_actor(self.cache[int(actor_id)])

    async def _pre_update_callback(self, actor):
        if hasattr(actor, "instance") and actor.instance is not None:
            await self._stop_actor(actor)

    async def _post_update_callback(self, actor):

        await self._init_actor(actor)