feature: avoid more py types

bdraco · bdraco · commit b9a2c0c18e44 · 2023-09-11T12:59:16.000-05:00
diff --git a/src/zeroconf/_protocol/outgoing.pxd b/src/zeroconf/_protocol/outgoing.pxd
@@ -31,7 +31,7 @@ cdef class DNSOutgoing:
     cdef public cython.dict names
     cdef public cython.list data
     cdef public unsigned int size
-    cdef public object allow_long
+    cdef public bint allow_long
     cdef public object state
     cdef public cython.list questions
     cdef public cython.list answers
@@ -48,18 +48,18 @@ cdef class DNSOutgoing:
 
     cdef _write_int(self, object value)
 
-    cdef _write_question(self, DNSQuestion question)
+    cdef cython.bint _write_question(self, DNSQuestion question)
 
     @cython.locals(
         d=cython.bytes,
         data_view=cython.list,
         length=cython.uint
     )
-    cdef _write_record(self, DNSRecord record, object now)
+    cdef cython.bint _write_record(self, DNSRecord record, object now)
 
     cdef _write_record_class(self, DNSEntry record)
 
-    cdef _check_data_limit_or_rollback(self, object start_data_length, object start_size)
+    cdef cython.bint _check_data_limit_or_rollback(self, cython.uint start_data_length, cython.uint start_size)
 
     cdef _write_questions_from_offset(self, object questions_offset)
 
diff --git a/src/zeroconf/_protocol/outgoing.py b/src/zeroconf/_protocol/outgoing.py
@@ -294,7 +294,8 @@ def write_name(self, name: str_) -> None:
 
     def _write_question(self, question: DNSQuestion_) -> bool:
         """Writes a question to the packet"""
-        start_data_length, start_size = len(self.data), self.size
+        start_data_length = len(self.data)
+        start_size = self.size
         self.write_name(question.name)
         self.write_short(question.type)
         self._write_record_class(question)
@@ -315,7 +316,8 @@ def _write_record(self, record: DNSRecord_, now: float_) -> bool:
         """Writes a record (answer, authoritative answer, additional) to
         the packet.  Returns True on success, or False if we did not
         because the packet because the record does not fit."""
-        start_data_length, start_size = len(self.data), self.size
+        start_data_length = len(self.data)
+        start_size = self.size
         self.write_name(record.name)
         self.write_short(record.type)
         self._write_record_class(record)
