Skip to content

Commit 0907d7b

Browse files
committed
Updated type hints
1 parent 1d8a11a commit 0907d7b

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

src/murfey/client/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from pathlib import Path
1414
from pprint import pprint
1515
from queue import Queue
16-
from typing import List, Literal
16+
from typing import Literal
1717
from urllib.parse import ParseResult, urlparse
1818

1919
import requests
@@ -41,7 +41,7 @@ def write_config(config: configparser.ConfigParser):
4141

4242

4343
def main_loop(
44-
source_watchers: List[murfey.client.watchdir.DirWatcher],
44+
source_watchers: list[murfey.client.watchdir.DirWatcher],
4545
appearance_time: float,
4646
transfer_all: bool,
4747
):

src/murfey/server/websocket.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import json
55
import logging
66
from datetime import datetime
7-
from typing import Any, Dict, TypeVar, Union
7+
from typing import Any, TypeVar
88

99
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
1010
from sqlmodel import Session, select
@@ -22,7 +22,7 @@
2222

2323
class ConnectionManager:
2424
def __init__(self):
25-
self.active_connections: Dict[int | str, WebSocket] = {}
25+
self.active_connections: dict[int | str, WebSocket] = {}
2626

2727
async def connect(
2828
self, websocket: WebSocket, client_id: int | str, register_client: bool = True
@@ -93,7 +93,8 @@ async def websocket_endpoint(websocket: WebSocket, client_id: int):
9393

9494
@ws.websocket("/connect/{client_id}")
9595
async def websocket_connection_endpoint(
96-
websocket: WebSocket, client_id: Union[int, str]
96+
websocket: WebSocket,
97+
client_id: int | str,
9798
):
9899
await manager.connect(websocket, client_id, register_client=False)
99100
await manager.broadcast(f"Client {client_id} joined")
@@ -157,7 +158,7 @@ async def close_ws_connection(client_id: int):
157158

158159

159160
@ws.delete("/connect/{client_id}")
160-
async def close_unrecorded_ws_connection(client_id: Union[int, str]):
161+
async def close_unrecorded_ws_connection(client_id: int | str):
161162
client_id_str = str(client_id).replace("\r\n", "").replace("\n", "")
162163
log.info(f"Disconnecting {client_id_str}")
163164
manager.disconnect(client_id)

0 commit comments

Comments
 (0)