key_utils.py

import os
import posixpath


def strip_key_metadata(keys):
    """移除 all_keys.json 中以下划线开头的元数据字段，返回新 dict。"""
    return {k: v for k, v in keys.items() if not k.startswith("_")}


def _is_safe_rel_path(path):
    """检查路径不包含 .. 等遍历组件。"""
    normalized = path.replace("\\", "/")
    return ".." not in posixpath.normpath(normalized).split("/")


def key_path_variants(rel_path):
    """生成同一路径的多种分隔符表示，兼容 Windows/Linux JSON key。"""
    normalized = rel_path.replace("\\", "/")
    variants = []
    for candidate in (
        rel_path,
        normalized,
        normalized.replace("/", "\\"),
        normalized.replace("/", os.sep),
    ):
        if candidate not in variants:
            variants.append(candidate)
    return variants


def get_key_info(keys, rel_path):
    """按相对路径查找数据库密钥，自动兼容不同平台分隔符。"""
    if not _is_safe_rel_path(rel_path):
        return None
    for candidate in key_path_variants(rel_path):
        if candidate in keys and not candidate.startswith("_"):
            return keys[candidate]
    return None