批量查询域名

import dns.resolver
import concurrent.futures

def query_dns(domain, record_type):
    try:
        answers = dns.resolver.resolve(domain, record_type)
        return {domain: [rdata.to_text() for rdata in answers]}
    except dns.resolver.NXDOMAIN:
        return {domain: "Domain does not exist"}
    except dns.resolver.NoAnswer:
        return {domain: "No answer for the query"}
    except Exception as e:
        return {domain: str(e)}

def batch_query(domains, record_type):
    results = {}
    with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
        future_to_domain = {executor.submit(query_dns, domain, record_type): domain for domain in domains}
        for future in concurrent.futures.as_completed(future_to_domain):
            domain = future_to_domain[future]
            try:
                data = future.result()
            except Exception as exc:
                data = {domain: str(exc)}
            results.update(data)
    return results

# 示例使用
domains = ["example.com", "google.com", "nonexistentdomain.xyz"]
record_type = "A"
print(batch_query(domains, record_type))