feat: add more cython typing to DNSRecord

bdraco · bdraco · commit 2c6ae75fc342 · 2023-06-18T21:55:08.000-05:00
diff --git a/src/zeroconf/_dns.pxd b/src/zeroconf/_dns.pxd
@@ -46,6 +46,14 @@ cdef class DNSRecord(DNSEntry):
 
     cpdef set_created_ttl(self, cython.float created, cython.float ttl)
 
+    cpdef get_remaining_ttl(self, cython.float current_time)
+
+    cpdef is_expired(self, cython.float current_time)
+
+    cpdef is_stale(self, cython.float current_time)
+
+    cpdef is_recent(self, cython.float current_time)
+
 cdef class DNSAddress(DNSRecord):
 
     cdef public cython.int _hash
diff --git a/src/zeroconf/_dns.py b/src/zeroconf/_dns.py
@@ -192,19 +192,19 @@ def get_expiration_time(self, percent: int_) -> float:
         return self.created + (percent * self.ttl * 10)
 
     # TODO: Switch to just int here
-    def get_remaining_ttl(self, now: float) -> int_or_float_:
+    def get_remaining_ttl(self, now: float_) -> int_or_float_:
         """Returns the remaining TTL in seconds."""
         return max(0, millis_to_seconds((self.created + (_EXPIRE_FULL_TIME_MS * self.ttl)) - now))
 
-    def is_expired(self, now: float) -> bool:
+    def is_expired(self, now: float_) -> bool:
         """Returns true if this record has expired."""
         return self.created + (_EXPIRE_FULL_TIME_MS * self.ttl) <= now
 
-    def is_stale(self, now: float) -> bool:
+    def is_stale(self, now: float_) -> bool:
         """Returns true if this record is at least half way expired."""
         return self.created + (_EXPIRE_STALE_TIME_MS * self.ttl) <= now
 
-    def is_recent(self, now: float) -> bool:
+    def is_recent(self, now: float_) -> bool:
         """Returns true if the record more than one quarter of its TTL remaining."""
         return self.created + (_RECENT_TIME_MS * self.ttl) > now
 
