Инициализация интернационализации. Перевод на en, ru для эндпоинтов api/v1

This commit is contained in:
parent 48e8551b0e
commit 9ac52c8a8b
16 changed files with 990 additions and 181 deletions

View file

@ -2,8 +2,10 @@ import asyncio
import logging
from typing import Set
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
from fastapi import APIRouter, WebSocket, WebSocketDisconnect, Depends
from starlette.websockets import WebSocketState
from utils.i18n import get_translations, _
from babel.support import Translations
from .network import update_container_position, get_network_data
@ -70,15 +72,13 @@ class WebSocketManager:
except WebSocketDisconnect:
self.disconnect(connection)
except Exception as e:
logger.error(f"Сообщение об ошибке трансляции: {e}")
logger.error(f"Broadcast error: {e}")
self.disconnect(connection)
manager = WebSocketManager()
@router.websocket("/api/v1")
async def websocket_endpoint(websocket: WebSocket):
async def websocket_endpoint(websocket: WebSocket, translations: Translations = Depends(get_translations("api_v1"))):
"""Обработчик WebSocket-соединения.
Управляет соединением клиента, принимает данные для обновления позиций
@ -86,6 +86,7 @@ async def websocket_endpoint(websocket: WebSocket):
Args:
websocket (WebSocket): Объект WebSocket для обработки соединения.
translations: Объект переводов для текущего языка.
Raises:
WebSocketDisconnect: Если клиент отключился.
@ -95,26 +96,25 @@ async def websocket_endpoint(websocket: WebSocket):
try:
while True:
data = await websocket.receive_json()
logger.info(f"Полученные данные: {data}")
logger.info(f"Received data: {data}")
if all(key in data for key in ["container_id", "container_x", "container_y"]):
await update_container_position(data)
network_data = await get_network_data()
logger.info(f"Текущие сетевые данные: {network_data}")
logger.info(f"Current network data: {network_data}")
await manager.broadcast(network_data)
await asyncio.sleep(2)
except WebSocketDisconnect:
logger.info("Клиент отключен")
logger.info("Client disconnected")
except Exception as e:
logger.error(f"Ошибка в websocket_endpoint:{e}")
logger.error(f"Error in websocket_endpoint: {e}")
finally:
manager.disconnect(websocket)
if websocket.client_state != WebSocketState.DISCONNECTED:
await websocket.close()
@router.on_event("startup")
async def startup_event():
"""Запуск таска для трансляции сетевых данных при старте приложения.
@ -126,13 +126,15 @@ async def startup_event():
"""
asyncio.create_task(broadcast_data())
async def broadcast_data():
async def broadcast_data(translations: Translations = Depends(get_translations("api_v1"))):
"""Таск для периодической трансляции сетевых данных.
Выполняет сканирование сети и отправляет данные всем подключённым клиентам
с интервалом в 2 секунды.
Args:
translations: Объект переводов для текущего языка.
Raises:
Exception: Если произошла ошибка при получении данных или трансляции.
"""
@ -140,13 +142,11 @@ async def broadcast_data():
try:
network_data = await get_network_data()
if "activation" not in network_data:
logger.error(
"Ошибка: Ключ 'activation' отсутствует в полученных данных сети"
)
logger.error(translations.gettext("Ошибка: Ключ 'activation' отсутствует в полученных данных сети"))
await asyncio.sleep(2)
continue
await manager.broadcast(network_data)
except Exception as e:
logger.error(f"Ошибка в broadcast_data:{e}")
logger.error(f"Error in broadcast_data: {e}")
await asyncio.sleep(2)