11

core/ui/basic_ui.py

@@ -0,0 +1,112 @@
+import os
+import tkinter as tk
+from tkinter import ttk, filedialog, messagebox
+
+import logging
+logger = logging.getLogger(__name__)
+
+class BasicUI:
+
+    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 add_button(
+                self, parent, button_text, 
+                row, col=0, command="", 
+                label_width=8, entry_width=20, colspan=1, sticky='w'
+            ):
+            """
+            添加一个按钮，并返回 (StringVar, Frame) 以便后续控制。
+            - parent: 父容器
+            - button_text: 按钮文本
+            - row, col: 放置在父容器的 grid 行列
+            - command: 调用的函数
+            - entry_width: 输入框宽度
+            - colspan: 该组控件在父容器上跨越的列数
+            - sticky: 对齐方式（默认左对齐）
+            """
+            group_frame = ttk.Frame(parent)
+            group_frame.grid(row=row, column=col, columnspan=colspan, sticky=sticky, padx=5, pady=3)
+
+            btn = ttk.Button(group_frame, text=button_text, command=command)
+            btn.grid(row=0, column=0, sticky='w')
+            return {
+                "frame": group_frame,
+                "btn": btn,
+            }
+        

core/ui/tab_ping.py

@@ -0,0 +1,88 @@
+import tkinter as tk
+from tkinter import ttk, filedialog, messagebox, scrolledtext
+
+from core.ui.basic_ui import BasicUI
+from core.Function.ping_fun import PingFun
+
+import logging
+logger = logging.getLogger(__name__)
+
+class PingTab(ttk.Frame, BasicUI):
+    def __init__(self, parent):
+        super().__init__(parent)   
+        self.ping_ui()
+        self.ping_fun = PingFun(self.result_box)
+
+    def ping_ui(self):
+        """ping界面布局"""
+        self.create_allIP_section()
+        self.create_assignIP_section()
+        self.create_outputping_section()
+# --------------------------------------UI界面布局函数--------------------------------------
+    def create_allIP_section(self):
+        # 区域标签
+        frame = ttk.LabelFrame(self, text="PING 目标IP")
+        frame.pack(side='top', fill='x', padx=10, pady=5)
+
+        self.entry_allIP = self.add_input(frame, "目标IP", row=0, col=0) 
+        self.allIP_startPing = self.add_button(frame, "开始ping", row=0, col=1, command=self.btn_allIP_startPing)
+        self.allIP_stopPing = self.add_button(frame, "停止ping", row=0, col=2, command=self.btn_allIP_stopPing)
+
+    def create_assignIP_section(self):
+        # 区域标签
+        frame = ttk.LabelFrame(self, text="指定IP PING 目标IP")
+        frame.pack(side='top', fill='x', padx=10, pady=5)
+
+        self.entry_assignIP_A = self.add_input(frame, "本地IP", row=0, col=0) 
+        self.entry_assignIP_B = self.add_input(frame, "目标IP", row=0, col=1) 
+        self.assignIP_startPing = self.add_button(frame, "开始ping", row=0, col=2, command=self.btn_assignIP_startPing)
+        self.assignIP_stopPing = self.add_button(frame, "停止ping", row=0, col=3, command=self.btn_assignIP_stopPing)
+
+    def create_outputping_section(self):
+        # 区域标签
+        frame = ttk.LabelFrame(self, text="PING 结果输出")
+        frame.pack(side='top', fill='x', padx=10, pady=5)
+
+        self.result_box = scrolledtext.ScrolledText(frame, width=100, height=20)
+        self.result_box.pack(pady=10)
+# --------------------------------------按钮回调函数--------------------------------------
+    def btn_allIP_startPing(self):   
+        if not self.entry_allIP['var'].get():
+            messagebox.showwarning("输入错误", "请输入IP或域名！")
+            return   
+        logger.info(f"开始Ping {self.entry_allIP['var'].get()}")
+        self.ping_fun.strat_ping(self.entry_allIP['var'].get(), callback=self.allIP_ping_callback)
+        self.allIP_startPing['btn'].config(state='disabled')
+
+    def btn_allIP_stopPing(self):   
+        logger.info(f"停止Ping {self.entry_allIP['var'].get()}")
+        self.ping_fun.stop_ping()
+        self.allIP_startPing['btn'].config(state='normal')
+
+    def allIP_ping_callback(self):
+        self.allIP_startPing['btn'].config(state='normal')
+
+    def btn_assignIP_startPing(self):         
+        if not self.entry_assignIP_B['var'].get():
+            messagebox.showwarning("输入错误", "请输入IP或域名！")
+            return   
+        logger.info(f"开始由{self.entry_assignIP_A['var'].get()} Ping {self.entry_assignIP_B['var'].get()}")
+        self.ping_fun.strat_ping(self.entry_assignIP_B['var'].get(),local_ip=self.entry_assignIP_A['var'].get(), 
+                                 callback=self.assignIP_ping_callback)
+        self.assignIP_startPing['btn'].config(state='disabled')
+
+    def btn_assignIP_stopPing(self):   
+        logger.info(f"停止由{self.entry_assignIP_A['var'].get()} Ping {self.entry_assignIP_B['var'].get()}")
+        self.ping_fun.stop_ping()
+        self.assignIP_startPing['btn'].config(state='normal')
+
+    def assignIP_ping_callback(self):
+        self.assignIP_startPing['btn'].config(state='normal')
+
+
+
+
+
+
+   
+        

core/ui/ui_main copy.py

@@ -0,0 +1,297 @@
+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")
+        

core/ui/ui_main.py

@@ -0,0 +1,25 @@
+import tkinter as tk
+from tkinter import ttk, filedialog, messagebox
+
+from core.ui.basic_ui import BasicUI
+from core.ui.tab_ping import PingTab
+
+import logging
+logger = logging.getLogger(__name__)
+
+class MainUI(BasicUI):
+    def __init__(self, root):
+        self.root = root
+        self.root.title("Network tools")   # 窗口标题
+        self.root.geometry("800x500")  # 窗口大小
+        root.resizable(False, False)    # 禁止水平和垂直调整大小
+        # 创建 Notebook 作为多标签页容器
+        self.tab_control = ttk.Notebook(root)
+        self.tab_control.pack(expand=1, fill="both")
+        # 实例化各个功能页
+        self.tabs = {
+            "Ping测试": PingTab(self.tab_control),
+        }
+        # 添加到 Notebook
+        for name, tab in self.tabs.items():
+            self.tab_control.add(tab, text=name)