import subprocess
import csv
import time

dns_list = ['8.8.8.8', '8.8.4.4', '208.67.222.222', '208.67.220.220']
domain_name = 'example.com'
output_file = 'dns_results.csv'

with open(output_file, 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['Domain Name', 'DNS Server', 'IP Address']) # added 'Domain Name' column

    for dns in dns_list:
        print(f"正在使用DNS服务器: {dns}")
        print(f"正在解析域名: {domain_name}")
        command = f'nslookup {domain_name} {dns}'
        result = subprocess.run(command, capture_output=True, text=True)
        ip_address = result.stdout.split('Address: ')[-1].split('\n')[0]
        writer.writerow([domain_name, dns, ip_address]) # added domain_name to the row
        time.sleep(1) # to avoid overwhelming the DNS servers

import csv
import os

dns_servers = ['1.0.0.1','8.8.8.8','119.29.29.29','1.12.12.12','233.5.5.5','9.9.9.9','208.67.222.222']
domains = ['imfine.eu.org'] # list of domains to lookup

results = [] # list to store results

for domain in domains:
    for server in dns_servers:
        print(f"\n等待DNS服务器：【{server}】 解析域名：【{domain}】")
        response = os.popen(f'nslookup {domain} {server}').read() # run nslookup command
        ip_address = response.split('addresses:')[-1].strip() # extract IP address from response
        results.append([domain, server, ip_address]) # add result to list
        print(f"解析结果: {ip_address}\n") # print IP address

# write results to csv file
try:
    with open('dns_results.csv', 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(['Domain', 'DNS Server', 'IP Address'])
        writer.writerows(results)
    print('DNS lookup complete. Results saved to dns_results.csv')
except Exception as e:
    print(f"Error occurred while writing to csv file: {e}")
    raise e