Code: Select all
import tkinter as tk
from tkinter import ttk
import re
from tkinterdnd2 import DND_FILES, TkinterDnD
import os
class FactorioConfigComparer(TkinterDnD.Tk):
def __init__(self):
super().__init__()
self.title("Factorio Config Comparer")
self.geometry("1000x600")
# 创建主框架
self.main_frame = ttk.Frame(self)
self.main_frame.pack(expand=True, fill='both', padx=10, pady=10)
# 创建左右两栏
self.create_panels()
# 创建对比结果区域
self.create_result_area()
# 存储当前加载的配置
self.left_config = None
self.right_config = None
def create_panels(self):
# 创建左右面板容器
panels_frame = ttk.Frame(self.main_frame)
panels_frame.pack(expand=True, fill='both')
# 左面板
left_frame = ttk.LabelFrame(panels_frame, text="Left Config File")
left_frame.pack(side='left', expand=True, fill='both', padx=5)
self.left_text = tk.Text(left_frame, height=10, wrap=tk.WORD)
self.left_text.pack(expand=True, fill='both', padx=5, pady=5)
self.left_text.insert('1.0', "Drop .cfg file here")
# 配置拖放
self.left_text.drop_target_register(DND_FILES)
self.left_text.dnd_bind('<<Drop>>', lambda e: self.on_drop(e, 'left'))
# 右面板
right_frame = ttk.LabelFrame(panels_frame, text="Right Config File")
right_frame.pack(side='left', expand=True, fill='both', padx=5)
self.right_text = tk.Text(right_frame, height=10, wrap=tk.WORD)
self.right_text.pack(expand=True, fill='both', padx=5, pady=5)
self.right_text.insert('1.0', "Drop .cfg file here")
# 配置拖放
self.right_text.drop_target_register(DND_FILES)
self.right_text.dnd_bind('<<Drop>>', lambda e: self.on_drop(e, 'right'))
def create_result_area(self):
# 创建结果显示区域
result_frame = ttk.LabelFrame(self.main_frame, text="Comparison Results")
result_frame.pack(expand=True, fill='both', pady=10)
# 创建两个子区域用于显示缺失的键
results_container = ttk.Frame(result_frame)
results_container.pack(expand=True, fill='both')
# 左侧缺失的键
left_missing_frame = ttk.LabelFrame(results_container, text="Keys missing in Left file")
left_missing_frame.pack(side='left', expand=True, fill='both', padx=5, pady=5)
self.left_missing_text = tk.Text(left_missing_frame, height=10, wrap=tk.WORD)
self.left_missing_text.pack(expand=True, fill='both', padx=5, pady=5)
# 右侧缺失的键
right_missing_frame = ttk.LabelFrame(results_container, text="Keys missing in Right file")
right_missing_frame.pack(side='left', expand=True, fill='both', padx=5, pady=5)
self.right_missing_text = tk.Text(right_missing_frame, height=10, wrap=tk.WORD)
self.right_missing_text.pack(expand=True, fill='both', padx=5, pady=5)
def parse_factorio_cfg(self, file_path):
"""解析 Factorio 配置文件，返回键值对字典"""
config_dict = {}
current_key = None
with open(file_path, 'r', encoding='utf-8') as f:
for line in f:
line = line.strip()
# 跳过空行和注释
if not line or line.startswith('#') or line.startswith('_'):
continue
# 检查是否是键值对
if '=' in line:
key, value = line.split('=', 1)
key = key.strip()
value = value.strip()
# 去掉值的引号（如果存在）
value = value.strip('"').strip("'")
config_dict[key] = value
return config_dict
def on_drop(self, event, side):
# 获取拖放的文件路径
file_path = event.data
# 移除可能的大括号和引号
file_path = file_path.strip('{}').strip('"')
# 检查文件是否存在且是.cfg文件
if not os.path.exists(file_path) or not file_path.lower().endswith('.cfg'):
self.show_error("Please drop a valid .cfg file")
return
# 读取配置文件
try:
config_dict = self.parse_factorio_cfg(file_path)
# 更新相应的文本框和存储的配置
if side == 'left':
self.left_text.delete('1.0', tk.END)
self.left_text.insert('1.0', f"Loaded: {os.path.basename(file_path)}\n")
self.left_text.insert(tk.END, f"Found {len(config_dict)} keys")
self.left_config = config_dict
else:
self.right_text.delete('1.0', tk.END)
self.right_text.insert('1.0', f"Loaded: {os.path.basename(file_path)}\n")
self.right_text.insert(tk.END, f"Found {len(config_dict)} keys")
self.right_config = config_dict
# 如果两个配置文件都已加载，则进行比较
if self.left_config and self.right_config:
self.compare_configs()
except Exception as e:
self.show_error(f"Error reading config file: {str(e)}")
def compare_configs(self):
# 清空结果区域
self.left_missing_text.delete('1.0', tk.END)
self.right_missing_text.delete('1.0', tk.END)
# 获取所有键
left_keys = set(self.left_config.keys())
right_keys = set(self.right_config.keys())
# 找出缺失的键
missing_in_left = right_keys - left_keys
missing_in_right = left_keys - right_keys
# 显示结果
if missing_in_left:
for key in sorted(missing_in_left):
self.left_missing_text.insert(tk.END, f"{key}={self.right_config[key]}\n")
else:
self.left_missing_text.insert('1.0', "No missing keys")
if missing_in_right:
for key in sorted(missing_in_right):
self.right_missing_text.insert(tk.END, f"{key}={self.left_config[key]}\n")
else:
self.right_missing_text.insert('1.0', "No missing keys")
def show_error(self, message):
"""显示错误消息"""
import tkinter.messagebox as messagebox
messagebox.showerror("Error", message)
if __name__ == "__main__":
app = FactorioConfigComparer()
app.mainloop()