当前位置：首页 > article >正文

多线程并发Ping脚本

article 2024/9/22 3:17:17

1. 前言

最近需要ping地址，还是挺多的，就使用python搞一个ping脚本，记录一下，以免丢失了。

2. 脚本介绍

首先检查是否存在True.txt或False.txt文件，并在用户确认后进行删除，然后从IP.txt的文件中读取IP地址，然后使用多线程并发地对这些IP地址进行ping测试。根据测试结果，将可连接的IP地址写入True.txt文件，不可连接的IP地址写入False.txt文件。

IP.txt文件需要每一行为一个IP地址，不可放置网段，并且通信与不通信是暂存在列表中的，若中间结束ping，则数据会丢失，只有当全部运行完后，会一次性写入，之前也测试过一边测试一边写入，会出现频繁的读取文件和写入文件，总会出现IP丢失的情况。此次经过测试，无丢失情况。

测试环境：python3.10

测试结果：IP.txt中共计1.2011个IP地址，20线程，共计耗时32分钟。

2.1. 文件夹内容

文件夹中，False.txt和True.txt这两个文件是不需要创建的，程序会自动创建，而且IP.txt是需要创建的，并且放置~地址，每行一个，正常回车换行，或者在表格中弄好，直接复制过来。

每次执行只需要替换IP.txt中的IP地址就可以了。

2.2. 脚本源码

这里电脑性能好，而且想快的话，直接多上一些线程，一开始我测试的时候，线程没设定好，直接给我干了2400多线程，电脑直接卡死，预估100多线程应该都没问题。

import threading
import subprocess
import os
from concurrent.futures import ThreadPoolExecutor

"""判断是否存在True.txt或False.txt文，由于如果之前存在可能会导致执行添加了，但是测试好像不会，不过添加一个也无妨。"""
def check_confirmation():
    if os.path.exists('True.txt') or os.path.exists('False.txt'):
        user_input = input("发现已存在的True.txt或False.txt文件，是否删除？(y/n): ")
        if user_input.lower() == 'y':
            if os.path.exists('True.txt'):
                os.remove('True.txt')
            if os.path.exists('False.txt'):
                os.remove('False.txt')
        elif user_input.lower() == 'n':
            exit()
        else:
            print("无效的输入。请输入 'y' 或 'n'。")
            check_confirmation()

def ping_ip(ip, true_ips, false_ips):
    """如果IP地址能够ping通，则将其添加到true_ips列表中；否则添加到false_ips列表中。"""
    command = ['ping', '-n', '1', ip]
    result = subprocess.call(command, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
    if result == 0:
        true_ips.append(ip)
        print(f"{ip} 可以ping通!")
    else:
        false_ips.append(ip)
        print(f"{ip} 无法ping通!")

def main():
    check_confirmation()
    with open('IP.txt', 'r') as file:
        ips = file.readlines()

    true_ips = []
    false_ips = []
    """控制线程，这里我写的是20线程。"""
    with ThreadPoolExecutor(max_workers=20) as executor:
        for ip in ips:
            ip = ip.strip()
            executor.submit(ping_ip, ip, true_ips, false_ips)
    """整个程序运行完将列表中的数据写入到相应的文件夹中。"""
    with open('True.txt', 'w') as file:
        for ip in true_ips:
            file.write(ip + '\n')

    with open('False.txt', 'w') as file:
        for ip in false_ips:
            file.write(ip + '\n')

if __name__ == "__main__":
    main()