mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-11-10 01:17:42 +01:00
81 lines
No EOL
2.6 KiB
Python
81 lines
No EOL
2.6 KiB
Python
import logging
|
|
import weakref
|
|
from collections import defaultdict
|
|
import json
|
|
import aiohttp
|
|
from aiohttp import web
|
|
from typing import Iterable, Callable
|
|
from cbpi_api import *
|
|
|
|
|
|
class CBPiWebSocket:
|
|
def __init__(self, cbpi) -> None:
|
|
self.cbpi = cbpi
|
|
self._callbacks = defaultdict(set)
|
|
self._clients = weakref.WeakSet()
|
|
self.logger = logging.getLogger(__name__)
|
|
self.cbpi.app.add_routes([web.get('/ws', self.websocket_handler)])
|
|
self.cbpi.bus.register_object(self)
|
|
|
|
@on_event(topic="#")
|
|
async def listen(self, topic, **kwargs):
|
|
from core.utils.encoder import ComplexEncoder
|
|
data = json.dumps(dict(topic=topic, data=dict(**kwargs)),skipkeys=True, check_circular=True, cls=ComplexEncoder)
|
|
self.logger.info("PUSH %s " % data)
|
|
self.send(data)
|
|
|
|
def send(self, data):
|
|
|
|
for ws in self._clients:
|
|
async def send_data(ws, data):
|
|
await ws.send_str(data)
|
|
self.cbpi.app.loop.create_task(send_data(ws, data))
|
|
|
|
def add_callback(self, func: Callable, event: str) -> None:
|
|
self._callbacks[event].add(func)
|
|
|
|
def register_object(self, obj):
|
|
for method in [getattr(obj, f) for f in dir(obj) if callable(getattr(obj, f)) and hasattr(getattr(obj, f), "ws")]:
|
|
self.add_callback(method, method.__getattribute__("key"))
|
|
|
|
async def emit(self, event: str, *args, **kwargs) -> None:
|
|
for func in self._event_funcs(event):
|
|
await func(*args, **kwargs)
|
|
|
|
def _event_funcs(self, event: str) -> Iterable[Callable]:
|
|
for func in self._callbacks[event]:
|
|
yield func
|
|
|
|
async def websocket_handler(self, request):
|
|
ws = web.WebSocketResponse()
|
|
await ws.prepare(request)
|
|
|
|
self._clients.add(ws)
|
|
|
|
c = len(self._clients) - 1
|
|
|
|
self.logger.info(ws)
|
|
self.logger.info(c)
|
|
try:
|
|
async for msg in ws:
|
|
if msg.type == aiohttp.WSMsgType.TEXT:
|
|
if msg.data == 'close':
|
|
|
|
await ws.close()
|
|
self.logger.info("WS Close")
|
|
else:
|
|
msg_obj = msg.json()
|
|
|
|
await self.cbpi.bus.fire(msg_obj["topic"], id=1, power=22)
|
|
# await self.fire(msg_obj["key"], ws, msg)
|
|
|
|
# await ws.send_str(msg.data)
|
|
elif msg.type == aiohttp.WSMsgType.ERROR:
|
|
self.logger.error('ws connection closed with exception %s' % ws.exception())
|
|
|
|
finally:
|
|
self._clients.discard(ws)
|
|
|
|
self.logger.info("Web Socket Close")
|
|
|
|
return ws |