refac: modernize imports, standardize type hints and docstrings

backend/open_webui/socket/main.py

@@ -1,55 +1,48 @@
 import asyncio
-import random
-
-import socketio
 import logging
+import random
 import sys
 import time
 from typing import Dict, Set
-from redis import asyncio as aioredis
+
 import pycrdt as Y
-
-from open_webui.models.users import Users, UserNameResponse
-from open_webui.models.channels import Channels
-from open_webui.models.chats import Chats
-from open_webui.models.notes import Notes, NoteUpdateForm
-from open_webui.utils.redis import (
-    get_sentinels_from_env,
-    get_sentinel_url_from_env,
-)
-
+import socketio
 from open_webui.config import (
     CORS_ALLOW_ORIGIN,
 )
-
 from open_webui.env import (
-    VERSION,
     ENABLE_WEBSOCKET_SUPPORT,
+    GLOBAL_LOG_LEVEL,
+    REDIS_KEY_PREFIX,
+    VERSION,
+    WEBSOCKET_EVENT_CALLER_TIMEOUT,
     WEBSOCKET_MANAGER,
-    WEBSOCKET_REDIS_URL,
     WEBSOCKET_REDIS_CLUSTER,
     WEBSOCKET_REDIS_LOCK_TIMEOUT,
-    WEBSOCKET_SENTINEL_PORT,
-    WEBSOCKET_SENTINEL_HOSTS,
-    REDIS_KEY_PREFIX,
     WEBSOCKET_REDIS_OPTIONS,
-    WEBSOCKET_SERVER_PING_TIMEOUT,
-    WEBSOCKET_SERVER_PING_INTERVAL,
-    WEBSOCKET_SERVER_LOGGING,
+    WEBSOCKET_REDIS_URL,
+    WEBSOCKET_SENTINEL_HOSTS,
+    WEBSOCKET_SENTINEL_PORT,
     WEBSOCKET_SERVER_ENGINEIO_LOGGING,
-    WEBSOCKET_EVENT_CALLER_TIMEOUT,
+    WEBSOCKET_SERVER_LOGGING,
+    WEBSOCKET_SERVER_PING_INTERVAL,
+    WEBSOCKET_SERVER_PING_TIMEOUT,
 )
-from open_webui.utils.auth import decode_token
+from open_webui.models.access_grants import AccessGrants
+from open_webui.models.channels import Channels
+from open_webui.models.chats import Chats
+from open_webui.models.notes import Notes, NoteUpdateForm
+from open_webui.models.users import UserNameResponse, Users
 from open_webui.socket.utils import RedisDict, RedisLock, YdocManager
 from open_webui.tasks import create_task, stop_item_tasks
-from open_webui.utils.redis import get_redis_connection
 from open_webui.utils.access_control import has_permission
-from open_webui.models.access_grants import AccessGrants
-
-
-from open_webui.env import (
-    GLOBAL_LOG_LEVEL,
+from open_webui.utils.auth import decode_token
+from open_webui.utils.redis import (
+    get_redis_connection,
+    get_sentinel_url_from_env,
+    get_sentinels_from_env,
 )
+from redis import asyncio as aioredis
 
 logging.basicConfig(stream=sys.stdout, level=GLOBAL_LOG_LEVEL)
 log = logging.getLogger(__name__)
@@ -371,15 +364,15 @@ async def connect(sid, environ, auth):
 
 @sio.on('user-join')
 async def user_join(sid, data):
-    auth = data['auth'] if 'auth' in data else None
+    auth = data.get('auth')
     if not auth or 'token' not in auth:
         return
 
-    data = decode_token(auth['token'])
-    if data is None or 'id' not in data:
+    token_data = decode_token(auth['token'])
+    if token_data is None or 'id' not in token_data:
         return
 
-    user = await Users.get_user_by_id(data['id'])
+    user = await Users.get_user_by_id(token_data['id'])
     if not user:
         return
 
@@ -845,7 +838,7 @@ async def _make_channel_emitter(request_info):
     THROTTLE_INTERVAL = 0.15  # ~6 updates/sec
 
     async def _emit_channel_update(content: str, done: bool = False):
-        from open_webui.models.messages import Messages, MessageForm
+        from open_webui.models.messages import MessageForm, Messages
 
         update_form = MessageForm(content=content)
         if done: