from aiohttp import web from core.api.actor import CBPiActor from core.api.decorator import on_event, request_mapping from core.controller.crud_controller import CRUDController from core.database.model import ActorModel from core.http_endpoints.http_api import HttpAPI from core.utils import parse_props class ActorHttp(HttpAPI): @request_mapping(path="/{id}/on", auth_required=False) async def http_on(self, request) -> web.Response: """ :param request: :return: """ id = int(request.match_info['id']) self.cbpi.bus.fire(topic="actor/%s/on" % id, id=id, power=99) return web.Response(status=204) @request_mapping(path="/{id}/off", auth_required=False) async def http_off(self, request) -> web.Response: """ :param request: :return: """ id = int(request.match_info['id']) self.cbpi.bus.fire(topic="actor/%s/off" % id, id=id) return web.Response(status=204) @request_mapping(path="/{id}/toggle", auth_required=False) async def http_toggle(self, request) -> web.Response: """ :param request: :return: """ id = int(request.match_info['id']) print("ID", id) self.cbpi.bus.fire(topic="actor/%s/toggle" % id, id=id) return web.Response(status=204) class ActorController(ActorHttp, CRUDController): ''' The main actor controller ''' model = ActorModel def __init__(self, cbpi): super(ActorController, self).__init__(cbpi) self.cbpi = cbpi self.state = False; self.cbpi.register(self, "/actor") self.types = {} self.actors = {} def register(self, name, clazz) -> None: ''' Register a new actor type :param name: actor name :param clazz: actor class :return: None ''' print("REGISTER", name) if issubclass(clazz, CBPiActor): print("ITS AN ACTOR") parse_props(clazz) self.types[name] = clazz async def init(self): ''' This method initializes all actors during startup. It creates actor instances :return: ''' await super(ActorController, self).init() print("INIT ACTOR") for name, clazz in self.types.items(): print("Type", name) for id, value in self.cache.items(): if value.type in self.types: clazz = self.types[value.type]["class"]; print(self.cache[id]) self.cache[id].instance = clazz(self.cbpi) @on_event(topic="actor/+/on") def on(self, id, power=100, **kwargs) -> None: print(id) id = int(id) if id in self.cache: print("POWER ON") actor = self.cache[id].instance actor.on(power) @on_event(topic="actor/+/toggle") def toggle(self, id, power=100, **kwargs) -> None: id = int(id) if id in self.cache: actor = self.cache[id].instance if actor.state is True: actor.off() else: actor.on() @on_event(topic="actor/+/off") def off(self, id, **kwargs) -> None: """ :param id: :param kwargs: """ id = int(id) if id in self.cache: actor = self.cache[id].instance actor.off()