video/core/establish.py

import os
import datetime
from pathlib import Path


# 配置IP文件路径（统一使用绝对路径）
IP_FILE_PATH = Path(r"D:\ccc\IP.txt")


def create_directory_structure():
    """创建项目所需的目录结构"""
    try:
        # 1. 创建根目录下的resource文件夹
        resource_dir = Path("resource")
        resource_dir.mkdir(exist_ok=True)
        print(f"确保resource目录存在: {resource_dir.absolute()}")

        # 2. 在resource下创建dect文件夹
        dect_dir = resource_dir / "dect"
        dect_dir.mkdir(exist_ok=True)
        print(f"确保dect目录存在: {dect_dir.absolute()}")

        # 3. 在dect下创建三个模型文件夹
        model_dirs = ["ocr", "face", "yolo"]
        for model in model_dirs:
            model_dir = dect_dir / model
            model_dir.mkdir(exist_ok=True)
            print(f"确保{model}模型目录存在: {model_dir.absolute()}")

        # 4. 读取ip.txt文件获取IP地址
        try:
            with open(IP_FILE_PATH, "r") as f:
                ip_addresses = [line.strip() for line in f if line.strip()]

            if not ip_addresses:
                print("警告: ip.txt文件中未找到有效的IP地址")
                return

            print(f"从ip.txt中读取到的IP地址: {ip_addresses}")

            # 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

                for model in model_dirs:
                    # 构建路径: resource/dect/{model}/{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()}")

        except FileNotFoundError:
            print(f"错误: 未找到ip.txt文件，请确保该文件存在于 {IP_FILE_PATH}")
        except Exception as e:
            print(f"处理IP和日期目录时发生错误: {str(e)}")

    except Exception as e:
        print(f"创建目录结构时发生错误: {str(e)}")


def get_image_save_path(model_type: str) -> tuple:
    """
    获取图片保存的完整路径和显示用路径

    参数:
        model_type: 模型类型，应为"ocr"、"face"或"yolo"

    返回:
        元组 (完整保存路径, 显示用路径)
    """
    try:
        # 读取IP地址（假设只有一个IP或使用第一个IP）
        with open(IP_FILE_PATH, "r") as f:
            ip_addresses = [line.strip() for line in f if line.strip()]

        if not ip_addresses:
            raise ValueError("ip.txt文件中未找到有效的IP地址")

        ip = ip_addresses[0]
        safe_ip = ip  # 直接使用原始IP格式

        # 获取当前日期和时间（精确到毫秒，确保文件名唯一）
        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]  # 去除最后三位，保留到毫秒

        # 构建基础目录路径
        base_dir = Path("resource") / "dect"
        # 构建完整路径: resource/dect/{model}/{ip}/{year}/{month}/{day}
        day_dir = base_dir / model_type / safe_ip / current_year / current_month / current_day
        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

        # 计算显示用路径（相对于resource/dect的路径）
        display_path = full_path.relative_to(base_dir)

        return str(full_path), str(display_path)

    except Exception as e:
        print(f"获取图片保存路径时发生错误: {str(e)}")
        return "", ""