Updated type hinting in Websocket API module to use pipe-based unions now that Python 3.9 has been dropped

Author: tieneupin

src/murfey/server/api/websocket.py

@@ -4,7 +4,7 @@
 import json
 import logging
 from datetime import datetime
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from fastapi import APIRouter, WebSocket, WebSocketDisconnect
 from sqlmodel import Session, select
@@ -27,7 +27,7 @@ def __init__(self):
     async def connect(
         self,
         websocket: WebSocket,
-        client_id: Union[int, str],
+        client_id: int | str,
         register_client: bool = True,
     ):
         await websocket.accept()
@@ -48,7 +48,7 @@ def _register_new_client(client_id: int):
         murfey_db.commit()
         murfey_db.close()
 
-    def disconnect(self, client_id: Union[int, str], unregister_client: bool = True):
+    def disconnect(self, client_id: int | str, unregister_client: bool = True):
         self.active_connections.pop(client_id)
         if unregister_client:
             murfey_db: Session = next(get_murfey_db_session())
@@ -97,7 +97,7 @@ async def websocket_endpoint(websocket: WebSocket, client_id: int):
 @ws.websocket("/connect/{client_id}")
 async def websocket_connection_endpoint(
     websocket: WebSocket,
-    client_id: Union[int, str],
+    client_id: int | str,
 ):
     await manager.connect(websocket, client_id, register_client=False)
     await manager.broadcast(f"Client {client_id} joined")
@@ -161,7 +161,7 @@ async def close_ws_connection(client_id: int):
 
 
 @ws.delete("/connect/{client_id}")
-async def close_unrecorded_ws_connection(client_id: Union[int, str]):
+async def close_unrecorded_ws_connection(client_id: int | str):
     client_id_str = str(client_id).replace("\r\n", "").replace("\n", "")
     log.info(f"Disconnecting {client_id_str}")
     manager.disconnect(client_id)