Add exception blocks for query errors.

pull/173/head
Fred Morris 2018-03-08 15:26:39 -08:00
parent a2ebcdb358
commit d0f618b648
1 changed files with 27 additions and 18 deletions

View File

@ -1,5 +1,5 @@
import json import json
from ._dnsdb_query.dnsdb_query import DnsdbClient from ._dnsdb_query.dnsdb_query import DnsdbClient, QueryError
misperrors = {'error': 'Error'} misperrors = {'error': 'Error'}
@ -41,26 +41,35 @@ def handler(q=False):
def lookup_name(client, name): def lookup_name(client, name):
res = client.query_rrset(name) # RRSET = entries in the left-hand side of the domain name related labels try:
for item in res: res = client.query_rrset(name) # RRSET = entries in the left-hand side of the domain name related labels
if item.get('rrtype') in ['A', 'AAAA', 'CNAME']: for item in res:
for i in item.get('rdata'): if item.get('rrtype') in ['A', 'AAAA', 'CNAME']:
yield(i.rstrip('.')) for i in item.get('rdata'):
if item.get('rrtype') in ['SOA']: yield(i.rstrip('.'))
for i in item.get('rdata'): if item.get('rrtype') in ['SOA']:
# grab email field and replace first dot by @ to convert to an email address for i in item.get('rdata'):
yield(i.split(' ')[1].rstrip('.').replace('.', '@', 1)) # grab email field and replace first dot by @ to convert to an email address
# res = client.query_rdata_name(name) # RDATA = entries on the right-hand side of the domain name related labels yield(i.split(' ')[1].rstrip('.').replace('.', '@', 1))
# for item in res: except QueryError as e:
# if item.get('rrtype') in ['A', 'AAAA', 'CNAME']: pass
# yield(item.get('rrname').rstrip('.'))
try:
res = client.query_rdata_name(name) # RDATA = entries on the right-hand side of the domain name related labels
for item in res:
if item.get('rrtype') in ['A', 'AAAA', 'CNAME']:
yield(item.get('rrname').rstrip('.'))
except QueryError as e:
pass
def lookup_ip(client, ip): def lookup_ip(client, ip):
res = client.query_rdata_ip(ip) try:
for item in res: res = client.query_rdata_ip(ip)
print(item) for item in res:
yield(item['rrname'].rstrip('.')) yield(item['rrname'].rstrip('.'))
except QueryError as e:
pass
def introspection(): def introspection():