diff --git a/src/pyg2o/server_v3.py b/src/pyg2o/server_v3.py index 219723e..c0a6a8d 100644 --- a/src/pyg2o/server_v3.py +++ b/src/pyg2o/server_v3.py @@ -1,4 +1,5 @@ import json +import logging from fastapi import WebSocket, FastAPI, Depends, HTTPException, WebSocketDisconnect, WebSocketException from fastapi.security import HTTPBasic, HTTPBasicCredentials from uuid import uuid4 @@ -11,6 +12,9 @@ class Server: self._server_username = server_username self._server_password = server_password self._client_password = client_password + + self._logger = logging.getLogger(__name__) + self._logger.addHandler(logging.NullHandler()) self._server_connection: WebSocket | None = None self._register_routes(app) @@ -73,20 +77,20 @@ class Server: await websocket.accept() self._server_connection = websocket + self._logger.info('PyG2O сервер подключился') - #TODO: Заменить принты на логирование try: while True: try: data = await websocket.receive_text() message_data = json.loads(data) - print('Server message:', message_data) + self._logger.info(f'Сообщение сервера: {message_data}') except json.JSONDecodeError as e: - print('JSON Decode exception:', e) + self._logger.info(f'Ошибка декодирования JSON: {e}') except WebSocketDisconnect: - print('Server socket disconnected') + self._logger.info('PyG2O сервер отключился') except WebSocketException as e: - print('Server socket exception:', e) + self._logger.info(f'Ошибка подключения PyG2O сервера: {e}') async def _process_server_message(self, message: dict): match message: