import json from flask_app.货物标.技术参数要求提取后处理函数 import postprocess_technical_table import re from collections import defaultdict #12.27之前版本 # def extract_matching_keys(data, good_list, special_keys=None, parent_key=''): # def get_suffix(n): # """ # 根据数字n返回对应的字母后缀。 # 1 -> 'a', 2 -> 'b', ..., 26 -> 'z', 27 -> 'aa', 28 -> 'ab', ... # """ # suffix = '' # while n > 0: # n, r = divmod(n - 1, 26) # suffix = chr(97 + r) + suffix # return suffix # # def count_matching_keys(data, patterns, special_keys, counter=None): # """递归统计匹配键的出现次数，仅统计值为列表的键""" # if counter is None: # counter = defaultdict(int) # # if isinstance(data, dict): # for key, value in data.items(): # clean_key = key.replace(" ", "") # 去除键中的空格 # if isinstance(value, list): # if clean_key not in special_keys and any(pattern.match(clean_key) for pattern in patterns): # counter[clean_key] += 1 # elif isinstance(value, dict): # count_matching_keys(value, patterns, special_keys, counter) # elif isinstance(data, list): # for item in data: # if isinstance(item, (dict, list)): # count_matching_keys(item, patterns, special_keys, counter) # # return counter # # def process_data(data, patterns, special_keys, key_counter, suffix_map, filtered_data, parent_key): # """递归处理数据并构建结果""" # # def get_suffix_label(key): # suffix_map[key] += 1 # return get_suffix(suffix_map[key]) # # if isinstance(data, dict): # for key, value in data.items(): # clean_key = key.replace(" ", "") # 去除键中的空格 # if isinstance(value, list): # # 处理值为列表的键 # if any(pattern.match(clean_key) for pattern in patterns): # # 检查是否以特殊符号开头 # if clean_key.startswith(('▲', '★','●','■','◆','☆','△','◇','○','□')): # symbol = clean_key[0] # stripped_key = clean_key[1:] # new_key = generate_key(stripped_key, parent_key, key_counter, suffix_map, special_keys) # # 将符号添加到每个字符串的开头 # new_value = [symbol + item for item in value] # filtered_data[new_key] = new_value # else: # new_key = generate_key(clean_key, parent_key, key_counter, suffix_map, special_keys) # filtered_data[new_key] = value # elif isinstance(value, dict): # # 继续递归处理嵌套字典 # new_parent_key = clean_key if parent_key == '' else f"{parent_key}的{clean_key}" # process_data(value, patterns, special_keys, key_counter, suffix_map, # filtered_data, new_parent_key) # elif isinstance(data, list): # for item in data: # if isinstance(item, (dict, list)): # process_data(item, patterns, special_keys, key_counter, suffix_map, # filtered_data, parent_key) # # def generate_key(key, parent_key, key_counter, suffix_map, special_keys): # """生成新的键名""" # if key in special_keys and parent_key: # return f"{parent_key}的{key}" # elif key_counter[key] > 1: # suffix = get_suffix(suffix_map[key] + 1) # suffix_map[key] += 1 # return f"{key}-{suffix}" # return key # # if special_keys is None: # special_keys = ["系统功能"] # 默认值为 ["系统功能"] # # # 去除 good_list 中的空格 # clean_good_list = [g.replace(" ", "") for g in good_list] # # # 构建匹配的正则表达式 # patterns = [re.compile(r'^' + re.escape(g) + r'(?:-\d+)?$') for g in clean_good_list] # # # 先统计所有匹配键的出现次数，仅统计值为列表的键 # key_counter = count_matching_keys(data, patterns, special_keys) # # # 初始化后缀映射 # suffix_map = {key: 0 for key, count in key_counter.items() if count > 1} # # # 用于存储最终结果 # filtered_data = {} # # # 递归处理数据 # process_data(data, patterns, special_keys, key_counter, suffix_map, filtered_data, parent_key) # # return filtered_data def test_extract_matching_keys(): # 定义测试数据 data = { "采购需求": { "显示系统": { "系统功能":["1","2"], "★LED全彩显示屏": [ "1、显示尺寸：6m±0.3（W）× 1.5m±0.2(H),单屏分辨率≥3744 × 1040；", "2 、像素间距≤1.53mm；", "3 、亮度≥450nits， 色温： 3000-15000K 可调， 对比度： 5000:1；", "4 、峰值功耗≤440W， 平均功耗≤146W， 带有智能（黑屏） 节电功 能， 开启智能节电功能比没开启节能 40%以上；", "5 、水平可视角度≥160 ° , 垂直可视角度≥140 ° ;", "6 、亮度均匀性≥97%， 刷新率≥3840 Hz ， 发光点中心距偏差＜3%；", "7 、色域覆盖率≥100%， 低亮高灰： 100%亮度时， 16 bits 灰度，20% 亮度时， 12bits 灰度；", "8 、铝底壳材质， 无风扇散热结构；", "9 、模组电源接口采用4P 接插头， 免工具维护， 同时有防呆设计， 预防接错电源线短路而导致的烧毁模组行为，采用集成 HUB 接收卡 控制， 支持通讯状态监控；", "10 、冗余备份， 支持双电网供电， 当其中一路交流电网跳闸后， 另 外一路电网继续供电， 实现不间断供电， 支持热备份， 当其中一块 电源失效后， 另一块电源继续工作， 从而实现不间断供电；", "11 、屏体发光模组采用 4.5 VDC 的安全电压供电；", "12 、彩色信号处理位数≥16bit；", "13 、具备故障自诊及排查功能；", "14 、 图像有降噪 、增强 、运动补偿 、色坐标变换处理 、钝 化处理无 几何失真和线性失真现象 、消除鬼影拖尾， 无“毛毛虫 ”“鬼影 ” 跟随现象；", "15 、防护等级符合 IP6X， 显示屏具有防潮 、防尘 、防腐蚀 、防电磁 干扰 、防静电等功能， 并具有过流、短路 、过压 、欠压保护等功能；", "16 、工作噪音声压等级一米处≤7.8 dB (A),盐雾符合 10 级要求， PCB 阻燃等级达到 UL 94 V-0 级要求， 通过 9 级烈度地震模拟实验。" ], "☆钢结构底座及铝型材支架": [ "1 、主体钢架结构及定制型材；", "2 、确保楼层承受力许可,按需加固楼层地面；", "3 、钢结构。" ] }, "摄像系统": { "系统功能": ["a", "b"], "☆钢结构底座及铝型材支架": [ "1 、主体钢架结构及定制型材；wwww", "2 、确保楼层承受力许可,按需加固楼层地面；", "3 、钢结构。" ] }, "视频处理系统": { "★高清视频拼控矩阵(16*16)": [ "1 、8U 切换主箱体， 支持输入 13 槽， 输出 4.5 槽， 支持 8 路高分采集， 支持冗余电源， 标配 1 个电源模块； 本项目配置输入接 口 16 路和 1 张字幕卡， 输出接口 16 路；", "2 、设备应为纯硬件 FPGA 架构， CrossPoint 全总线交换技术， 背板 等效带宽；", "3 、单张板卡支持 4 通道输入或输出， 紧凑型机箱,模拟视频单板卡 支持 16 路同时输入， 单卡支持 2 种信号源任意组合；", "4、输入输出板卡可热插拔，输入板卡热插拔恢复时间 ＜2s，输出板 卡热插拔恢复时间＜8s；", "5 、开机时间≤10s， 启动电源至输出最总画面的时间间隔；", "6 、平均故障时间间隔 （ MTBF ） 不小于 96000 小时， 保证设备能够 稳定运行；", "7 、最大单机背板信号处理带宽不小于 720Gbps；", "8、对各个输入通道采用纯硬件处理技术，采用独享带宽方式为每个 输入通道分配带宽， 切换过程中对其他信号无影响， 实现了对输入 通道的实时处理；", "9、支持集中采集 DVI、VGA、CVBS、HDBaseT、HDMI、SDI、YPbPr 、 光纤等 2K 信号，Dual-link DVI、HDMI 1.4、DisplayPort 等 4K 信号；", "10 、支持 DVI 、HDBaseT 、HDMI 、SDI 、光纤 、CVBS 、Ypbp r 等常见的 2K 信号输出， Dual-link DVI 、HDMI 1.4 等 4K 信号输出；", "11、设备可实现任意一路画面的任意比例缩放、漫游、 跨屏 、叠加、 开窗；", "12 、设备支持图像无缝实时切换功能， 无缝切换时间＜20 ms ；", "13 、支持场景保存及快速调用， 支持场景轮巡， 适应于不同的应用 场景；", "14 、支持信号源预监功能， 支持浏览所有输入信号源的实时预览画 面；", "15 、支持大屏图像回显， 可显示整面拼接墙的显示图像；", "16 、支持设置拼接屏的拼缝补偿， 可精确到 1 个像素；", "17 、支持 RRTA 分辨率实时全兼容技术， 单台 设备应支持同时控制 4 组不同分辨率的大屏幕显示；", "18 、设备具备静态底图功能， 设备支持超大分辨率底图显示， 横纵 分辨率最大 65535 像素。" ], "分量信号接口器": [ "用于现有视频会议专业对接高清矩阵接口器" ], "高清四画面分割器": [ "画面预览使用， 具有画中画 、独立单画面放大功能。" ] }, } } good_list = [ "★LED全彩显示屏", "控制盒及电源", "大屏播控系统", "配电柜(含PLC)", "☆钢结构底座及铝型材支架", "电缆及信号线缆", "控制终端", "50寸液晶电视机", "50寸电视机地面推车", "高清监视器", "★高清摄像机", "摄像机三脚架", "摄像机壁装架", "摄像机控制键盘", "★高清视频拼控矩阵(16*16)", "分量信号接口器", "高清四画面分割器", "数字会议发言主机", "方形短杆代表话筒", "专用连接线缆", "手持无线话筒", "▲多点控制器", "★多串口控制服务器", "★综合会议管理调度平台", "★高清会议终端(主会场)", "★高清会议终端(分会场)", "65寸电视机移动推车(9楼)", "65寸液晶电视机(分会场)", "控制平板及软件", "鹅颈话筒", "时序电源", "多媒体地插盒", "线材辅料", "墙体拆除及修复", "系统功能" ] special_keys = ["系统功能", "dairy"] # 假设 'dairy' 也是特殊键 # 注意：根据您的函数逻辑，特殊键会被排除，且嵌套的 'fruits' 会被处理 # 这里我们需要根据函数实际行为调整预期输出 # 让我们根据函数逻辑重新定义预期输出 # 函数会生成新的键名，对于重复的 'fruits' 会添加后缀 # 'grains' 内的 'whole' 和 'whole-1' 也会被处理为 'whole-a', 'whole-b' # 'chips' 和 'chips-1' 会被处理为 'chips-a', 'chips-b' # 'dairy' 是特殊键，应被排除 # '系统功能' 和 '系统 功能' 是特殊键，应被排除 # 运行函数 result = postprocess_technical_table(data, good_list, special_keys) # 打印结果 print("测试用例: 提取匹配键并处理各种情况") print("good_list:", good_list) print("special_keys:", special_keys) print("实际输出:", json.dumps(result,ensure_ascii=False,indent=4)) # 运行测试 if __name__ == "__main__": test_extract_matching_keys()