From 7e5aef167663dc209fda3bd0aeecaa802e50d924 Mon Sep 17 00:00:00 2001 From: "Stuart D. Gathman" Date: Tue, 12 Jun 2012 23:09:15 -0400 Subject: [PATCH] Trap NXDOMAIN exception with dnspython. --- dkim/dnsplug.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/dkim/dnsplug.py b/dkim/dnsplug.py index b59ef0a..8f258b0 100644 --- a/dkim/dnsplug.py +++ b/dkim/dnsplug.py @@ -27,10 +27,12 @@ __all__ = [ def get_txt_dnspython(name): """Return a TXT record associated with a DNS name.""" - a = dns.resolver.query(name, dns.rdatatype.TXT,raise_on_no_answer=False) - for r in a.response.answer: - if r.rdtype == dns.rdatatype.TXT: - return b"".join(r.items[0].strings) + try: + a = dns.resolver.query(name, dns.rdatatype.TXT,raise_on_no_answer=False) + for r in a.response.answer: + if r.rdtype == dns.rdatatype.TXT: + return b"".join(r.items[0].strings) + except dns.resolver.NXDOMAIN: pass return None