import os
from pathlib import Path
from weclone.utils.log import logger


def check_image_file_exists(file_path: str) -> str | bool:
    """
    检查传入的文件路径，提取文件名（去掉前缀和扩展名），
    然后检查 dataset/images 目录下是否存在对应的文件（不论扩展名）
    """
    try:
        # 直接使用 os.path.normpath 处理路径，然后转换为正斜杠
        normalized_path = os.path.normpath(file_path).replace("\\", "/")

        # 提取文件名
        filename_with_ext = os.path.basename(normalized_path)
        filename_without_ext = Path(filename_with_ext).stem

        # 使用 glob 查找精确匹配该文件名的文件（不论扩展名）
        images_dir = Path("dataset") / "media" / "images"
        matching_files = list(images_dir.glob(f"{filename_without_ext}.*"))

        if len(matching_files) > 0:
            # 获取相对于dataset/media的路径，只保留images/文件名
            full_path = matching_files[0]
            relative_path = full_path.relative_to(Path("dataset") / "media")
            return str(relative_path)
        else:
            return False

    except Exception as e:
        logger.error(f"检查图片文件时出错: {file_path}, 错误: {e}")
        return False


if __name__ == "__main__":
    path = "Storage\Image\2021-08\6ce3f785b4230246639c3dd0d4a8848c.dat"
    print(check_image_file_exists(path))