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("utf-8", "ignore")
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'.\domain.csv' # 输入域名列表文件
file_nslookup = r'.\nslookup1.csv' # 输出解析结果
with open(file_domain, 'r', newline='', encoding='utf-8') 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='utf-8') as rf:
domain_csv = csv.DictReader(rf, dialect=csv.excel)
dns_list = []
for row in domain_csv:
print(row['DNS'])
if row['DNS'] != '':
dns_list.append(row['DNS'])
with open(file_nslookup, 'w+', newline='', encoding='utf-8') 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)
except Exception as e:
print("error")
print('执行完毕')
