99import websockets
1010from channels .db import database_sync_to_async
1111from channels .generic .websocket import AsyncJsonWebsocketConsumer , AsyncWebsocketConsumer
12+ from websockets import client as client
1213
1314from django .conf import settings
1415
@@ -28,7 +29,7 @@ def __init__(self, *args: Any, **kwargs: Any):
2829 self .site : Optional [Site ] = None
2930 self .connected = False
3031
31- self .status_websocket : Optional [websockets . client .WebSocketClientProtocol ] = None
32+ self .status_websocket : Optional [client .WebSocketClientProtocol ] = None
3233
3334 async def connect (self ) -> None :
3435 if not self .scope ["user" ].is_authenticated :
@@ -251,7 +252,7 @@ def __init__(self, *args: Any, **kwargs: Any):
251252 self .site : Optional [Site ] = None
252253 self .connected = False
253254
254- self .terminal_websock : Optional [websockets . client .WebSocketClientProtocol ] = None
255+ self .terminal_websock : Optional [client .WebSocketClientProtocol ] = None
255256
256257 async def connect (self ) -> None :
257258 if not self .scope ["user" ].is_authenticated :
@@ -397,7 +398,7 @@ def __init__(self, *args: Any, **kwargs: Any):
397398 self .site : Optional [Site ] = None
398399 self .connected = False
399400
400- self .monitor_websocks : List [websockets . client .WebSocketClientProtocol ] = []
401+ self .monitor_websocks : List [client .WebSocketClientProtocol ] = []
401402
402403 async def connect (self ) -> None :
403404 if not self .scope ["user" ].is_authenticated :
@@ -461,7 +462,7 @@ async def open_monitor_connections(self) -> None:
461462
462463 async def monitor_mainloop (
463464 self ,
464- monitor_websock : websockets . client .WebSocketClientProtocol ,
465+ monitor_websock : client .WebSocketClientProtocol ,
465466 ) -> None :
466467 while True :
467468 try :
@@ -516,7 +517,7 @@ def __init__(self, *args: Any, **kwargs: Any):
516517
517518 self .site : Optional [Site ] = None
518519
519- self .logs_websock : Optional [websockets . client .WebSocketClientProtocol ] = None
520+ self .logs_websock : Optional [client .WebSocketClientProtocol ] = None
520521
521522 async def connect (self ) -> None :
522523 if not self .scope ["user" ].is_authenticated :
@@ -616,7 +617,7 @@ def __init__(self, *args: Any, **kwargs: Any):
616617
617618 self .site_ids : List [int ] = []
618619
619- self .monitor_websock : Optional [websockets . client .WebSocketClientProtocol ] = None
620+ self .monitor_websock : Optional [client .WebSocketClientProtocol ] = None
620621
621622 async def connect (self ) -> None :
622623 if not self .scope ["user" ].is_authenticated :
@@ -674,7 +675,7 @@ async def open_monitor_connection(self) -> None:
674675
675676 async def monitor_mainloop (
676677 self ,
677- monitor_websock : websockets . client .WebSocketClientProtocol ,
678+ monitor_websock : client .WebSocketClientProtocol ,
678679 ) -> None :
679680 while True :
680681 try :
0 commit comments