This commit is contained in:
urec56 2024-03-02 15:37:09 +03:00
parent ca58c62787
commit 81d4ed8d8a

View file

@ -19,7 +19,6 @@ class ConnectionManager(WebSocket):
self.active_connections[chat_id].remove(websocket)
async def broadcast(self, user_id: int, chat_id: int, message: dict):
print(message)
new_message = await self.add_message_to_database(
user_id=user_id, chat_id=chat_id, message=message['message'], image_url=message['image_url']
)
@ -45,7 +44,7 @@ async def websocket_endpoint(chat_id: int, user_id: int, websocket: WebSocket):
try:
while True:
data = await websocket.receive_json()
print(data)
await manager.broadcast(user_id=user_id, chat_id=chat_id, message=data)
except WebSocketDisconnect:
manager.disconnect(chat_id, websocket)