type lint

diff --git a/dns/btreezone.py b/dns/btreezone.py
index 1d981651d545e0c4e4bbee1d653da95d2bf9e6b1..de1107f12726c5fefeb2699c9e6ba80d6dc90dac 100644 (file)
--- a/dns/btreezone.py
+++ b/dns/btreezone.py
@@ -210,7 +210,9 @@ class WritableVersion(dns.zone.WritableVersion):
         if not replacement:
             assert isinstance(zone, dns.versioned.Zone)
             version = zone._versions[-1]
-            self.nodes: dns.btree.BTreeDict[dns.name.Name, Node] = dns.btree.BTreeDict(
+            self.nodes: dns.btree.BTreeDict[dns.name.Name, Node] = dns.btree.BTreeDict[
+                dns.name.Name, Node
+            ](
                 original=version.nodes  # type: ignore
             )
             self.delegations = Delegations(original=version.delegations)  # type: ignore

diff --git a/dns/dnssec.py b/dns/dnssec.py
index 35427b58d0283d479a18823fd82bb466bd687a18..2738e3dab687b21a019ee056d52382f963b5b1b9 100644 (file)
--- a/dns/dnssec.py
+++ b/dns/dnssec.py
@@ -1016,6 +1016,8 @@ def sign_zone(
                     dnskey_ttl = dnskey.ttl
                 else:
                     soa = _txn.get(zone.origin, dns.rdatatype.SOA)
+                    if soa is None:
+                        raise ValueError("zone does not have an SOA at the origin")
                     dnskey_ttl = soa.ttl
             for _, dnskey in keys:
                 _txn.add(zone.origin, dnskey_ttl, dnskey)

diff --git a/dns/transaction.py b/dns/transaction.py
index 3a5dec5d8d623caac59583ccd0a7be725a55d5d3..222b6757b1537d5618cf18058776f6f870f8e73a 100644 (file)
--- a/dns/transaction.py
+++ b/dns/transaction.py
@@ -79,13 +79,13 @@ class AlreadyEnded(dns.exception.DNSException):
     """Tried to use an already-ended transaction."""
 
 
-def _ensure_immutable_rdataset(rdataset):
+def _ensure_immutable_rdataset(rdataset) -> dns.rdataset.ImmutableRdataset | None:
     if rdataset is None or isinstance(rdataset, dns.rdataset.ImmutableRdataset):
         return rdataset
     return dns.rdataset.ImmutableRdataset(rdataset)
 
 
-def _ensure_immutable_node(node):
+def _ensure_immutable_node(node) -> dns.node.ImmutableNode | None:
     if node is None or node.is_immutable():
         return node
     return dns.node.ImmutableNode(node)
@@ -129,7 +129,7 @@ class Transaction:
         name: dns.name.Name | str | None,
         rdtype: dns.rdatatype.RdataType | str,
         covers: dns.rdatatype.RdataType | str = dns.rdatatype.NONE,
-    ) -> dns.rdataset.Rdataset:
+    ) -> dns.rdataset.Rdataset | None:
         """Return the rdataset associated with *name*, *rdtype*, and *covers*,
         or ``None`` if not found.