Разработана конечная точка для прослушивания сценариев

Этот коммит содержится в:
Глеб Иваницкий 2024-08-26 17:20:13 +03:00
родитель fb131822cc
Коммит b9f96398e3

Просмотреть файл

@ -34,7 +34,7 @@ class ConnectionManager:
async def broadcast( async def broadcast(
self, self,
data: dict, data: any,
): ):
for connection in self.connections: for connection in self.connections:
asyncio.ensure_future(connection.send_json(data)) asyncio.ensure_future(connection.send_json(data))
@ -47,53 +47,34 @@ api = FastAPI(
title=config.Main.app_name, title=config.Main.app_name,
) )
polls = [ scripts = [
{ {
'id': 1, 'id': 1,
'name': 'Текущее состояние сотрудников', 'name': 'Текущее состояние сотрудников',
'daysOfWeek': 'ПН-ПТ', 'time': '0 11 * * 1-5',
'time': '11:00', 'messageNumber': '1 вопрос',
'questionNumber': '1 вопрос',
'isEnabled': True, 'isEnabled': True,
}, },
{ {
'id': 2, 'id': 2,
'name': 'Планы на обед', 'name': 'Планы на обед',
'daysOfWeek': 'ПН-ПТ', 'time': '45 11 * * 1-5',
'time': '11:45-12:00', 'messageNumber': '2 вопроса',
'questionNumber': '2 вопроса',
'isEnabled': False, 'isEnabled': False,
}, },
] ]
@api.get( @api.websocket(
path='/api/polls', path='/ws/scripts',
)
async def _():
return polls
@api.put(
path='/api/polls',
) )
async def _( async def _(
poll: dict, websocket: WebSocket,
): ):
for i, p in enumerate(polls): await connection_manager.connect(websocket)
if p['id'] == poll['id']: try:
polls[i] = poll await connection_manager.broadcast(scripts)
while True:
await connection_manager.broadcast(await websocket.receive_json())
# @app.websocket( except WebSocketDisconnect:
# path='/ws/sync', connection_manager.disconnect(websocket)
# )
# async def _(
# websocket: WebSocket,
# ):
# await connection_manager.connect(websocket)
# try:
# while True:
# await connection_manager.broadcast(await websocket.receive_json())
# except WebSocketDisconnect:
# connection_manager.disconnect(websocket)