Improve type annotations

Author: puddly

zigpy_znp/frames.py

@@ -11,7 +11,7 @@ class GeneralFrame:
     header: t.CommandHeader
     data: t.Bytes
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         # We're frozen so `self.header = ...` is disallowed
         if not isinstance(self.header, t.CommandHeader):
             object.__setattr__(self, "header", t.CommandHeader(self.header))

zigpy_znp/logger.py

@@ -3,7 +3,7 @@
 _TRACE = 5
 
 
-def _find_trace_level():
+def _find_trace_level() -> int:
     if logging.getLevelName(_TRACE) != f"Level {_TRACE}":
         # If a level 5 exists, use it
         return _TRACE

zigpy_znp/types/struct.py

@@ -18,7 +18,7 @@ def real_cls(cls) -> type:
         return next(c for c in cls.__mro__ if c.__name__ != "Optional")
 
     @classmethod
-    def _annotations(cls) -> typing.List[type]:
+    def _annotations(cls) -> typing.Dict[str, typing.Any]:
         # First get our proper subclasses
         subclasses = []
 
@@ -195,7 +195,7 @@ def serialize(self) -> bytes:
         )
 
     @classmethod
-    def deserialize(cls, data: bytes) -> "Struct":
+    def deserialize(cls, data: bytes) -> typing.Tuple["Struct", bytes]:
         instance = cls()
 
         for field in cls.fields():
@@ -239,7 +239,7 @@ class StructField:
     dynamic_type: typing.Optional[typing.Callable[[Struct], type]] = None
     requires: typing.Optional[typing.Callable[[Struct], bool]] = None
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         # Fail to initialize if the concrete type is invalid
         if self.dynamic_type is None:
             self.concrete_type

zigpy_znp/uart.py

@@ -140,7 +140,7 @@ def _extract_frame(self) -> frames.TransportFrame:
 
         return frame
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return f"<{type(self).__name__} for {self._api}>"