企业网络Python自动化实战之企业IP地址查询器手把手教学

在维护企业网络的时候，我们需要根据IP地址来确定IP地址所属地区和区域，如果需要实现这种需求，依靠记忆是不可能实现的。我看周边很多同事大多数依赖excel表格的搜索功能来实现，但是一次性打开很多的表，不但占用很多电脑资源，效率也不高。为了提高效率，我们可以用python的办法来制作一个IP地址查询器。

一、操作前准备

1.1、在操作前，我们需要下载一个pycharm的IDE工具，我用的版本如下图所示：

1.2 本次实验我们需要用到的模块是ipaddress和pandas；大家需要在电脑上安装这两个模块。操作办法是：

在导航栏中输入pandas；ipaddress模块是内置模块，不需要安装了。

二、操作步骤

2.1 安装完成以上模块后，就可以通过import调用pandas了。以下是完整的代码：其中

INPUT_IP.txt文本是用来将我们需要确认的IP地址存放的文本；NFZX_2_IP.xls是存放网段信息的excel文件。excel文件设置如下图所示，第一列为IP地址网段；第二列为IP网段的使用属性。

import ipaddress
import pandas as pd

def read_ip_addresses(file_path):
    """读取 IP 地址文件"""
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            return [line.strip() for line in file.readlines()]
    except FileNotFoundError:
        print(f"文件 {file_path} 不存在")
        return []
    except Exception as e:
        print(f"读取文件 {file_path} 时发生错误: {e}")
        return []

def read_excel_file(file_path):
    """读取 Excel 文件中的网段和对应内容"""
    try:
        xls_file = pd.ExcelFile(file_path)
        df = xls_file.parse('Sheet1')
        return df.iloc[:, 0], df.iloc[:, 1]
    except FileNotFoundError:
        print(f"文件 {file_path} 不存在")
        return [], []
    except Exception as e:
        print(f"读取文件 {file_path} 时发生错误: {e}")
        return [], []

def check_ip_in_network(ip, network):
    """检查 IP 地址是否属于网段"""
    try:
        ip_obj = ipaddress.ip_address(ip)
        network_obj = ipaddress.ip_network(network)
        return ip_obj in network_obj
    except ValueError as e:
        print(f"IP 地址或网段格式错误: {e}")
        return False

def main():
    # 文件路径
    ip_file_path = 'INPUT_IP.txt'
    excel_file_path = 'NFZX_2_IP.xls'

    # 读取 IP 地址和 Excel 文件
    ip_addresses = read_ip_addresses(ip_file_path)
    network_column, content_column = read_excel_file(excel_file_path)

    # 检查每个 IP 是否属于任何网段，并返回对应内容
    results = []
    for ip in ip_addresses:
        for network, content in zip(network_column, content_column):
            if check_ip_in_network(ip, network):
                results.append(f"{ip}: {content}")
                break  # 一旦找到匹配的网段，就可以停止检查

    # 输出结果
    for result in results:
        print(result)

if __name__ == "__main__":
    main()