Dnspython:设置查询超时/生命周期

时间:2012-01-24 15:31:36

标签: python timeout lifetime dnspython

我有一个小脚本,可以检查大量域名列表中的MX记录,一切正常,但是当脚本找到没有记录的域时,跳转到下一个域需要很长时间。

我尝试过添加:

query.lifetime = 1.0
or
query.timeout = 1.0

但这似乎没有做任何事情。有谁知道如何配置此设置?

我的脚本在下面,感谢您的时间。

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")

1 个答案:

答案 0 :(得分:17)

您已经在已经执行了查询后设置了超时。所以这不会做任何事情!

您要做的是创建一个Resolver对象,设置超时,然后调用其query()方法。 dns.resolver.query()只是一个便捷函数,可以实例化默认的Resolver对象并调用其query()方法,因此如果您不想使用默认Resolver,则需要手动执行此操作。

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

然后在你的循环中使用它:

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

您应该能够为所有查询使用相同的Resolver对象。