Add get_options() helper.

rthalley · Feb 20, 2024 · 86a2841 · 86a2841
1 parent 5180041
commit 86a2841
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/dns/message.py b/dns/message.py
@@ -912,11 +912,13 @@ def set_opcode(self, opcode: dns.opcode.Opcode) -> None:
         self.flags &= 0x87FF
         self.flags |= dns.opcode.to_flags(opcode)
 
+    def get_options(self, otype: dns.edns.OptionType) -> List[dns.edns.Option]:
+        """Return the list of options of the specified type."""
+        return [option for option in self.options if option.otype == otype]
+
     def extended_errors(self) -> List[dns.edns.EDEOption]:
         """Return the list of Extended DNS Error (EDE) options in the message"""
-        return [
-            option for option in self.options if option.otype == dns.edns.OptionType.EDE
-        ]
+        return self.get_options(dns.edns.OptionType.EDE)
 
     def _get_one_rr_per_rrset(self, value):
         # What the caller picked is fine.