mypy

bdraco · bdraco · commit 4db915bb9ba2 · 2021-08-05T09:48:28.000-05:00
diff --git a/zeroconf/_core.py b/zeroconf/_core.py
@@ -319,13 +319,13 @@ def _respond_query(
     @property
     def _socket_description(self) -> str:
         """A human readable description of the socket."""
+        assert self.transport is not None
         fileno = self.transport.get_extra_info('socket').fileno()
         sockname = self.transport.get_extra_info('sockname')
         return f"{fileno} ({sockname})"
 
     def error_received(self, exc: Exception) -> None:
         """Likely socket closed or IPv6."""
-        assert self.transport is not None
         # We preformat the message string with the socket as we want
         # log_exception_once to log a warrning message once PER EACH
         # different socket in case there are problems with multiple
diff --git a/zeroconf/_logger.py b/zeroconf/_logger.py
@@ -23,6 +23,7 @@
 import logging
 import sys
 from typing import Any, Dict, Union, cast
+from types import TracebackType
 
 log = logging.getLogger(__name__.split('.', maxsplit=1)[0])
 log.addHandler(logging.NullHandler())
@@ -63,7 +64,7 @@ def log_warning_once(cls, *args: Any) -> None:
         logger(*args)
 
     @classmethod
-    def log_exception_once(cls, exc_info, *args: Any) -> None:
+    def log_exception_once(cls, exc_info: TracebackType, *args: Any) -> None:
         msg_str = args[0]
         if msg_str not in cls._seen_logs:
             cls._seen_logs[msg_str] = 0
