护网IP去重,去白名单

import csv
from tkinter import *
from tkinter import ttk
from tkinter.messagebox import showinfo, askyesno

# 从文件中读取IP列表并查询
def read_ip_list():
    ip_list = []
    with open('ip_list.txt', 'r') as file:
        for line in file:
            ip = line.strip()
            if not search_ip(ip):
                ip_list.append(ip)
                print(ip)
    store_ip_list(ip_list)
    showinfo("完成", "成功处理IP列表")

# 查询IP是否存在于库中
def search_ip(ip):
    with open('ip_database.csv', 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            if row[0] == ip:
                return True
    return False

# 存储IP列表到库中
def store_ip_list(ip_list):
    with open('ip_database.csv', 'a', newline='') as file:
        writer = csv.writer(file)
        for ip in ip_list:
            writer.writerow([ip])

# 单个IP查询
def search_single_ip():
    ip = entry.get()
    if search_ip(ip):
        showinfo("Result", f"IP {ip} exists in the database")
    else:
        showinfo("Result", f"IP {ip} does not exist in the database")

# 删除单个IP
def delete_single_ip():
    ip = entry.get()
    if search_ip(ip):
        confirmation = askyesno("Confirmation", f"Are you sure you want to delete IP {ip}?")
        if confirmation:
            with open('ip_database.csv', 'r') as file:
                lines = file.readlines()
            with open('ip_database.csv', 'w') as file:
                for line in lines:
                    if line.strip() != ip:
                        file.write(line)
            showinfo("Success", f"IP {ip} has been deleted from the database")
    else:
        showinfo("Error", f"IP {ip} does not exist in the database")

# 创建图形化界面
window = Tk()
window.title("IP去重工具")
window.geometry("500x400")

# 创建标题标签
title_label = ttk.Label(window, text="IP去重工具", font=("Helvetica", 16))
title_label.pack(pady=10)

# 创建按钮框架
button_frame = Frame(window)
button_frame.pack(pady=10)

read_button = ttk.Button(button_frame, text="读取IP列表", command=read_ip_list)
read_button.grid(row=0, column=1, padx=5)

# 创建输入框和按钮框架
input_frame = Frame(window)
input_frame.pack(pady=10)

entry = ttk.Entry(input_frame)
entry.grid(row=0, column=0, padx=5)

search_button = ttk.Button(input_frame, text="单个查询IP", command=search_single_ip)
search_button.grid(row=0, column=1, padx=5)

delete_button = ttk.Button(input_frame, text="删除单个IP", command=delete_single_ip)
delete_button.grid(row=0, column=2, padx=5)

# 创建状态标签
status_label = ttk.Label(window, text="")
status_label.pack(pady=10)

window.mainloop()