批量处理多DNS多域名的nslookup解析

利用EXCLE生成CSV文档，批量处理nslookup解析。并保存为CSV文档，方便进行查看：

输入文档格式：

data\domain.csv

最终输出文档情况：

data\nlookup.csv

代码：

# coding=gbk
import subprocess
import csv


def get_nslookup(domain, dns):
    res = subprocess.Popen("nslookup {0} {1}".format(domain, dns), stdin=subprocess.PIPE,
                           stdout=subprocess.PIPE).communicate()[0]
    response = res.decode("gbk")
    res_list = response.split("s:")
    row_nslookup = [domain, dns]
    row_ip = res_list[2].split()[:-1]
    row_nslookup.extend(row_ip)
    return row_nslookup


if __name__ == "__main__":
    file_domain = r'data\domain.csv'        # 输入文件
    file_nslookup = r'data\nslookup.csv'    # 输出文件
    with open(file_domain, 'r', newline='', encoding='gbk') as rf:
        domain_csv = csv.DictReader(rf, dialect=csv.excel)
        domain_list = [row['domain'] for row in domain_csv]

    with open(file_domain, 'r', newline='', encoding='gbk') as rf:
        domain_csv = csv.DictReader(rf, dialect=csv.excel)
        dns_list = []
        for row in domain_csv:
            print(row['DNS'])
            if row['DNS'] != '':        # 通常DNS数量少于需要监测的域名数量，做去空处理
                dns_list.append(row['DNS'])

    with open(file_nslookup, 'w+', newline='', encoding='gbk') as wf:
        nslookup_csv = csv.writer(wf, dialect=csv.excel)
        header = ['domain', 'DNS', 'nslookup_res...']
        nslookup_csv.writerow(header)
        for domain in domain_list:
            for dns in dns_list:
                print('解析中：域名：{0}___DNS：{1}'.format(domain, dns))
                row_nslookup = get_nslookup(domain, dns)
                nslookup_csv.writerow(row_nslookup)

print('执行完毕')

标签：domain,dns,nslookup,域名,DNS,csv,row
来源： https://blog.csdn.net/feelingwild/article/details/106984616