fix import

src/middlewared/middlewared/apps/websocket_app.py

@@ -14,7 +14,7 @@
     RpcWebSocketAppEvent,
 )
 from middlewared.job import Job
-from middlewared.logger import logger
+from middlewared.logger import Logger
 
 from middlewared.schema import Error as SchemaError
 from middlewared.service_exception import (
@@ -51,7 +51,7 @@ def __init__(
         self.request = request
         self.response = response
         self.handshake = False
-        self.logger = logger.Logger("application").getLogger()
+        self.logger = Logger("websock_app").getLogger()
         # Allow atmost 10 concurrent calls and only queue up until 20
         self._softhardsemaphore = SoftHardSemaphore(10, 20)
         self._py_exceptions = False

src/middlewared/middlewared/main.py

@@ -70,7 +70,7 @@
 
 from truenas_api_client import json
 
-from . import logger
+from .logger import Logger
 
 SYSTEMD_EXTEND_USECS = 240000000  # 4mins in microseconds
 
@@ -99,9 +99,7 @@ def __init__(
         print_version=True,
     ):
         super().__init__()
-        self.logger = logger.Logger(
-            'middlewared', debug_level, log_format
-        ).getLogger()
+        self.logger = Logger('middlewared', debug_level, log_format).getLogger()
         if print_version:
             self.logger.info('Starting %s middleware', sw_version())
         self.loop_debug = loop_debug