Dnspython: Thiết lập thời gian chờ truy vấn/toàn thời gian

Question

Tôi có một tập lệnh nhỏ kiểm tra danh sách tên miền lớn cho bản ghi MX của họ, mọi thứ hoạt động tốt nhưng khi tập lệnh tìm thấy tên miền không có bản ghi, sẽ mất khá nhiều thời gian để bỏ qua cai tiêp theo.

Tôi đã thử thêm:

query.lifetime = 1.0 
or 
query.timeout = 1.0 

nhưng điều này dường như không làm gì cả. Có ai biết cách thiết lập cấu hình này không?

Tập lệnh của tôi bên dưới, cảm ơn thời gian của bạn.

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 1

Bạn đang đặt thời gian chờ sau bạn đã thực hiện truy vấn. Vì vậy, đó sẽ không làm bất cứ điều gì!

Thay vào đó, bạn muốn tạo đối tượng Resolver, đặt thời gian chờ và sau đó gọi phương thức query(). dns.resolver.query() chỉ là một chức năng thuận tiện để khởi tạo đối tượng mặc định Resolver và gọi phương thức query() của nó, vì vậy bạn cần thực hiện điều đó theo cách thủ công nếu bạn không muốn mặc định Resolver.

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

Sau đó sử dụng này trong vòng lặp của bạn:

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

Bạn sẽ có thể sử dụng cùng một đối tượng Resolver cho tất cả các truy vấn.