拒绝困顿，拥抱智能：打造快速批量IP归属地查询小工具

后端

2024-01-17 15:55:21

快速批量查询IP归属地信息的实用工具

背景

安全工作中，分析海量IP地址是至关重要的。获取IP归属地信息是关键一步，但手动查询极其耗时费力。为了提高效率，我开发了一个小工具，可以快速批量查询IP归属地信息。

工具设计

功能需求：

批量查询IP归属地信息
自动去重，避免重复查询
按国内外汇总，便于后续分析
智能识别错误IP，防止影响分析
操作简单，可快速处理海量数据

技术选型：

编程语言：Python（简单、强大、可移植性强）
第三方库：ipwhois（用于查询IP归属地信息）

工具实现

导入必要的Python库：

import ipwhois
import csv

加载IP地址列表：

ip_list = []
with open('ip_list.txt', 'r') as f:
    for line in f:
        ip_list.append(line.strip())

定义查询IP归属地信息的函数：

def get_ip_info(ip):
    try:
        ip_info = ipwhois.IPWhois(ip).lookup()
        return ip_info.country, ip_info.city
    except:
        return None, None

批量查询IP归属地信息：

ip_info_list = []
for ip in ip_list:
    country, city = get_ip_info(ip)
    ip_info_list.append([ip, country, city])

自动去重：

ip_info_list = list(set(ip_info_list))

按国内外汇总：

domestic_ip_list = [ip_info for ip_info in ip_info_list if ip_info[1] == '中国']
foreign_ip_list = [ip_info for ip_info in ip_info_list if ip_info[1] != '中国']

智能识别错误IP：

error_ip_list = [ip_info[0] for ip_info in ip_info_list if ip_info[1] is None or ip_info[2] is None]

输出结果：

with open('ip_info.csv', 'w', newline='') as f:
    csv_writer = csv.writer(f)
    csv_writer.writerow(['IP', '国家', '城市'])
    for ip_info in ip_info_list:
        csv_writer.writerow(ip_info)

with open('error_ip.txt', 'w') as f:
    for error_ip in error_ip_list:
        f.write(error_ip + '\n')