William Grant
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
# This software is provided 'as-is', without any express or implied
|
|
# warranty. In no event will the author be held liable for any damages
|
|
# arising from the use of this software.
|
|
#
|
|
# Permission is granted to anyone to use this software for any purpose,
|
|
# including commercial applications, and to alter it and redistribute it
|
|
# freely, subject to the following restrictions:
|
|
#
|
|
# 1. The origin of this software must not be misrepresented; you must not
|
|
# claim that you wrote the original software. If you use this software
|
|
# in a product, an acknowledgment in the product documentation would be
|
|
# appreciated but is not required.
|
|
# 2. Altered source versions must be plainly marked as such, and must not be
|
|
# misrepresented as being the original software.
|
|
# 3. This notice may not be removed or altered from any source distribution.
|
|
#
|
|
# Copyright (c) 2008 Greg Hewgill http://hewgill.com
|
|
#
|
|
# This has been modified from the original software.
|
|
# Copyright (c) 2011 William Grant <me@williamgrant.id.au>
|
|
|
|
|
|
__all__ = [
|
|
'get_txt'
|
|
]
|
|
|
|
|
|
def get_txt_dnspython(name):
|
|
"""Return a TXT record associated with a DNS name."""
|
|
a = dns.resolver.query(name, dns.rdatatype.TXT)
|
|
for r in a.response.answer:
|
|
if r.rdtype == dns.rdatatype.TXT:
|
|
return b"".join(r.items[0].strings)
|
|
return None
|
|
|
|
|
|
def get_txt_pydns(name):
|
|
"""Return a TXT record associated with a DNS name."""
|
|
# Older pydns releases don't like a trailing dot.
|
|
if name.endswith('.'):
|
|
name = name[:-1]
|
|
DNS.ParseResolvConf()
|
|
response = DNS.DnsRequest(name, qtype='txt').req()
|
|
if not response.answers:
|
|
return None
|
|
return response.answers[0]['data'][0]
|
|
|
|
|
|
# Prefer dnspython if it's there, otherwise use pydns.
|
|
try:
|
|
import dns.resolver
|
|
get_txt = get_txt_dnspython
|
|
except ImportError:
|
|
import DNS
|
|
get_txt = get_txt_pydns
|