parent
24b7906d62
commit
0c05efe3cc
@ -0,0 +1,267 @@
|
|||||||
|
import ipaddress
|
||||||
|
import re
|
||||||
|
import sqlite3
|
||||||
|
from tkinter import messagebox
|
||||||
|
|
||||||
|
import ttkbootstrap as tk
|
||||||
|
from ttkbootstrap import *
|
||||||
|
from ttkbootstrap import ttk
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
from SimObjs import SimRouter
|
||||||
|
|
||||||
|
|
||||||
|
def round_rectangle(cv, x1, y1, x2, y2, radius=30, **kwargs):
|
||||||
|
"""
|
||||||
|
绘制圆角矩形
|
||||||
|
:param cv: canvas对象
|
||||||
|
:param radius: 圆角值
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
points = [x1 + radius, y1,
|
||||||
|
x1 + radius, y1,
|
||||||
|
x2 - radius, y1,
|
||||||
|
x2 - radius, y1,
|
||||||
|
x2, y1,
|
||||||
|
x2, y1 + radius,
|
||||||
|
x2, y1 + radius,
|
||||||
|
x2, y2 - radius,
|
||||||
|
x2, y2 - radius,
|
||||||
|
x2, y2,
|
||||||
|
x2 - radius, y2,
|
||||||
|
x2 - radius, y2,
|
||||||
|
x1 + radius, y2,
|
||||||
|
x1 + radius, y2,
|
||||||
|
x1, y2,
|
||||||
|
x1, y2 - radius,
|
||||||
|
x1, y2 - radius,
|
||||||
|
x1, y1 + radius,
|
||||||
|
x1, y1 + radius,
|
||||||
|
x1, y1]
|
||||||
|
|
||||||
|
return cv.create_polygon(points, **kwargs, smooth=True)
|
||||||
|
|
||||||
|
def validate_ip_address(ip_address):
|
||||||
|
"""
|
||||||
|
匹配ip地址格式是否规范
|
||||||
|
:param ip_address: IP地址
|
||||||
|
:return: Boolean
|
||||||
|
"""
|
||||||
|
# 定义IP地址的正则表达式模式
|
||||||
|
pattern_with_subnet = r'^(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/(\d{1,2})$'
|
||||||
|
pattern_without_subnet = r'^(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})$'
|
||||||
|
# 使用re模块进行匹配
|
||||||
|
match_with_subnet = re.match(pattern_with_subnet, ip_address)
|
||||||
|
match_without_subnet = re.match(pattern_without_subnet, ip_address)
|
||||||
|
if match_with_subnet:
|
||||||
|
# 带有子网掩码的IP地址
|
||||||
|
# 检查每个组件的取值范围是否在 0-255 之间
|
||||||
|
for group in match_with_subnet.groups()[:4]:
|
||||||
|
if not (0 <= int(group) <= 255):
|
||||||
|
return False
|
||||||
|
# 检查子网掩码的取值范围是否在 0-32 之间
|
||||||
|
subnet_mask = int(match_with_subnet.groups()[4])
|
||||||
|
if not (0 <= subnet_mask <= 32):
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
elif match_without_subnet:
|
||||||
|
# 不带子网掩码的IP地址
|
||||||
|
# 检查每个组件的取值范围是否在 0-255 之间
|
||||||
|
for group in match_without_subnet.groups():
|
||||||
|
if not (0 <= int(group) <= 255):
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
# IP地址格式不正确
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
class ExportUtil():
|
||||||
|
def __init__(self, path):
|
||||||
|
self.conn = sqlite3.connect('./network.db')
|
||||||
|
self.path = path
|
||||||
|
|
||||||
|
def get_table_names(self):
|
||||||
|
cursor = self.conn.cursor()
|
||||||
|
cursor.execute("SELECT name FROM sqlite_master WHERE type='table';") # 如果你使用SQLite数据库
|
||||||
|
tables = cursor.fetchall()
|
||||||
|
cursor.close()
|
||||||
|
return [table[0] for table in tables]
|
||||||
|
|
||||||
|
def export(self):
|
||||||
|
tables = self.get_table_names()
|
||||||
|
with pd.ExcelWriter(self.path, engine='openpyxl') as writer:
|
||||||
|
for table in tables:
|
||||||
|
table_name = table
|
||||||
|
# a. 从数据库中获取表的数据并存储在DataFrame中
|
||||||
|
query = f"SELECT * FROM {table_name}"
|
||||||
|
df = pd.read_sql(query, self.conn)
|
||||||
|
# b. 使用Pandas将数据写入Excel文件的不同sheet中
|
||||||
|
df.to_excel(writer, sheet_name=table_name, index=False)
|
||||||
|
|
||||||
|
def import_data(self):
|
||||||
|
excel_file = pd.ExcelFile(self.path)
|
||||||
|
sheet_names = excel_file.sheet_names
|
||||||
|
cursor = self.conn.cursor()
|
||||||
|
for sheet_name in sheet_names:
|
||||||
|
# 4. 使用 Pandas 读取工作表数据
|
||||||
|
df = pd.read_excel(excel_file, sheet_name=sheet_name)
|
||||||
|
# 5. 获取工作表的列名
|
||||||
|
columns = df.columns.tolist()
|
||||||
|
# 6. 构造插入语句
|
||||||
|
columns_str = ', '.join(columns)
|
||||||
|
placeholders = ', '.join(['?' for _ in range(len(columns))])
|
||||||
|
sql = f"INSERT INTO {sheet_name} ({columns_str}) VALUES ({placeholders})"
|
||||||
|
# 7. 将数据插入数据库
|
||||||
|
for index, row in df.iterrows():
|
||||||
|
# 8. 使用动态生成的 SQL 语句将数据插入数据库
|
||||||
|
cursor.execute(sql, tuple(row))
|
||||||
|
self.conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
class RouterConfigWindow(tk.Toplevel):
|
||||||
|
def __init__(self, parent, router_obj):
|
||||||
|
super().__init__(parent)
|
||||||
|
self.geometry("435x433+350+200")
|
||||||
|
self.title(f"{router_obj.ObjLabel}路由表配置")
|
||||||
|
self.router_obj = router_obj
|
||||||
|
self.interface_entries = []
|
||||||
|
self.router_table = {}
|
||||||
|
self.create_interface_inputs()
|
||||||
|
self.create_router_table()
|
||||||
|
|
||||||
|
def create_interface_inputs(self):
|
||||||
|
label_text = ["接口1", "接口2", "接口3", "接口4"]
|
||||||
|
for i in range(4):
|
||||||
|
label = tk.Label(self, text=label_text[i], font=("黑体", 16))
|
||||||
|
label.grid(row=i, column=0, padx=10, pady=5, sticky="w")
|
||||||
|
entry = tk.Entry(self, width=20, font=("黑体", 16),)
|
||||||
|
entry.grid(row=i, column=1, padx=10, pady=5, sticky="w")
|
||||||
|
self.interface_entries.append(entry)
|
||||||
|
button = tk.Button(self, text="添加", font=("黑体", 16), command=lambda index=i: self.add_router_entry(index))
|
||||||
|
button.grid(row=i, column=2, padx=10, pady=5)
|
||||||
|
lab = LabelFrame(self, text="示例")
|
||||||
|
lab.grid(row=4, column=0, columnspan=3, sticky=W, padx=20)
|
||||||
|
Label(lab, text="10.1.2.0/24 或者 10.1.2.12" if self.router_obj.ObjType == 2 else "MAC11", font=("黑体", 16)).pack()
|
||||||
|
|
||||||
|
def create_router_table(self):
|
||||||
|
def on_right_click(event):
|
||||||
|
row = self.router_treeview.identify_row(event.y) # 获取鼠标位置的行索引
|
||||||
|
if row:
|
||||||
|
self.router_treeview.selection_set(row) # 选中该行
|
||||||
|
delete_menu.post(event.x_root, event.y_root) # 在鼠标位置弹出删除菜单
|
||||||
|
|
||||||
|
def delete_row():
|
||||||
|
selected_items = self.router_treeview.selection() # 获取选中的行
|
||||||
|
for item in selected_items:
|
||||||
|
ifs, network = int(self.router_treeview.item(item)["values"][0][-1:]), self.router_treeview.item(item)["values"][1]
|
||||||
|
self.router_obj.delete_config(ifs, network)
|
||||||
|
self.router_treeview.delete(item)
|
||||||
|
self.router_table_frame = tk.Frame(self)
|
||||||
|
self.router_table_frame.grid(row=5, column=0, columnspan=3, padx=10, pady=5)
|
||||||
|
style = ttk.Style()
|
||||||
|
style.configure("Custom.Treeview.Heading", font=("宋体", 15))
|
||||||
|
style.configure("Custom.Treeview", rowheight=30, font=("宋体", 15))
|
||||||
|
self.router_treeview = ttk.Treeview(self.router_table_frame, style="Custom.Treeview", columns=("Interface", "Route"), show="headings")
|
||||||
|
self.router_treeview.heading("Interface", text="接口")
|
||||||
|
self.router_treeview.heading("Route", text="网段")
|
||||||
|
self.router_treeview.pack(side="left", fill="both")
|
||||||
|
scrollbar = ttk.Scrollbar(self.router_table_frame, orient="vertical", command=self.router_treeview.yview)
|
||||||
|
scrollbar.pack(side="right", fill="y")
|
||||||
|
self.router_treeview.configure(yscrollcommand=scrollbar.set)
|
||||||
|
self.router_table = self.router_obj.router_table
|
||||||
|
self.router_treeview.bind("<Button-3>", on_right_click)
|
||||||
|
# 创建删除菜单
|
||||||
|
delete_menu = tk.Menu(self.master, tearoff=False)
|
||||||
|
delete_menu.add_command(label="删除", command=delete_row)
|
||||||
|
self.update_router_table()
|
||||||
|
|
||||||
|
def add_router_entry(self, index):
|
||||||
|
entry_text = self.interface_entries[index].get()
|
||||||
|
try:
|
||||||
|
ipaddress.ip_network(entry_text)
|
||||||
|
if isinstance(self.router_obj, SimRouter):
|
||||||
|
if not validate_ip_address(entry_text):
|
||||||
|
messagebox.showerror("注意", message="添加的网段信息格式不合格")
|
||||||
|
self.interface_entries[index].delete(0, tk.END)
|
||||||
|
self.focus_set()
|
||||||
|
return
|
||||||
|
if entry_text:
|
||||||
|
if index + 1 in self.router_table:
|
||||||
|
self.router_table[index + 1].append(entry_text)
|
||||||
|
else:
|
||||||
|
self.router_table[index + 1] = [entry_text]
|
||||||
|
self.interface_entries[index].delete(0, tk.END)
|
||||||
|
self.router_obj.add_config(entry_text, index + 1)
|
||||||
|
self.update_router_table()
|
||||||
|
self.master.message.show_message(f"{self.router_obj.ObjLabel}添加配置{entry_text}成功！")
|
||||||
|
except:
|
||||||
|
messagebox.showerror("注意", message="网段格式错误！网段示例如下：\n10.1.2.0/24\n10.1.2.12")
|
||||||
|
return
|
||||||
|
|
||||||
|
def update_router_table(self):
|
||||||
|
self.router_treeview.delete(*self.router_treeview.get_children())
|
||||||
|
for i, entrys in self.router_table.items():
|
||||||
|
for entry in entrys:
|
||||||
|
self.router_treeview.insert("", "end", values=(f"接口{i}", entry))
|
||||||
|
|
||||||
|
|
||||||
|
class SwitchConfigWindow(RouterConfigWindow):
|
||||||
|
def __init__(self, parent, router_obj):
|
||||||
|
super().__init__(parent, router_obj)
|
||||||
|
self.geometry("435x433+350+200")
|
||||||
|
self.title(f"{router_obj.ObjLabel}交换表配置")
|
||||||
|
self.router_obj = router_obj
|
||||||
|
self.interface_entries = []
|
||||||
|
self.router_table = {}
|
||||||
|
self.create_interface_inputs()
|
||||||
|
self.create_router_table()
|
||||||
|
|
||||||
|
def create_router_table(self):
|
||||||
|
def on_right_click(event):
|
||||||
|
row = self.router_treeview.identify_row(event.y) # 获取鼠标位置的行索引
|
||||||
|
if row:
|
||||||
|
self.router_treeview.selection_set(row) # 选中该行
|
||||||
|
delete_menu.post(event.x_root, event.y_root) # 在鼠标位置弹出删除菜单
|
||||||
|
|
||||||
|
def delete_row():
|
||||||
|
selected_items = self.router_treeview.selection() # 获取选中的行
|
||||||
|
for item in selected_items:
|
||||||
|
ifs, network = int(self.router_treeview.item(item)["values"][0][-1:]), self.router_treeview.item(item)["values"][1]
|
||||||
|
self.router_obj.delete_config(ifs, network)
|
||||||
|
self.router_treeview.delete(item)
|
||||||
|
|
||||||
|
self.router_table_frame = tk.Frame(self)
|
||||||
|
self.router_table_frame.grid(row=5, column=0, columnspan=3, padx=10, pady=5)
|
||||||
|
self.router_treeview = ttk.Treeview(self.router_table_frame, columns=("Interface", "Route"), show="headings")
|
||||||
|
self.router_treeview.heading("Interface", text="接口")
|
||||||
|
self.router_treeview.heading("Route", text="mac")
|
||||||
|
self.router_treeview.pack(side="left", fill="both")
|
||||||
|
scrollbar = ttk.Scrollbar(self.router_table_frame, orient="vertical", command=self.router_treeview.yview)
|
||||||
|
scrollbar.pack(side="right", fill="y")
|
||||||
|
self.router_treeview.configure(yscrollcommand=scrollbar.set)
|
||||||
|
self.router_table = self.router_obj.mac_table
|
||||||
|
self.router_treeview.bind("<Button-3>", on_right_click)
|
||||||
|
# 创建删除菜单
|
||||||
|
delete_menu = tk.Menu(self.master, tearoff=False)
|
||||||
|
delete_menu.add_command(label="删除", command=delete_row)
|
||||||
|
self.update_router_table()
|
||||||
|
|
||||||
|
|
||||||
|
def add_router_entry(self, index):
|
||||||
|
entry_text = self.interface_entries[index].get()
|
||||||
|
if isinstance(self.router_obj, SimRouter):
|
||||||
|
if not validate_ip_address(entry_text):
|
||||||
|
messagebox.showerror("注意", message="添加的网段信息格式不合格")
|
||||||
|
self.interface_entries[index].delete(0, tk.END)
|
||||||
|
self.focus_set()
|
||||||
|
return
|
||||||
|
if entry_text:
|
||||||
|
if index + 1 in self.router_table:
|
||||||
|
self.router_table[index + 1].append(entry_text)
|
||||||
|
else:
|
||||||
|
self.router_table[index + 1] = [entry_text]
|
||||||
|
self.interface_entries[index].delete(0, tk.END)
|
||||||
|
self.router_obj.add_config(entry_text, index + 1)
|
||||||
|
self.master.message.show_message(f"{self.router_obj.ObjLabel}添加配置{entry_text}成功！")
|
||||||
|
self.update_router_table()
|
||||||
Binary file not shown.
@ -0,0 +1,104 @@
|
|||||||
|
import sqlite3
|
||||||
|
import sys
|
||||||
|
|
||||||
|
import pandas as pd
|
||||||
|
from pandas import DataFrame
|
||||||
|
|
||||||
|
conn = sqlite3.connect(sys.path[0]+"/network.db")
|
||||||
|
|
||||||
|
def execute_sql(sql):
|
||||||
|
"""
|
||||||
|
执行sql语句
|
||||||
|
:param sql:
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute(sql)
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def search(sql) -> DataFrame:
|
||||||
|
return pd.read_sql(sql, conn)
|
||||||
|
|
||||||
|
|
||||||
|
def delete_obj(obj_id):
|
||||||
|
cursor = conn.cursor()
|
||||||
|
delete_obj_sql = f"delete from sim_objs where ObjID='{obj_id}'"
|
||||||
|
cursor.execute(delete_obj_sql)
|
||||||
|
delete_conn_sql = f"delete from sim_conn where conn_id in (select conn_id from conn_config where node_id='{obj_id}')"
|
||||||
|
cursor.execute(delete_conn_sql)
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def truncate_db():
|
||||||
|
init_database()
|
||||||
|
|
||||||
|
def init_database():
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("""
|
||||||
|
DROP TABLE IF EXISTS `conn_config`;
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE `conn_config` (
|
||||||
|
`conn_id` varchar(55) NULL DEFAULT NULL,
|
||||||
|
`node_id` varchar(55) NULL DEFAULT NULL,
|
||||||
|
`node_ifs` int(0) NULL DEFAULT NULL,
|
||||||
|
`ip` varchar(55) NULL DEFAULT NULL,
|
||||||
|
`mac` varchar(128) NULL DEFAULT NULL,
|
||||||
|
`conn_port` varchar(32) NULL DEFAULT NULL,
|
||||||
|
`addr` varchar(255) NULL DEFAULT NULL,
|
||||||
|
CONSTRAINT `conn_config_sim_conn_conn_id_fk` FOREIGN KEY (`conn_id`) REFERENCES `sim_conn` (`conn_id`) ON DELETE CASCADE ON UPDATE CASCADE
|
||||||
|
) ;
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
DROP TABLE IF EXISTS `mac_table`;
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE `mac_table` (
|
||||||
|
`obj_id` varchar(55) NULL DEFAULT NULL,
|
||||||
|
`node_ifs` int(0) NULL DEFAULT NULL,
|
||||||
|
`mac` varchar(55) NULL DEFAULT NULL,
|
||||||
|
CONSTRAINT `mac_table_sim_objs_ObjID_fk` FOREIGN KEY (`obj_id`) REFERENCES `sim_objs` (`ObjID`) ON DELETE CASCADE ON UPDATE CASCADE
|
||||||
|
) ;
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
DROP TABLE IF EXISTS `router_table`;
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE `router_table` (
|
||||||
|
`obj_id` varchar(55) NULL DEFAULT NULL,
|
||||||
|
`node_ifs` int(0) NULL DEFAULT NULL,
|
||||||
|
`segment` varchar(55) NULL DEFAULT NULL,
|
||||||
|
CONSTRAINT `router_table_sim_objs_ObjID_fk` FOREIGN KEY (`obj_id`) REFERENCES `sim_objs` (`ObjID`) ON DELETE CASCADE ON UPDATE CASCADE
|
||||||
|
) ;
|
||||||
|
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
DROP TABLE IF EXISTS `sim_conn`;
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE `sim_conn` (
|
||||||
|
`conn_id` varchar(255) NOT NULL ,
|
||||||
|
`ConfigCorrect` int(0) NULL DEFAULT NULL ,
|
||||||
|
PRIMARY KEY (`conn_id`)
|
||||||
|
) ;
|
||||||
|
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
DROP TABLE IF EXISTS `sim_objs`;
|
||||||
|
""")
|
||||||
|
cursor.execute("""
|
||||||
|
CREATE TABLE `sim_objs` (
|
||||||
|
`ObjID` varchar(50) NOT NULL,
|
||||||
|
`ObjType` int(0) NULL DEFAULT NULL,
|
||||||
|
`ObjLabel` varchar(20) NULL DEFAULT NULL,
|
||||||
|
`ObjX` int(0) NULL DEFAULT NULL,
|
||||||
|
`ObjY` int(0) NULL DEFAULT NULL,
|
||||||
|
`ConfigCorrect` int(0) NULL DEFAULT NULL,
|
||||||
|
PRIMARY KEY (`ObjID`)
|
||||||
|
) ;
|
||||||
|
""")
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
init_database()
|
||||||
Binary file not shown.
@ -0,0 +1,9 @@
|
|||||||
|
import random
|
||||||
|
|
||||||
|
def random_mac():
|
||||||
|
# 生成六组两位的十六进制数
|
||||||
|
return ":".join(["%02x" % random.randint(0, 255) for _ in range(6)])
|
||||||
|
|
||||||
|
# 生成MAC地址
|
||||||
|
mac_address = random_mac()
|
||||||
|
print(mac_address)
|
||||||
Loading…
Reference in new issue