Merge pull request #974 from zyv/sdp-service-alias

Make sdp.Server.Service a type alias

Author: zxzxwu

bumble/device.py

@@ -2644,11 +2644,13 @@ def host(self, host: Host) -> None:
             host.link_key_provider = self.get_link_key
 
     @property
-    def sdp_service_records(self):
+    def sdp_service_records(self) -> dict[int, sdp.Server.Service]:
         return self.sdp_server.service_records
 
     @sdp_service_records.setter
-    def sdp_service_records(self, service_records):
+    def sdp_service_records(
+        self, service_records: dict[int, sdp.Server.Service]
+    ) -> None:
         self.sdp_server.service_records = service_records
 
     def lookup_connection(self, connection_handle: int) -> Connection | None:

bumble/sdp.py

@@ -22,7 +22,7 @@
 import struct
 from collections.abc import Iterable, Sequence
 from dataclasses import dataclass, field
-from typing import TYPE_CHECKING, Any, ClassVar, NewType, TypeVar
+from typing import TYPE_CHECKING, Any, ClassVar, TypeAlias, TypeVar
 
 from typing_extensions import Self
 
@@ -1076,7 +1076,7 @@ async def __aexit__(self, *args) -> None:
 class Server:
     CONTINUATION_STATE = bytes([0x01, 0x00])
     channel: l2cap.ClassicChannel | None
-    Service = NewType('Service', list[ServiceAttribute])
+    Service: TypeAlias = list[ServiceAttribute]
     service_records: dict[int, Service]
     current_response: None | bytes | tuple[int, list[int]]