From 3910d0b6171955c0b3355dea444afcd56f657bc5 Mon Sep 17 00:00:00 2001
From: frxz751113 <156018267+frxz751113@users.noreply.github.com>
Date: Mon, 26 Aug 2024 22:03:04 +0800
Subject: [PATCH] =?UTF-8?q?Update=20=E6=94=B6=E9=9B=86.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 py/收集.py | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/py/收集.py b/py/收集.py
index e3e9c86..0fe7add 100644
--- a/py/收集.py
+++ b/py/收集.py
@@ -294,16 +294,15 @@ from tqdm import tqdm
 # 初始化2字典
 detected_ips = {}
 # 存储文件路径
-file_path = "网络收集.txt"
+file_path = "2.txt"
 output_file_path = "网络收集.txt"
 
 def get_ip_key(url):
-    """从URL中提取后面的16个字符作为IP地址键"""
-    # 找到'://'之后的位置
+    """从URL中提取IP地址，并构造一个唯一的键"""
+    # 找到'//'到第一个'/'之间的字符串
     start = url.find('://') + 3  # '://'.length 是 3
-    # 截取start位置后面16个字符
-    ip_key = url[start:start + 10]
-    return ip_key
+    end = url.find('/', start)  # 找到第一个'/'的位置
+    return url[start:end] if end != -1 else None
     
 # 打开输入文件和输出文件
 with open(file_path, 'r', encoding='utf-8') as file:
@@ -433,7 +432,7 @@ def parse_file(input_file_path, output_file_name):
 parse_file('网络收集.txt', '网络收集.txt')
 
 ################################################################################################任务结束，删除不必要的过程文件
-files_to_remove = ["2#.txt", "汇总.txt"]
+files_to_remove = ["2.txt", "汇总.txt"]
 
 for file in files_to_remove:
     if os.path.exists(file):