从服务器读取IP并将检测数据写入数据库

core/establish.py

@@ -2,15 +2,11 @@ import os
 import datetime
 from pathlib import Path
 
-
-# 配置IP文件路径（统一使用绝对路径）
-IP_FILE_PATH = Path(r"D:\ccc\IP.txt")
-
-
+from service.device_service import get_unique_client_ips
 def create_directory_structure():
-    """创建项目所需的目录结构"""
+    """创建项目所需的目录结构，为所有客户端IP预创建基础目录"""
     try:
-        # 1. 创建根目录下的resource文件夹
+        # 1. 创建根目录下的resource文件夹（存在则跳过，不覆盖子内容）
         resource_dir = Path("resource")
         resource_dir.mkdir(exist_ok=True)
         print(f"确保resource目录存在: {resource_dir.absolute()}")
@@ -27,87 +23,95 @@ def create_directory_structure():
             model_dir.mkdir(exist_ok=True)
             print(f"确保{model}模型目录存在: {model_dir.absolute()}")
 
-        # 4. 读取ip.txt文件获取IP地址
+        # 4. 调用外部方法获取所有客户端IP地址
         try:
-            with open(IP_FILE_PATH, "r") as f:
-                ip_addresses = [line.strip() for line in f if line.strip()]
+            # 调用外部ip_read()方法获取所有客户端IP地址列表
+            all_ip_addresses = get_unique_client_ips()
 
-            if not ip_addresses:
-                print("警告: ip.txt文件中未找到有效的IP地址")
+            # 确保返回的是列表类型
+            if not isinstance(all_ip_addresses, list):
+                all_ip_addresses = [all_ip_addresses]
+
+            # 过滤有效IP（去除空字符串和空格）
+            valid_ips = [ip.strip() for ip in all_ip_addresses if ip.strip()]
+
+            if not valid_ips:
+                print("警告: 未获取到有效的客户端IP地址")
                 return
 
-            print(f"从ip.txt中读取到的IP地址: {ip_addresses}")
+            print(f"获取到的所有客户端IP地址: {valid_ips}")
 
-            # 5. 获取当前日期
+            # 5. 获取当前日期（年、月）
             now = datetime.datetime.now()
             current_year = str(now.year)
             current_month = str(now.month)
 
-            # 6. 为每个IP在每个模型文件夹下创建年->月的目录结构
-            for ip in ip_addresses:
-                # 直接使用原始IP格式
-                safe_ip = ip
+            # 6. 为每个客户端IP在每个模型文件夹下创建年->月的基础目录结构
+            for ip in valid_ips:
+                # 处理IP地址中的特殊字符（将.替换为_，避免路径问题）
+                safe_ip = ip.replace(".", "_")
 
                 for model in model_dirs:
-                    # 构建路径: resource/dect/{model}/{ip}/{year}/{month}
+                    # 构建路径: resource/dect/{model}/{safe_ip}/{year}/{month}
                     ip_dir = dect_dir / model / safe_ip
                     year_dir = ip_dir / current_year
                     month_dir = year_dir / current_month
 
-                    # 创建目录（如果不存在）
+                    # 递归创建目录（存在则跳过，不覆盖）
                     month_dir.mkdir(parents=True, exist_ok=True)
-                    print(f"创建/确保目录存在: {month_dir.absolute()}")
+                    print(f"为客户端IP {ip} 创建/确保目录存在: {month_dir.absolute()}")
 
-        except FileNotFoundError:
-            print(f"错误: 未找到ip.txt文件，请确保该文件存在于 {IP_FILE_PATH}")
         except Exception as e:
-            print(f"处理IP和日期目录时发生错误: {str(e)}")
+            print(f"处理客户端IP和日期目录时发生错误: {str(e)}")
 
     except Exception as e:
-        print(f"创建目录结构时发生错误: {str(e)}")
+        print(f"创建基础目录结构时发生错误: {str(e)}")
 
 
-def get_image_save_path(model_type: str) -> tuple:
+def get_image_save_path(model_type: str, client_ip: str) -> tuple:
     """
-    获取图片保存的完整路径和显示用路径
+    获取图片保存的「完整路径」和「显示用短路径」
 
     参数:
         model_type: 模型类型，应为"ocr"、"face"或"yolo"
+        client_ip: 检测到违禁的客户端IP地址（原始格式，如192.168.1.101）
 
     返回:
-        元组 (完整保存路径, 显示用路径)
+        元组 (完整保存路径, 显示用短路径)；若出错则返回 ("", "")
     """
     try:
-        # 读取IP地址（假设只有一个IP或使用第一个IP）
-        with open(IP_FILE_PATH, "r") as f:
-            ip_addresses = [line.strip() for line in f if line.strip()]
+        # 1. 验证客户端IP有效性（检查是否在已知IP列表中）
+        all_ip_addresses = get_unique_client_ips()
+        if not isinstance(all_ip_addresses, list):
+            all_ip_addresses = [all_ip_addresses]
+        valid_ips = [ip.strip() for ip in all_ip_addresses if ip.strip()]
 
-        if not ip_addresses:
-            raise ValueError("ip.txt文件中未找到有效的IP地址")
+        if client_ip.strip() not in valid_ips:
+            raise ValueError(f"客户端IP {client_ip} 不在已知IP列表中，无法保存文件")
 
-        ip = ip_addresses[0]
-        safe_ip = ip  # 直接使用原始IP格式
+        # 2. 处理IP地址（与目录创建逻辑一致，将.替换为_）
+        safe_ip = client_ip.strip().replace(".", "_")
 
-        # 获取当前日期和时间（精确到毫秒，确保文件名唯一）
+        # 3. 获取当前日期和毫秒级时间戳（确保文件名唯一）
         now = datetime.datetime.now()
         current_year = str(now.year)
         current_month = str(now.month)
         current_day = str(now.day)
-        # 生成时间戳字符串（格式：年月日时分秒毫秒）
-        timestamp = now.strftime("%Y%m%d%H%M%S%f")[:-3]  # 去除最后三位，保留到毫秒
+        # 时间戳格式：年月日时分秒毫秒（如20250910143050123）
+        timestamp = now.strftime("%Y%m%d%H%M%S%f")[:-3]
 
-        # 构建基础目录路径
-        base_dir = Path("resource") / "dect"
-        # 构建完整路径: resource/dect/{model}/{ip}/{year}/{month}/{day}
+        # 4. 定义基础目录（用于生成相对路径）
+        base_dir = Path("resource") / "dect"  # 显示路径会去掉这个前缀
+        # 构建日级目录（完整路径：resource/dect/{model}/{safe_ip}/{年}/{月}/{日}）
         day_dir = base_dir / model_type / safe_ip / current_year / current_month / current_day
-        day_dir.mkdir(parents=True, exist_ok=True)
+        day_dir.mkdir(parents=True, exist_ok=True)  # 确保日目录存在
 
-        # 构建图片文件名（简化名称，去掉resource_dect_前缀）
-        image_filename = f"{model_type}_{safe_ip}_{current_year}_{current_month}_{current_day}_{timestamp}.jpg"
-        full_path = day_dir / image_filename
+        # 5. 构建唯一文件名
+        image_filename = f"dect_{model_type}_{safe_ip}_{current_year}{current_month}{current_day}_{timestamp}.jpg"
 
-        # 计算显示用路径（相对于resource/dect的路径）
-        display_path = full_path.relative_to(base_dir)
+        # 6. 生成完整路径（用于实际保存图片）和显示路径（用于打印）
+        full_path = day_dir / image_filename  # 完整路径：resource/dect/.../xxx.jpg
+        display_path = full_path.relative_to(base_dir)  # 短路径：{model}/.../xxx.jpg（去掉resource/dect）
 
         return str(full_path), str(display_path)