Dnspython: Setting query timeout/lifetime

Christopher Long picture Christopher Long · Jan 24, 2012 · Viewed 17.3k times · Source

I have a small script that checks a large list of domains for their MX records, everything works fine but when the script finds a domain with no record, it takes quite a long time to skip to the next one.

I have tried adding:

query.lifetime = 1.0
or
query.timeout = 1.0

but this doesn't seem to do anything. Does anyone know how this setting is configured?

My script is below, thanks for your time.

import dns.resolver
from dns.exception import DNSException
import dns.query
import csv

domains = csv.reader(open('domains.csv', 'rU'))
output = open('output.txt', 'w')
for row in domains:
    try:
        domain = row[0]
        query = dns.resolver.query(domain,'MX')
        query.lifetime = 1.0
    except DNSException:
        print "nothing here"
    for rdata in query:
            print domain, " ", rdata.exchange, 'has preference', rdata.preference
            output.writelines(domain)
            output.writelines(",")
            output.writelines(rdata.exchange.to_text())
            output.writelines("\n")

Answer

kindall picture kindall · Jan 24, 2012

You're setting the timeout after you've already performed the query. So that's not gonna do anything!

What you want to do instead is create a Resolver object, set its timeout, and then call its query() method. dns.resolver.query() is just a convenience function that instantiates a default Resolver object and invokes its query() method, so you need to do that manually if you don't want a default Resolver.

resolver = dns.resolver.Resolver()
resolver.timeout = 1
resolver.lifetime = 1

Then use this in your loop:

try:
    domain = row[0]
    query = resolver.query(domain,'MX')
except:
    # etc.

You should be able to use the same Resolver object for all queries.