initialises SocketIOHandler in web server

This commit is contained in:
Mose Müller 2023-11-27 15:39:00 +01:00
parent e9df89765d
commit acaac6f0a6

View File

@ -1,3 +1,4 @@
import asyncio
import logging import logging
from pathlib import Path from pathlib import Path
from typing import Any, TypedDict from typing import Any, TypedDict
@ -12,6 +13,7 @@ from pydase import DataService
from pydase.data_service.data_service import process_callable_attribute from pydase.data_service.data_service import process_callable_attribute
from pydase.data_service.state_manager import StateManager from pydase.data_service.state_manager import StateManager
from pydase.utils.helpers import get_object_attr_from_path_list from pydase.utils.helpers import get_object_attr_from_path_list
from pydase.utils.logging import SocketIOHandler
from pydase.version import __version__ from pydase.version import __version__
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -91,6 +93,11 @@ class WebAPI:
self.setup_socketio() self.setup_socketio()
self.setup_fastapi_app() self.setup_fastapi_app()
self.setup_logging_handler()
def setup_logging_handler(self) -> None:
logger = logging.getLogger()
logger.addHandler(SocketIOHandler(self.__sio))
def setup_socketio(self) -> None: def setup_socketio(self) -> None:
# the socketio ASGI app, to notify clients when params update # the socketio ASGI app, to notify clients when params update