### Python批量处理多DNS多域名的nslookup解析实现
在互联网技术领域中，域名解析是一项基本且重要的功能，它能够将人类易于记忆的域名转换成机器可读的IP地址。而`nslookup`命令作为检查域名解析问题的一种常用工具，在诊断网络故障、排查域名解析错误等方面发挥着重要作用。本文将详细介绍如何使用Python脚本实现对多个DNS服务器上多个域名的批量`nslookup`解析，并将结果保存至CSV文件中，便于后续的数据分析与管理。
#### 一、需求背景
在实际工作中，特别是对于网络安全管理员或网站运维人员来说，经常会遇到需要批量查询不同域名在不同DNS服务器上的解析结果的情况。例如，当需要检查某个域名在全球范围内的解析一致性时，就需要用到这种批量查询的能力。传统的手动操作不仅耗时费力，而且容易出错，因此开发一个自动化脚本来完成这项任务显得尤为重要。
#### 二、关键技术点
1. **Python环境搭建**：确保安装了Python环境。
2. **子进程调用**：使用`subprocess`模块来执行`nslookup`命令。
3. **CSV文件读写**：利用`csv`模块进行CSV文件的操作。
4. **数据结构设计**：合理组织存储查询结果的数据结构。
5. **异常处理**：考虑到网络不稳定等因素，增加适当的异常处理逻辑。
#### 三、具体实现步骤
1. **定义函数**：编写一个函数`get_nslookup`用于执行`nslookup`命令并返回解析结果。
- 输入参数：`domain`（待解析的域名）和`dns`（使用的DNS服务器）。
- 使用`subprocess.Popen`执行`nslookup`命令。
- 解析命令输出，提取IP地址等关键信息。
- 返回包含域名、DNS服务器及解析结果的列表。
2. **读取输入文件**：从CSV文件中读取待查询的域名和DNS服务器信息。
- 输入文件`domain.csv`格式示例：
```csv
domain,DNS
example.com,8.8.8.8
test.org,1.1.1.1
```
3. **执行查询**：遍历所有域名和DNS服务器，调用`get_nslookup`函数获取解析结果。
- 输出结果包括：域名、DNS服务器、以及该域名在该DNS服务器下的所有解析结果（IP地址等）。
4. **保存输出文件**：将所有查询结果保存至CSV文件`nslookup.csv`中，便于后续的数据分析。
#### 四、核心代码解析
```python
#coding=gbk
import subprocess
import csv
def get_nslookup(domain, dns):
# 使用subprocess执行nslookup命令
res = subprocess.Popen(f"nslookup {domain} {dns}", stdin=subprocess.PIPE, stdout=subprocess.PIPE).communicate()[0]
response = res.decode("gbk")
# 解析命令输出，提取关键信息
res_list = response.split("\n")
row_nslookup = [domain, dns]
row_ip = res_list[2].split()[:-1] # 提取IP地址
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]
# 获取DNS列表
dns_list = []
for row in domain_csv:
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(f'解析中：域名：{domain}___DNS：{dns}')
row_nslookup = get_nslookup(domain, dns)
nslookup_csv.writerow(row_nslookup)
print('执行完毕')
```
#### 五、总结
通过上述方法，我们可以有效地实现对多个DNS服务器上多个域名的批量`nslookup`解析，并将结果以CSV文件的形式保存下来。这种方式不仅可以大大提高工作效率，还能减少人为错误，非常适合在网络管理和维护场景中应用。同时，该脚本还具有良好的扩展性，可以根据具体需求进行相应的调整和优化。
