handle missing dns.resolver library quietly

This commit is contained in:
Jay Lee
2014-09-23 21:08:58 -04:00
parent 7f6683e21d
commit 37ade0059f

44
gam.py
View File

@ -5022,28 +5022,34 @@ def doSiteVerifyAttempt():
print u'Method: %s' % verify_data[u'method'] print u'Method: %s' % verify_data[u'method']
print u'Token: %s' % verify_data[u'token'] print u'Token: %s' % verify_data[u'token']
if verify_data[u'method'] == u'DNS_CNAME': if verify_data[u'method'] == u'DNS_CNAME':
import dns.resolver
resolver = dns.resolver.Resolver()
resolver.nameservers = nameservers=[u'8.8.8.8', u'8.8.4.4']
cname_token = verify_data[u'token']
cname_list = cname_token.split(u' ')
cname_subdomain = cname_list[0]
try: try:
answers = resolver.query(u'%s.%s' % (cname_subdomain, a_domain), u'A') import dns.resolver
for answer in answers: resolver = dns.resolver.Resolver()
print u'DNS Record: %s' % answer resolver.nameservers = nameservers=[u'8.8.8.8', u'8.8.4.4']
except dns.resolver.NXDOMAIN: cname_token = verify_data[u'token']
print u'ERROR: No such domain found in DNS!' cname_list = cname_token.split(u' ')
cname_subdomain = cname_list[0]
try:
answers = resolver.query(u'%s.%s' % (cname_subdomain, a_domain), u'A')
for answer in answers:
print u'DNS Record: %s' % answer
except (dns.resolver.NXDOMAIN, dns.resolver.NoAnswer):
print u'ERROR: No such domain found in DNS!'
except ImportError:
pass
elif verify_data[u'method'] == u'DNS_TXT': elif verify_data[u'method'] == u'DNS_TXT':
import dns.resolver
resolver = dns.resolver.Resolver()
resolver.nameservers = nameservers=[u'8.8.8.8', u'8.8.4.4']
try: try:
answers = resolver.query(a_domain, u'TXT') import dns.resolver
for answer in answers: resolver = dns.resolver.Resolver()
print u'DNS Record: %s' % str(answer).replace(u'"', u'') resolver.nameservers = nameservers=[u'8.8.8.8', u'8.8.4.4']
except dns.resolver.NXDOMAIN: try:
print u'ERROR: no such domain found in DNS!' answers = resolver.query(a_domain, u'TXT')
for answer in answers:
print u'DNS Record: %s' % str(answer).replace(u'"', u'')
except dns.resolver.NXDOMAIN:
print u'ERROR: no such domain found in DNS!'
except ImportError:
pass
return return
print u'SUCCESS!' print u'SUCCESS!'
print u'Verified: %s' % verify_result[u'site'][u'identifier'] print u'Verified: %s' % verify_result[u'site'][u'identifier']