Routes was moved to self class

This commit is contained in:
2023-04-19 11:32:43 +02:00
parent e9ed4206db
commit 6ef15c5e9d
3 changed files with 92 additions and 64 deletions

80
routes.py Normal file
View File

@@ -0,0 +1,80 @@
import websockets
import json
from storage import Storage
class Routes():
__connections = set()
__storages = {}
def __init__(self):
self.__commands = {
"add": self._add_entity,
"get": self._get_entity,
"mod": self._mod_entity,
"del": self._del_entity,
"describe": self._describe
}
def __storage(self, name):
if not name in self.__storages:
self.__storages[name] = Storage(name)
return self.__storages[name]
def __get_entity(self, event):
return event.get("entity", None)
def __get_data(self, event):
return event.get("data", {})
def __get_event(self, name, event, data):
return { "entity": name, "event": event, "data": data }
async def _add_entity(self, websocket, event):
name = self.__get_entity(event)
data = self.__get_data(event)
if name and data:
self.__storage(name).create(data)
await broadcast(websocket, self.__get_event(name, "created", data))
async def _get_entity(self, websocket, event):
name = self.__get_entity(event)
ts = self.__get_data(event).get("ts", 0)
data = self.__storage(name).read(ts)
await websocket.send(json.dumps(self.__get_event(name, "received", data )))
async def _mod_entity(self, websocket, event):
name = self.__get_entity(event)
data = self.__get_data(event)
if name and data:
self.__storage(name).update(data)
await broadcast(websocket, self.__get_event(name, "modified", data))
async def _del_entity(self, websocket, event):
name = self.__get_entity(event)
entity_id = self.__get_data(event).get("id", None)
if self.__storage(name).delete(entity_id):
await broadcast(websocket, self.__get_event(name, "deleted", entity_id ))
async def _describe(self, websocket, event):
connections = list(self.__connections)
data = { c.user["id"]: connections.count(c) for c in connections }
await websocket.send(json.dumps(self.__get_event("connections", "described", data)))
async def _broadcast(self, _, event):
websockets.broadcast(self.__connections, json.dumps(event))
async def add_connection(self, connection):
self.__connections.add(connection)
await self._broadcast(connection, self.__get_event("users", "connected", connection.user))
async def remove_connection(self, connection):
self.__connections.remove(connection)
await self._broadcast(connection, self.__get_event("users", "disconnected", connection.user))
async def call(self, command, websocket, event):
handler = self.__commands.get(command, self._broadcast)
await handler(websocket, event)
def users(self):
return self.__storage("users").read()