text_widget.delete(1.0, tk.END)
text_widget.insert(tk.END, pretty_xml)
text_widget.config(state=tk.DISABLED)
# 高亮空ID节点（可选） self.highlight_empty_id_nodes(text_widget, pretty_xml) except Exception as e: messagebox.showerror("显示错误", f"格式化XML显示失败: {str(e)}") def highlight_empty_id_nodes(self, text_widget, xml_content): """高亮显示id为空的data节点""" if not xml_content: return # 查找所有id属性为空的data节点 pattern = r'(<data\b.*?\bid="")' matches = re.finditer(pattern, xml_content) # 设置高亮样式 text_widget.tag_configure("empty_id", background="#ffe6e6") # 应用高亮 for match in matches: start_index = f"1.0 + {match.start()} chars" end_index = f"1.0 + {match.end()} chars" text_widget.tag_add("empty_id", start_index, end_index) def init_data_structures(self): """初始化数据结构""" self.rawdata_template = [] self.rawdata_source = None self.event_template = [] self.event_source = None self.current_filter = "" self.variable_mapping = {} self.module_mapping = {} self.init_module() self.event_categories = [] self.special_mapping_id = {} self.special_mapping_name = {} self.init_special_mapping() def init_module(self): """初始化moduleID与SVEC模板中一致""" self.module_mapping["System"] = "00" self.module_mapping["CryoPump"] = "00" self.module_mapping["HeatExchanger"] = "00" self.module_mapping["EFEM"] = "01" self.module_mapping["LP1"] = "01" self.module_mapping["LP2"] = "01" self.module_mapping["LP3"] = "01" self.module_mapping["Transfer"] = "02" self.module_mapping["TC"] = "02" self.module_mapping["Buffer"] = "03" self.module_mapping["BF"] = "03" self.module_mapping["Bufferr"] = "03" self.module_mapping["LA"] = "11" self.module_mapping["LB"] = "12" self.module_mapping["LC"] = "13" self.module_mapping["LD"] = "14" self.module_mapping["LAB"] = "15" self.module_mapping["LCD"] = "16" self.module_mapping["Ch1"] = "21" self.module_mapping["Ch2"] = "22" self.module_mapping["Ch3"] = "23" self.module_mapping["Ch4"] = "24" self.module_mapping["Ch5"] = "25" self.module_mapping["Ch6"] = "26" self.module_mapping["ChA"] = "27" self.module_mapping["ChB"] = "28" self.module_mapping["ChC"] = "29" self.module_mapping["ChD"] = "30" self.module_mapping["ChE"] = "31" self.module_mapping["ChF"] = "32" def get_module_id(self, module_name): """根据module_name和映射表获取id""" if module_name not in self.module_mapping: return None else: return self.module_mapping[module_name] # ================ RawdataConfig 功能方法 ================ def load_rawdata_template(self): """加载Rawdata模板文件(.txt)""" file_path = filedialog.askopenfilename( title="选择Rawdata模板文件", filetypes=[("Text files", "*.txt"), ("All files", "*.*")] ) if not file_path: return try: with open(file_path, 'r', encoding='utf-8') as file: lines = file.readlines() self.rawdata_template = [] self.variable_mapping = {} # 清空表格 for item in self.template_table.get_children(): self.template_table.delete(item) # 解析模板文件 for line in lines: line = line.strip() if not line: continue # 分割行数据 parts = line.split() if len(parts) < 2: continue # 提取数据 variable = parts[0] id = parts[1] if len(parts) > 1 else "" mapping_info = parts[2] if len(parts) > 2 else None self.rawdata_template.append({ "variable": variable, "id": id, "mapping": mapping_info }) if mapping_info: mapping_parts = mapping_info.split(';') for map in mapping_parts: if map in self.variable_mapping: messagebox.showwarning("信息有误","请确认模板文件中mapping列是否有重复信息"+map) return else: self.variable_mapping[map] = variable # 添加到表格 self.template_table.insert("", tk.END, values=(variable, id, mapping_info)) self.status_var.set(f"已加载Rawdata模板: {os.path.basename(file_path)} - {len(self.rawdata_template)} 条记录") except Exception as e: messagebox.showerror("加载错误", f"加载Rawdata模板文件失败: {str(e)}") def load_rawdata_source(self): """加载Rawdata源文件(.xml)""" file_path = filedialog.askopenfilename( title="选择Rawdata源文件", filetypes=[("XML files", "*.xml"), ("All files", "*.*")] ) if not file_path: return try: self.rawdata_tree = ET.parse(file_path) self.rawdata_root = self.rawdata_tree.getroot() self.rawdata_source = self.rawdata_tree # 在文本框中显示XML内容 self.display_xml_content(self.rawdata_tree, self.rawdata_xml_text) self.status_var.set(f"已加载Rawdata源文件: {os.path.basename(file_path)}") except Exception as e: messagebox.showerror("加载错误", f"加载Rawdata源文件失败: {str(e)}") def add_id_to_rawdata(self): """为Rawdata源文件添加ID属性""" if not self.rawdata_source: messagebox.showwarning("无数据", "请先加载Rawdata源文件") return # 添加ID属性 count = 0 for data in self.rawdata_root.findall('.//data'): if 'id' not in data.attrib: # 在属性开头位置添加id属性 new_attrib = OrderedDict() new_attrib['id'] = '' for key, value in data.attrib.items(): new_attrib[key] = value # 更新属性 data.attrib.clear() for key, value in new_attrib.items(): data.set(key, value) count += 1 # 更新XML显示 self.display_xml_content(self.rawdata_tree, self.rawdata_xml_text) self.status_var.set(f"已为{count}个data标签添加id属性") def filter_template_vars(self): """筛选模板Variable""" if not self.rawdata_template: messagebox.showwarning("无数据", "请先加载Rawdata模板文件") return # 提取Variable后半部分并去重 var_parts = set() for item in self.rawdata_template: variable = item["variable"] parts = variable.split('_', 1) # 只分割第一个下划线 if len(parts) > 1: var_parts.add(parts[1]) # 更新列表 self.template_filter_var.set(tuple(sorted(var_parts))) self.status_var.set(f"已筛选出 {len(var_parts)} 个模板Variable") def filter_source_vars(self): """筛选源文件Variable""" if not self.rawdata_source: messagebox.showwarning("无数据", "请先加载Rawdata源文件") return # 提取Variable后半部分并去重 var_parts = set() for data in self.rawdata_root.findall('.//data'): variable = data.get('variable', '') if variable: parts = variable.split('_', 1) # 只分割第一个下划线 if len(parts) > 1: var_parts.add(parts[1]) # 更新列表 self.source_filter_var.set(tuple(sorted(var_parts))) self.status_var.set(f"已筛选出 {len(var_parts)} 个源文件Variable") def compare_vars(self): """对比模板和源文件Variable""" template_vars = self.template_filter_box.get(0, tk.END) source_vars = self.source_filter_box.get(0, tk.END) template_mapping_vars = self.variable_mapping.values() if not template_vars or not source_vars: messagebox.showwarning("无数据", "请先筛选模板和源文件Variable") return # 清空对比结果 self.compare_result.config(state=tk.NORMAL) self.compare_result.delete(1.0, tk.END) # 查找源文件中有但模板中没有的Variable missing_in_template = set(source_vars) - set(template_vars) - set(template_mapping_vars) # 更新源文件列表，标红缺失项 self.source_filter_box.delete(0, tk.END) for var in sorted(source_vars): self.source_filter_box.insert(tk.END, var) if var in missing_in_template: self.source_filter_box.itemconfig(tk.END, fg='red') # 显示对比结果 self.compare_result.insert(tk.END, "对比结果:\n") self.compare_result.insert(tk.END, f"模板Variable数量: {len(template_vars)}\n") self.compare_result.insert(tk.END, f"源文件Variable数量: {len(source_vars)}\n") self.compare_result.insert(tk.END, f"源文件中有但模板中缺失的数量: {len(missing_in_template)}\n\n") if missing_in_template: self.compare_result.insert(tk.END, "缺失的Variable:\n") for var in sorted(missing_in_template): self.compare_result.insert(tk.END, f" - {var}\n") self.compare_result.config(state=tk.DISABLED) self.status_var.set(f"对比完成: 发现 {len(missing_in_template)} 个缺失项") def export_highlighted(self): """导出标红内容""" # 获取标红的项 highlighted_items = [] for i in range(self.source_filter_box.size()): if self.source_filter_box.itemcget(i, "fg") == "red": highlighted_items.append(self.source_filter_box.get(i)) if not highlighted_items: messagebox.showinfo("无数据", "没有标红的项可导出") return # 选择保存位置 file_path = filedialog.asksaveasfilename( title="保存标红内容", defaultextension=".txt", filetypes=[("Text files", "*.txt"), ("All files", "*.*")] ) if not file_path: return # 写入文件 try: with open(file_path, 'w', encoding='utf-8') as file: file.write("标红的Variable项:\n") for item in highlighted_items: file.write(f"{item}\n") self.status_var.set(f"已导出 {len(highlighted_items)} 个标红项到 {os.path.basename(file_path)}") messagebox.showinfo("导出成功", "标红项已成功导出") except Exception as e: messagebox.showerror("导出错误", f"导出文件失败: {str(e)}") def fill_ids(self): """填充ID属性""" if not self.rawdata_source: messagebox.showwarning("无数据", "请先加载Rawdata源文件") return if not self.rawdata_root: messagebox.showwarning("无数据", "请先加载Rawdata源文件") return if not self.rawdata_template: messagebox.showwarning("无数据", "请先加载模板文件") return fill_count = 0 not_exist = [] not_exist_all = [] for data_elem in self.rawdata_tree.findall('.//data'): variable = data_elem.get('variable') parts = variable.split('_', 1) if len(parts) < 2 or variable in self.special_mapping_id: if variable in self.special_mapping_id and variable in self.special_mapping_name: temp_var = self.special_mapping_name[variable] temp_id = self.special_mapping_id[variable] else: continue else: module = parts[0] var = parts[1] temp_var, temp_id = self.get_id(module, var) if temp_var and temp_id: data_elem.set('variable', temp_var) data_elem.set('id', temp_id) fill_count += 1 else: if temp_id is None and var not in not_exist: not_exist.append(var) if temp_id is None: not_exist_all.append(variable) # 更新XML显示 self.display_xml_content(self.rawdata_tree, self.rawdata_xml_text) self.status_var.set(f"已为{fill_count}个data标签添加id属性，剩余{len(not_exist)}无对应值。总共剩余{len(not_exist_all)}个未设置") def filter_empty_id_nodes(self): """删除id属性为空的data节点并刷新显示""" if not self.rawdata_source: messagebox.showwarning("无数据", "请先加载Rawdata源文件") return # 获取XML根节点 root = self.rawdata_tree.getroot() # 查找所有id属性为空的data节点 empty_id_nodes = [] for data_node in root.iter('data'): if data_node.get('id', '') == '': empty_id_nodes.append(data_node) if not empty_id_nodes: messagebox.showinfo("无操作", "未找到id为空的data节点") return # 删除这些节点（使用替代方法获取父节点） for node in empty_id_nodes: for parent in root.iter(): if node in list(parent): # 检查节点是否是当前父节点的直接子节点 parent.remove(node) break # 刷新XML显示 self.display_xml_content(self.rawdata_tree, self.rawdata_xml_text) # 更新状态 self.status_var.set(f"已删除 {len(empty_id_nodes)} 个id为空的data节点") def export_rawdata(self): """导出Rawdata源文件""" if not self.rawdata_source: messagebox.showwarning("无数据", "没有可导出的数据") return file_path = filedialog.asksaveasfilename( title="保存Rawdata文件", defaultextension=".xml", filetypes=[("XML files", "*.xml"), ("All files", "*.*")] ) if not file_path: return try: self.rawdata_tree.write(file_path, encoding='utf-8', xml_declaration=True) self.status_var.set(f"Rawdata文件已成功导出到: {os.path.basename(file_path)}") messagebox.showinfo("导出成功", "Rawdata配置文件已成功导出") except Exception as e: messagebox.showerror("导出错误", f"导出文件失败: {str(e)}") def init_special_mapping(self): """初始化特殊映射""" self.special_mapping_id['LP1_BypassReadID'] = '201210000' self.special_mapping_name['LP1_BypassReadID'] = 'EFEM_LP1BypassReadID' self.special_mapping_id['LP1_LoadPortTout'] = '201210001' self.special_mapping_name['LP1_LoadPortTout'] = 'EFEM_LP1LoadPortTout' self.special_mapping_id['LP2_BypassReadID'] = '201211000' self.special_mapping_name['LP2_BypassReadID'] = 'EFEM_LP2BypassReadID' self.special_mapping_id['LP2_LoadPortTout'] = '201211001' self.special_mapping_name['LP2_LoadPortTout'] = 'EFEM_LP2LoadPortTout' self.special_mapping_id['LP3_BypassReadID'] = '201212000' self.special_mapping_name['LP3_BypassReadID'] = 'EFEM_LP3BypassReadID' self.special_mapping_id['LP3_LoadPortTout'] = '201212001' self.special_mapping_name['LP3_LoadPortTout'] = 'EFEM_LP3LoadPortTout' self.special_mapping_id['CJobSubstProcStatusList'] = '100000004' self.special_mapping_name['CJobSubstProcStatusList'] = 'CJobSubstProcStatusList' self.special_mapping_id['ToolState'] = '100000000' self.special_mapping_name['ToolState'] = 'ToolState' self.special_mapping_id['Aligner_AngleLA'] = '201209000' self.special_mapping_name['Aligner_AngleLA'] = 'EFEM_AlignerAngleLA' self.special_mapping_id['Aligner_AngleLB'] = '201209001' self.special_mapping_name['Aligner_AngleLB'] = 'EFEM_AlignerAngleLB' self.special_mapping_id['Aligner_AngleX'] = '201209002' self.special_mapping_name['Aligner_AngleX'] = 'EFEM_AlignerAngleX' self.special_mapping_id['Aligner_NotchSupport'] = '201209003' self.special_mapping_name['Aligner_NotchSupport'] = 'EFEM_AlignerNotchSupport' self.special_mapping_id['Aligner_Bypass'] = '201209004' self.special_mapping_name['Aligner_Bypass'] = 'EFEM_AlignerBypass' self.special_mapping_id['CryoPump_CompressorPressure'] = '101122000' self.special_mapping_name['CryoPump_CompressorPressure'] = 'System_CryoPumpCompressorPressure' self.special_mapping_id['CryoPump_CompressorDiffPressure'] = '101122001' self.special_mapping_name['CryoPump_CompressorDiffPressure'] = 'System_CryoPumpCompressorDiffPressure' self.special_mapping_id['CryoPump_CompressorSupplyPressure'] = '101122002' self.special_mapping_name['CryoPump_CompressorSupplyPressure'] = 'System_CryoPumpCompressorSupplyPressure' self.special_mapping_id['HeatExchanger_RS'] = '100409000' self.special_mapping_name['HeatExchanger_RS'] = 'System_HeatExchangerRS' def get_id(self, module_name, var): """获取ID""" if var in self.variable_mapping: #来自mapping，含有前缀CHXXX temp_var = self.variable_mapping[var] variable = temp_var.replace('CHXXX', module_name) # Ch1_LotID else: temp_var = "CHXXX_" + var #来自配置，不含有前缀CHXXX variable = module_name + '_' + var # Ch1_LotID module_id = self.get_module_id(module_name) temp_id = self.get_id_by_variable(temp_var) if temp_id and module_id: if 'XX' in temp_id: id = temp_id.replace('XX', str(module_id)) return variable.replace('CHXXX', module_name), id else: return None, None def get_id_by_variable(self, variable): """根据变量名获取ID""" for item in self.rawdata_template: if item['variable'] == variable: return item['id'] return None def split_export(self): """拆分并导出文件""" if not self.rawdata_source: messagebox.showwarning("无数据", "没有可拆分的数据") return # 创建分组字典，前缀为data节点列表 groups = defaultdict(list) # 遍历所有data节点 for data in self.rawdata_root.findall('.//data'): variable = data.get('variable','') if '_' in variable: prefix = variable.split('_', 1)[0] groups[prefix].append(data) # 为每个分组创建新的xml文件 for prefix, data in groups.items(): # 创建新根节点（复制原始根节点标签和属性） new_root = ET.Element(self.rawdata_root.tag, attrib=self.rawdata_root.attrib) # 复制原始根节点的命名空间 for ns, uri in self.rawdata_root.nsmap.items(): if ns: ET.register_namespace(ns, uri) # 添加原始非 data 节点 for child in self.rawdata_root: if child.tag != 'data': new_root.append(child) # 添加分组中的 data 节点 for node in data: new_root.append(node) # 创建 XML 树并写入文件 new_tree = ET.ElementTree(new_root) filename = f"RawData_{prefix}.xml" new_tree.write(filename, encoding='utf-8', xml_declaration=True) messagebox.showinfo("成功", f"已拆分为 {len(groups)} 个文件") # ================ EventConfig 功能方法 ================ def load_event_template(self): """加载Event模板文件(.txt)""" file_path = filedialog.askopenfilename( title="选择Event模板文件", filetypes=[("Text files", "*.txt"), ("All files", "*.*")] ) if not file_path: return try: with open(file_path, 'r', encoding='utf-8') as file: lines = file.readlines() self.event_template = [] self.event_mapping = {} # 清空表格 for item in self.event_template_table.get_children(): self.event_template_table.delete(item) # 解析模板文件 for line in lines: line = line.strip() if not line: continue # 分割行数据 parts = line.split() if len(parts) < 2: continue # 提取数据 name = parts[0] id = parts[1] if len(parts) > 1 else "" mapping = parts[2] if len(parts) > 2 else None # 提取type name_parts = name.split('_',1) if len(name_parts) < 2: continue type = name_parts[1] self.event_categories.append(type) if mapping: mapping_parts = mapping.split(';') for map in mapping_parts: if map in self.event_mapping: messagebox.showwarning("数据有误","请确认Event模板数据中mapping列是否有重复数据") else: self.event_mapping[map] = name self.event_template.append({ "name": name, "id": id, "type": type, "mapping": mapping }) # 添加到表格 self.event_template_table.insert("", tk.END, values=(name, id, type, mapping)) self.event_categories = sorted(self.event_categories, key=len, reverse=True) self.status_var.set(f"已加载Event模板: {os.path.basename(file_path)} - {len(self.event_template)} 条记录") except Exception as e: messagebox.showerror("加载错误", f"加载Event模板文件失败: {str(e)}") def load_event_source(self): """加载Event源文件(.xml)""" file_path = filedialog.askopenfilename( title="选择Event源文件", filetypes=[("XML files", "*.xml"), ("All files", "*.*")] ) if not file_path: return try: self.event_tree = ET.parse(file_path) self.event_root = self.event_tree.getroot() self.event_source = self.event_tree # 在文本框中显示XML内容 self.display_xml_content(self.event_tree, self.event_xml_text) self.status_var.set(f"已加载Event源文件: {os.path.basename(file_path)}") except Exception as e: messagebox.showerror("加载错误", f"加载Event源文件失败: {str(e)}") def add_id_to_event(self): """为Event源文件添加ID属性""" if not self.event_source: messagebox.showwarning("无数据", "请先加载Event源文件") return # 添加ID属性 count = 0 for event in self.event_root.findall('.//event'): if 'id' not in event.attrib: # 在属性开头位置添加id属性 new_attrib = OrderedDict() new_attrib['id'] = '' for key, value in event.attrib.items(): new_attrib[key] = value # 更新属性 event.attrib.clear() for key, value in new_attrib.items(): event.set(key, value) count += 1 # 更新XML显示 self.display_xml_content(self.event_tree, self.event_xml_text) self.status_var.set(f"已为{count}个event元素添加ID属性") def get_id_from_event_template(self, event): """从Event模板获取ID""" for item in self.event_template: if item['name'] == event: return item['id'] return None def get_id_event(self, module, cat): """获取Event ID""" if cat in self.event_mapping: temp_name = self.event_mapping[cat] else: temp_name = "CHXXX_" + cat module_id = self.get_module_id(module) temp_id = self.get_id_from_event_template(temp_name) name = module + '_' + cat if temp_id and module_id: id = temp_id.replace('XX', str(module_id)) return name, id else: return None, None def split_by_type(self, event_name): """按类型拆分""" if not self.event_categories: return None, None for cat in self.event_categories: if event_name.endswith(cat): module = event_name[:-len(cat)] return module, cat return None, None def fill_event_attributes(self): """填充Event属性""" if not self.event_source: messagebox.showwarning("无数据", "请先加载EventConfig源文件") return if not self.event_template: messagebox.showwarning("无数据", "请先加载EventConfig模板文件") return fill_count = 0 not_exist = [] not_exist_all = [] for data_elem in self.event_tree.findall('.//event'): name = data_elem.get('name') module, cat = self.split_by_type(name) if not module or not cat: continue temp_name, temp_id = self.get_id_event(module, cat) if temp_name and temp_id: data_elem.set('name', temp_name) data_elem.set('id', temp_id) fill_count += 1 else: if temp_id is None and cat not in not_exist: not_exist.append(cat) if temp_id is None: not_exist_all.append(name) # 更新XML显示 self.display_xml_content(self.event_tree, self.event_xml_text) self.status_var.set(f"已为{fill_count}个event补充id属性，剩余{len(not_exist)}个不在模板中，总共剩余{len(not_exist_all)}个") def export_event(self): """导出Event源文件""" if not self.event_source: messagebox.showwarning("无数据", "没有可导出的数据") return file_path = filedialog.asksaveasfilename( title="保存Event文件", defaultextension=".xml", filetypes=[("XML files", "*.xml"), ("All files", "*.*")] ) if not file_path: return try: self.event_tree.write(file_path, encoding='utf-8', xml_declaration=True) self.status_var.set(f"Event文件已成功导出到: {os.path.basename(file_path)}") messagebox.showinfo("导出成功", "Event配置文件已成功导出") except Exception as e: messagebox.showerror("导出错误", f"导出文件失败: {str(e)}") def split_export_event(self): """拆分并导出Event文件""" if not self.event_source: messagebox.showwarning("无数据", "没有可拆分的数据") return # 创建分组字典，前缀为event节点列表 groups = defaultdict(list) # 遍历所有event节点 for event in self.event_root.findall('.//event'): name = event.get('name','') if '_' in name: prefix = name.split('_', 1)[0] groups[prefix].append(event) # 为每个分组创建新的xml文件 for prefix, data in groups.items(): # 创建新根节点（复制原始根节点标签和属性） new_root = ET.Element(self.event_root.tag, attrib=self.event_root.attrib) # 复制原始根节点的命名空间 for ns, uri in self.event_root.nsmap.items(): if ns: ET.register_namespace(ns, uri) # 添加原始非 event 节点 for child in self.event_root: if child.tag != 'event': new_root.append(child) # 添加分组中的 event 节点 for node in data: new_root.append(node) # 创建 XML 树并写入文件 new_tree = ET.ElementTree(new_root) filename = f"Event_{prefix}.xml" new_tree.write(filename, encoding='utf-8', xml_declaration=True) messagebox.showinfo("成功", f"已拆分为 {len(groups)} 个文件")
if name == “main”:
root = tk.Tk()
app = ConfigUpdaterApp(root)
root.mainloop() 改为代码格式，缩进格式之类
最低0.47元/天 解锁文章
扫一扫
1552
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
