297 lines
13 KiB
Python
297 lines
13 KiB
Python
import tkinter as tk
|
||
from tkinter import ttk, filedialog, messagebox
|
||
import os
|
||
|
||
import logging
|
||
from core.template_manager import TemplateManager
|
||
from core.csv_manager import CSVManager
|
||
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class MainUI:
|
||
def __init__(self, root, base_dir):
|
||
self.root = root
|
||
self.root.title("KingSCADA 点表生成工具") # 窗口标题
|
||
self.root.geometry("1000x400") # 窗口大小
|
||
root.resizable(False, False) # 禁止水平和垂直调整大小
|
||
|
||
self.base_dir = base_dir
|
||
self.template_manager = TemplateManager(base_dir)
|
||
self.csv_manager = CSVManager(base_dir)
|
||
# 初始化导入数据模板数据
|
||
self.template_data = []
|
||
self.csv_data = []
|
||
|
||
self.build_ui()
|
||
logging.info(f"工具运行根目录{base_dir}")
|
||
|
||
def build_ui(self):
|
||
# 第一行容器：模板区 + CSV区
|
||
top_row = ttk.Frame(self.root)
|
||
top_row.pack(side='top', fill='x', padx=5, pady=5)
|
||
|
||
self.create_template_section(parent=top_row)
|
||
self.create_csv_section(parent=top_row)
|
||
|
||
# 第二行：参数区
|
||
self.create_input_section()
|
||
|
||
# 第三行：生成区
|
||
self.create_generate_section()
|
||
|
||
# ---------------- 模板区 ----------------
|
||
def create_template_section(self, parent):
|
||
frame = ttk.LabelFrame(parent, text="配置文件选择", padding=5)
|
||
frame.pack(side='left', padx=5, pady=5, anchor='nw')
|
||
|
||
self.device_cb = self._add_combobox(frame, "设备类型：", row=0, col=0, listbox=self.template_manager.get_device_types(), inivar=-1)
|
||
# 选择完成事件
|
||
self.device_cb["combobox"].bind('<<ComboboxSelected>>', self.on_device_selected)
|
||
|
||
self.template_cb = self._add_combobox(frame, "模板文件：", row=0, col=1, listbox=[], inivar=-1)
|
||
#选择完成事件
|
||
self.template_cb['combobox'].bind('<<ComboboxSelected>>', self.on_template_selected)
|
||
# 模板显示表格
|
||
self.template_table = ttk.Treeview(frame, columns=("name","desc","type","access","address"), show="headings", height=5)
|
||
col_defs = {
|
||
"name": ("名称", 100),
|
||
"desc": ("描述", 130),
|
||
"type": ("类型", 100),
|
||
"access": ("读写", 80),
|
||
"address": ("地址", 78),
|
||
}
|
||
for col, (text, width) in col_defs.items():
|
||
self.template_table.heading(col, text=text)
|
||
self.template_table.column(col, width=width, anchor="center")
|
||
self.template_table.grid(row=1, column=0, columnspan=4, sticky='nsew', pady=(9,5))
|
||
|
||
def on_device_selected(self, event=None):
|
||
"""设备类型选择完成事件"""
|
||
# 获取目录下的所有模板文件
|
||
device = self.device_cb['var'].get()
|
||
self.template_cb['combobox']['values'] = self.template_manager.get_templates_by_device(device)
|
||
#更新参数区的内容
|
||
if device == "SIEMENS" :
|
||
for row in self.template_table.get_children(): #清空表格
|
||
self.template_table.delete(row)
|
||
self.template_cb['var'].set("") #清空模板选择
|
||
self.deviceseries['combobox']['values'] = self.deviceseries_siemens #更新设备系列选项
|
||
self.channeldriver['combobox']['values'] = self.channeldriver_siemens #更新通道驱动选项
|
||
self.deviceseries['var'].set(self.deviceseries_siemens[0]) #设置默认值
|
||
self.channeldriver['var'].set(self.channeldriver_siemens[0]) #设置默认值
|
||
self.db_num["frame"].grid() #显示DB块号输入框
|
||
if device == "AB" :
|
||
for row in self.template_table.get_children():
|
||
self.template_table.delete(row)
|
||
self.template_cb['var'].set("")
|
||
self.deviceseries['combobox']['values'] = self.deviceseries_ab
|
||
self.channeldriver['combobox']['values'] = self.channeldriver_ab
|
||
self.deviceseries['var'].set(self.deviceseries_ab[0])
|
||
self.channeldriver['var'].set(self.channeldriver_ab[0])
|
||
self.db_num["frame"].grid_remove() #隐藏DB块号输入框
|
||
|
||
|
||
def on_template_selected(self, event=None):
|
||
"""模板文件选择完成事件"""
|
||
device = self.device_cb['var'].get()
|
||
template = self.template_cb['var'].get()
|
||
self.template_data = self.template_manager.load_template(device, template)
|
||
self.refresh_template_table()
|
||
|
||
def refresh_template_table(self):
|
||
"""刷新模板显示表格"""
|
||
for row in self.template_table.get_children():
|
||
self.template_table.delete(row)
|
||
try:
|
||
for item in self.template_data:
|
||
self.template_table.insert('', 'end', values=(item['name'], item['desc'], item['type'], item['access'], item['address']))
|
||
except Exception as e:
|
||
for row in self.template_table.get_children():
|
||
self.template_table.delete(row)
|
||
logger.error(f"加载模板异常{e}")
|
||
messagebox.showwarning("加载出错", f"加载模板异常{e}", icon="error")
|
||
|
||
# ---------------- CSV区 ----------------
|
||
def create_csv_section(self, parent):
|
||
frame = ttk.LabelFrame(parent, text="CSV 数据导入", padding=5)
|
||
frame.pack(side='left', padx=5, pady=5, anchor='nw')
|
||
|
||
btn = ttk.Button(frame, text="选择CSV文件", command=self.load_csv_file)
|
||
btn.grid(row=0, column=0, sticky='w')
|
||
|
||
self.csv_table = ttk.Treeview(frame, columns=("code","desc","offset"), show="headings", height=5)
|
||
col_defs = {
|
||
"code": ("设备代号", 150),
|
||
"desc": ("设备名称", 180),
|
||
"offset": ("拼接地址", 120),
|
||
}
|
||
for col, (text, width) in col_defs.items():
|
||
self.csv_table.heading(col, text=text)
|
||
self.csv_table.column(col, width=width, anchor="center")
|
||
self.csv_table.grid(row=1, column=0, columnspan=4, sticky='nsew', pady=5)
|
||
|
||
def load_csv_file(self):
|
||
filepath = filedialog.askopenfilename(filetypes=[("CSV Files", "*.csv")])
|
||
if not filepath:
|
||
return
|
||
self.csv_data = self.csv_manager.load_csv(filepath)
|
||
self.refresh_csv_table()
|
||
|
||
def refresh_csv_table(self):
|
||
"""刷新CSV显示表格"""
|
||
for row in self.csv_table.get_children():
|
||
self.csv_table.delete(row)
|
||
try:
|
||
for row in self.csv_data:
|
||
self.csv_table.insert('', 'end', values=(row['设备代号'], row['设备描述'], row['拼接地址']))
|
||
except Exception as e:
|
||
for row in self.csv_table.get_children():
|
||
self.csv_table.delete(row)
|
||
logger.error(f"导入数据异常{e}")
|
||
messagebox.showwarning("导入出错", f"导入数据异常{e}\n检查第一行列名是否正确", icon="error")
|
||
|
||
# ---------------- 参数区 ----------------
|
||
def create_input_section(self):
|
||
frame = ttk.LabelFrame(self.root, text="参数输入", padding=10)
|
||
frame.pack(side='top', fill='x', padx=10, pady=5)
|
||
|
||
self.start_id = self._add_input(frame, "起始ID", row=0, col=0, inivar=1001)
|
||
self.dev_name = self._add_input(frame, "设备名称", row=0, col=1, inivar="PLC1")
|
||
self.group_name = self._add_input(frame, "分组路径", row=0, col=2, inivar="TEST.一期")
|
||
self.group_name_en = self._add_combobox(frame, "设备分组", row=0, col=3, listbox=["禁用", "启用"])
|
||
|
||
self.link = self._add_combobox(frame, "采集链路", row=1, col=0, listbox=["以太网", "COM"])
|
||
#选择完成事件
|
||
self.link["combobox"].bind('<<ComboboxSelected>>', self.on_link_selected)
|
||
self.link_com = self._add_input(frame, "串口号", row=1, col=1, inivar="11")
|
||
self.link_ip = self._add_input(frame, "IP地址", row=1, col=1, inivar="192.168.10.11")
|
||
|
||
self.deviceseries_siemens = ["S7-1500", "S7-1200", "S7-300"]
|
||
self.channeldriver_siemens = ["S71500Tcp", "S71200Tcp", "S7300Tcp"]
|
||
self.deviceseries_ab = ["AB-ControlLogixTCP"]
|
||
self.channeldriver_ab = ["ControlLogix"]
|
||
self.deviceseries = self._add_combobox(frame, "设备系列", row=2, col=0, listbox=self.deviceseries_siemens)
|
||
self.channeldriver = self._add_combobox(frame, "通道驱动", row=2, col=1, listbox=self.channeldriver_siemens)
|
||
self.db_num = self._add_input(frame, "DB块号", row=2, col=2, inivar="3")
|
||
|
||
def on_link_selected(self, event=None):
|
||
"""链路选择完成事件"""
|
||
link_var = self.link["var"].get()
|
||
if link_var == "以太网" :
|
||
self.link_com["frame"].grid_remove()
|
||
self.link_ip["frame"].grid()
|
||
if link_var == "COM" :
|
||
self.link_ip["frame"].grid_remove()
|
||
self.link_com["frame"].grid()
|
||
|
||
def _add_input(
|
||
self, parent, label_text,
|
||
row, col=0, inivar="",
|
||
label_width=8, entry_width=20, colspan=1, sticky='w'
|
||
):
|
||
"""
|
||
添加一个带文本标签和输入框的组合控件，并返回 (StringVar, Frame) 以便后续控制。
|
||
- parent: 父容器
|
||
- label: 标签文字
|
||
- row, col: 放置在父容器的 grid 行列
|
||
- inivar: 初始值
|
||
- entry_width: 输入框宽度
|
||
- colspan: 该组控件在父容器上跨越的列数
|
||
- sticky: 对齐方式（默认左对齐）
|
||
"""
|
||
group_frame = ttk.Frame(parent)
|
||
group_frame.grid(row=row, column=col, columnspan=colspan, sticky=sticky, padx=5, pady=3)
|
||
|
||
# 标签
|
||
label = ttk.Label(group_frame, text=label_text + "：", width=label_width, anchor='w').grid(row=0, column=0, sticky='w', padx=(0, 5))
|
||
|
||
# 输入框
|
||
var = tk.StringVar(value=inivar)
|
||
entry = ttk.Entry(group_frame, textvariable=var, width=entry_width)
|
||
entry.grid(row=0, column=1, sticky='w')
|
||
|
||
return {
|
||
"frame": group_frame,
|
||
"label": label,
|
||
"entry": entry,
|
||
"var": var
|
||
}
|
||
|
||
def _add_combobox(
|
||
self, parent, label_text, row, col=0,
|
||
listbox=[], inivar=0, width=17, colspan=1,
|
||
sticky='w', label_width=8, state="readonly"
|
||
):
|
||
"""
|
||
创建一组 [标签 + 下拉框] 控件。
|
||
返回 dict，方便外部单独或统一控制。
|
||
|
||
- values: 下拉选项列表
|
||
- default: 初始值（可选）
|
||
- state: "readonly" 表示只能从列表选，"normal" 可手动输入
|
||
"""
|
||
frame = ttk.Frame(parent)
|
||
frame.grid(row=row, column=col, columnspan=colspan, sticky=sticky, padx=5, pady=3)
|
||
|
||
# 标签
|
||
label = ttk.Label(
|
||
frame,
|
||
text=label_text + "：",
|
||
width=label_width,
|
||
anchor='w'
|
||
)
|
||
label.grid(row=0, column=0, sticky='w', padx=(0, 5))
|
||
|
||
# 变量 + 下拉框
|
||
var = tk.StringVar()
|
||
combobox = ttk.Combobox(
|
||
frame,
|
||
textvariable=var,
|
||
values=listbox,
|
||
width=width,
|
||
state=state
|
||
)
|
||
combobox.grid(row=0, column=1, sticky='w')
|
||
if inivar>=0:
|
||
var.set(listbox[inivar])
|
||
|
||
return {
|
||
"frame": frame,
|
||
"label": label,
|
||
"combobox": combobox,
|
||
"var": var
|
||
}
|
||
|
||
|
||
# ---------------- 生成区 ----------------
|
||
def create_generate_section(self):
|
||
frame = ttk.Frame(self.root)
|
||
frame.pack(fill='x', padx=10, pady=5)
|
||
|
||
btn = ttk.Button(frame, text="生成点表文件", command=self.generate_csv)
|
||
btn.pack(anchor='center')
|
||
|
||
def generate_csv(self):
|
||
if not self.template_data or not self.csv_data:
|
||
messagebox.showwarning("警告", "请先加载模板和CSV数据！")
|
||
return
|
||
|
||
inputs = {
|
||
"start_id": self.start_id["var"].get(), #起始ID
|
||
"ip": self.link_ip["var"].get(),
|
||
"device_name": self.dev_name["var"].get(), #设备名称
|
||
"group_name": self.group_name["var"].get(), #分组路径
|
||
"link":self.link["var"].get(), #链路选择
|
||
"link_ip":self.link_ip["var"].get(), #IP地址
|
||
"link_com":self.link_com["var"].get(), #串口号
|
||
"deviceseries": self.deviceseries["var"].get(), #设备系类
|
||
"channeldriver": self.channeldriver["var"].get(), #通道驱动
|
||
"db_num": self.db_num["var"].get(), #DB块号
|
||
"device": self.device_cb["var"].get(), #设备类型
|
||
"group_name_en": self.group_name_en["var"].get() #设备分组是否启用
|
||
}
|
||
|
||
output_path = self.csv_manager.generate_output(self.template_data, inputs)
|
||
messagebox.showwarning("生成成功", output_path, icon="info")
|
||
|