Fix mypy nags in ritz.py

Uninett · Oct 13, 2023 · cca622b · cca622b
1 parent 8794553
commit cca622b
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/zinolib/ritz.py b/src/zinolib/ritz.py
@@ -84,7 +84,7 @@
 from time import mktime
 import re
 from os.path import expanduser
-from typing import NamedTuple
+from typing import NamedTuple, List, Tuple, Union
 import codecs
 import select
 
@@ -108,7 +108,9 @@ class ProtocolError(Exception):
 
 
 NotifierResponse = NamedTuple("NotifierResponse", [("id", int), ("type", str), ("info", str)])
-DataResponse = NamedTuple("DataResponse", [("data", str), ("header", str)])
+DataResponseData = Union[List[str], str]
+DataResponseHeader = Union[Tuple[()], Tuple[int, str]]
+DataResponse = NamedTuple("DataResponse", [("data", DataResponseData), ("header", DataResponseHeader)])
 
 
 class caseState(enum.Enum):
@@ -377,9 +379,9 @@ def _request(self, command: bytes, recv_buffer=4096, **_):
         """
         global logger
         buffer = ""
-        data = True
-        header = False
-        r = []
+        data = b"DUMMY"  # truthy, for mypy
+        header: DataResponseHeader = ()
+        r: List[str] = []
         logger.debug("send: %s" % command.__repr__())
         if command:
             delimiter = bytes(self.DELIMITER, 'ascii')