mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-11-14 02:58:16 +01:00
126 lines
3.4 KiB
Python
126 lines
3.4 KiB
Python
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()
|