Repository navigation
Expand file tree
/
Copy pathgui.py
More file actions
1324 lines (1134 loc) · 55 KB
/
Copy pathgui.py
File metadata and controls
1324 lines (1134 loc) · 55 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
# gui.py
from collector import HardwareCollector
from data_handler import DataHandler
import tkinter as tk
from tkinter import ttk, messagebox, simpledialog
import threading
import os
import re
import json
import webbrowser
from datetime import datetime
def _is_lan_url(u):
"""判断一个地址是否是局域网地址(http://IP:端口)"""
return bool(re.match(r"^https?://\d{1,3}(\.\d{1,3}){3}(:\d+)?$", u or ""))
class AssetToolGUI:
def __init__(self, root):
self.root = root
# 获取 exe 所在的绝对路径,确保能读取到 brands.json
from collector import get_data_file_path
self.brands_path = get_data_file_path("brands.json")
self.root.title("计算机资产管理工具 (Pro版)")
self.root.geometry("1500x720")
# --- 1. 样式与图标设置 ---
self.setup_styles()
# --- 2. 初始化核心模块 ---
self.handler = DataHandler()
self.collector = HardwareCollector()
self.computers_data = self.handler.load_data()
# --- 3. 变量初始化 ---
self.init_variables()
# --- 4. 构建界面 ---
self.setup_ui()
self.refresh_table()
threading.Thread(target=self.auto_scan_and_prompt, daemon=True).start()
# --- 5. 初始加载 ---
# self.refresh_table()
# 绑定窗口关闭协议
self.root.protocol("WM_DELETE_WINDOW", self.on_closing)
# 启动后稍微延迟一点获取基础信息,避免界面卡顿
self.root.after(200, self.load_basic_info)
def auto_scan_and_prompt(self):
"""后台扫描 -> 更新表格 -> 自动引导录入"""
# 1. 先通过 after 通知主线程更新状态文字
self.root.after(0, lambda: self.status_var.set("🚀 正在自动扫描硬件,请稍候..."))
# 2. 执行耗时扫描 (在子线程跑,不会卡死界面)
try:
hardware_data = self.collector.get_hardware_info()
# 3. 回到主线程更新 UI 和处理弹窗
self.root.after(0, lambda: self.finish_auto_task(hardware_data))
except Exception as e:
error_msg = str(e)
self.root.after(0, lambda m=error_msg: self.status_var.set(f"❌ 扫描出错: {m}"))
def finish_auto_task(self, data):
"""当后台扫描任务彻底完成时调用"""
# 1. 更新顶部 info_vars 面板(让用户看到实时数据)
for key in self.info_vars:
if key in data:
self.info_vars[key].set(data[key])
# 2. 将新扫描的数据放入内存列表
# 这里调用写好的去重逻辑函数
self.update_table_with_data(data)
self.status_var.set("✅ 自动扫描完成,请补充资产归属")
# 3. 自动引导录入
self.prompt_user_info(data)
def prompt_user_info(self, data):
"""自动弹窗引导用户输入部门和姓名"""
from tkinter import simpledialog
# 获取计算机名作为提示
pc_name = data.get("计算机名称", "本机")
# 弹出输入框
dept = simpledialog.askstring("归档引导", "已完成查询,请输入所属【部门】:", parent=self.root)
if not dept:
self.status_var.set("⚠️ 用户取消了信息补充")
return
user = simpledialog.askstring("归档引导", f"请输入【现使用人】姓名:", parent=self.root)
if not user: return
# 1. 完善数据对象
data["部门"] = dept if dept else "-"
data["现使用人"] = user if user else "-"
data["收集时间"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
self.save_to_json()
self.refresh_table()
def init_variables(self):
"""初始化界面显示的变量"""
self.info_vars = {
"计算机名称": tk.StringVar(value="-"),
"ip地址": tk.StringVar(value="-"),
"MAC地址": tk.StringVar(value="-"),
"当前用户": tk.StringVar(value="-"),
"品牌": tk.StringVar(value="-"),
"型号": tk.StringVar(value="-"),
"设备类型": tk.StringVar(value="-"),
"序列号": tk.StringVar(value="-"),
"系统UUID": tk.StringVar(value="-"),
"主板序列号": tk.StringVar(value="-"),
"BIOS版本": tk.StringVar(value="-"),
"CPU": tk.StringVar(value="-"),
"内存": tk.StringVar(value="-"),
"内存详情": tk.StringVar(value="-"),
"硬盘": tk.StringVar(value="-"),
"硬盘健康": tk.StringVar(value="-"),
"操作系统": tk.StringVar(value="-"),
"显卡": tk.StringVar(value="-"),
"主板信息": tk.StringVar(value="-"),
}
self.status_var = tk.StringVar(value="✅ 系统就绪")
self.count_var = tk.StringVar(value="0")
self.path_var = tk.StringVar(value=f"数据存储于: {os.path.basename(self.handler.data_dir)}")
def update_table_with_data(self, new_item):
"""将扫描到的单条数据同步到表格中"""
# 1. 检查数据是否完整
if not new_item: return
# 2. 更新本地内存数据
self.computers_data.append(new_item)
# 3. 真正刷新 Treeview 控件
self.refresh_table()
def setup_styles(self):
"""配置美化样式"""
style = ttk.Style()
# Windows 默认主题
if "vista" in style.theme_names():
style.theme_use("vista")
# 字体配置
base_font = ('微软雅黑', 10)
bold_font = ('微软雅黑', 10, 'bold')
# 按钮样式
style.configure('Primary.TButton', font=bold_font, foreground="#005fb8") # 蓝色重点
style.configure('Success.TButton', font=bold_font, foreground="#2e7d32") # 绿色导出
style.configure('Danger.TButton', font=bold_font, foreground="#c62828") # 红色删除
style.configure('Normal.TButton', font=base_font)
# 表格样式
style.configure("Treeview.Heading", font=bold_font)
style.configure("Treeview", font=base_font, rowheight=25)
def setup_ui(self):
"""构建精细化布局"""
self.status_var = tk.StringVar(value="准备就绪")
status_bar = tk.Label(self.root, textvariable=self.status_var, bd=1, relief=tk.SUNKEN, anchor=tk.W)
status_bar.pack(side=tk.BOTTOM, fill=tk.X)
# self.status_label.config(text="🚀 正在自动扫描硬件,请稍候...")
# === 顶部:当前机器信息面板 ===
info_frame = ttk.LabelFrame(self.root, text="📋 本机实时状态", padding=15)
info_frame.pack(fill="x", padx=10, pady=5)
# self.tree = ttk.Treeview(self.main_frame, columns=self.columns, show='headings')
# 使用 Grid 布局模仿旧版样式的网格展示
cols = [
"计算机名称",
"ip地址",
"MAC地址",
"当前用户",
"品牌",
"型号",
"设备类型",
"序列号",
"系统UUID",
"主板序列号",
"BIOS版本",
"操作系统",
"CPU",
"内存",
"内存详情",
"硬盘",
"硬盘健康",
"显卡",
"主板信息",
]
for i, key in enumerate(cols):
row = i // 3 # 每行3个
col = (i % 3) * 2
# 标签名
ttk.Label(info_frame, text=f"{key}:", font=('微软雅黑', 10, 'bold'), foreground="#555").grid(row=row,
column=col,
sticky="e",
padx=5, pady=5)
# 值
ttk.Label(info_frame, textvariable=self.info_vars[key], foreground="#005fb8").grid(row=row, column=col + 1,
sticky="w", padx=(0, 20),
pady=5)
# === 中部:功能按钮区 ===
btn_frame = ttk.Frame(self.root)
btn_frame.pack(fill="x", padx=10, pady=10)
# 左侧操作按钮
ttk.Button(btn_frame, text="🔍 深度采集本机", command=self.run_scan_thread, style='Primary.TButton',
width=15).pack(side="left", padx=5)
ttk.Button(btn_frame, text="✏️ 手动录入信息", command=self.open_manual_input_window, style='Normal.TButton',
width=15).pack(side="left", padx=5)
# 新增:添加品牌映射按钮
ttk.Button(btn_frame, text="📝 录入型号", command=self.add_custom_brand_mapping).pack(side="left", padx=5)
ttk.Button(btn_frame, text="🏷️添加品牌", command=self.add_new_brand_only).pack(side="left", padx=5)
ttk.Button(btn_frame, text="🔍 查看品牌库", command=self.show_all_brands_window, style='Normal.TButton').pack(side="left", padx=5)
ttk.Button(btn_frame, text="🌐 批量采集", command=self.open_batch_window, style='Primary.TButton').pack(side="left", padx=5)
# 右侧功能按钮
ttk.Button(btn_frame, text="📂 打开数据目录", command=self.open_data_folder, style='Normal.TButton').pack(
side="right", padx=5)
ttk.Button(btn_frame, text="📊 导出 Excel", command=self.export_excel, style='Success.TButton').pack(
side="right", padx=5)
ttk.Button(btn_frame, text="💾 保存数据", command=self.save_to_json, style='Normal.TButton').pack(
side="right", padx=5
)
ttk.Button(btn_frame, text="✏️ 编辑选中记录", command=self.edit_record, style='Normal.TButton').pack(
side="right", padx=5
)
ttk.Button(btn_frame, text="🗑 删除选中记录", command=self.delete_record, style='Danger.TButton').pack(
side="right", padx=5
)
# === 底部:数据表格 ===
table_frame = ttk.LabelFrame(self.root, text="🗄️ 资产列表", padding=5)
table_frame.pack(fill="both", expand=True, padx=10, pady=5)
# 定义列
self.columns = (
"计算机名称",
"ip地址",
"MAC地址",
"当前用户",
"品牌",
"型号",
"设备类型",
"序列号",
"系统UUID",
"主板序列号",
"BIOS版本",
"CPU",
"内存",
"内存详情",
"硬盘",
"硬盘健康",
"显卡",
"主板信息",
"操作系统",
"部门",
"现使用人",
"收集时间",
)
self.tree = ttk.Treeview(table_frame, columns=self.columns, show="headings", selectmode="extended")
# 滚动条
vsb = ttk.Scrollbar(table_frame, orient="vertical", command=self.tree.yview)
hsb = ttk.Scrollbar(table_frame, orient="horizontal", command=self.tree.xview)
self.tree.configure(yscrollcommand=vsb.set, xscrollcommand=hsb.set)
# 布局表格
self.tree.grid(row=0, column=0, sticky="nsew")
vsb.grid(row=0, column=1, sticky="ns")
hsb.grid(row=1, column=0, sticky="ew")
table_frame.grid_rowconfigure(0, weight=1)
table_frame.grid_columnconfigure(0, weight=1)
# 设置表头
col_widths = {
"计算机名称": 120,
"ip地址": 140,
"MAC地址": 130,
"当前用户": 80,
"品牌": 80,
"型号": 160,
"设备类型": 80,
"序列号": 150,
"系统UUID": 230,
"主板序列号": 150,
"BIOS版本": 110,
"CPU": 220,
"内存": 80,
"内存详情": 180,
"硬盘": 220,
"硬盘健康": 220,
"显卡": 220,
"主板信息": 200,
"操作系统": 160,
"部门": 80,
"现使用人": 80,
"收集时间": 150,
}
for col in self.columns:
self.tree.heading(col, text=col)
self.tree.column(col, width=col_widths.get(col, 100), anchor="center")
# 绑定事件:右键菜单 + 双击编辑
self.tree.bind("<Button-3>", self.show_context_menu)
self.tree.bind("<Double-1>", self.on_row_double_click)
# === 最底部:状态栏 ===
status_bar = ttk.Frame(self.root)
status_bar.pack(fill="x", side="bottom", padx=10, pady=5)
ttk.Label(status_bar, textvariable=self.path_var, foreground="gray").pack(side="left")
ttk.Label(status_bar, textvariable=self.status_var, foreground="#2e7d32").pack(side="right", padx=10)
ttk.Label(status_bar, textvariable=self.count_var).pack(side="right")
ttk.Label(status_bar, text="总记录数: ").pack(side="right")
ttk.Button(btn_frame, text="ℹ️ 关于", command=self.show_about).pack(side="right", padx=5)
# --- 逻辑功能 ---
def load_basic_info(self):
"""加载基础信息显示在顶部面板"""
info = self.collector.get_hardware_info()
# 更新顶部面板的变量
self.info_vars["计算机名称"].set(info.get('计算机名称', '-'))
self.info_vars["ip地址"].set(info.get('ip地址', '-'))
self.info_vars["MAC地址"].set(info.get('MAC地址', '-'))
self.info_vars["当前用户"].set(info.get('当前用户', '-'))
self.info_vars["品牌"].set(info.get('品牌', '-'))
self.info_vars["型号"].set(info.get('型号', '-'))
self.info_vars["设备类型"].set(info.get('设备类型', '-'))
self.info_vars["序列号"].set(info.get('序列号', '-'))
self.info_vars["系统UUID"].set(info.get('系统UUID', '-'))
self.info_vars["主板序列号"].set(info.get('主板序列号', '-'))
self.info_vars["BIOS版本"].set(info.get('BIOS版本', '-'))
self.info_vars["CPU"].set(info.get('CPU', '-'))
self.info_vars["内存"].set(info.get('内存', '-'))
self.info_vars["内存详情"].set(info.get('内存详情', '-'))
self.info_vars["硬盘"].set(info.get('硬盘', '-'))
self.info_vars["硬盘健康"].set(info.get('硬盘健康', '-'))
self.info_vars["显卡"].set(info.get('显卡', '-'))
self.info_vars["主板信息"].set(info.get('主板信息', '-'))
def run_scan_thread(self):
"""多线程执行深度扫描"""
self.status_var.set("⏳ 正在进行深度硬件扫描...")
t = threading.Thread(target=self._scan_logic, daemon=True)
t.start()
def _scan_logic(self):
try:
# 1. 采集
data = self.collector.get_hardware_info()
# 2. 补充默认空字段
data["部门"] = "-"
data["现使用人"] = "-"
data["收集时间"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
# 4. 更新UI
self.root.after(0, self.on_scan_complete, data)
except Exception as e:
self.root.after(0, lambda: messagebox.showerror("错误", str(e)))
def detect_brand(self, model):
"""简单的品牌推断逻辑"""
model = model.lower()
if "lenovo" in model or "think" in model: return "联想"
if "dell" in model or "optiplex" in model: return "戴尔"
if "hp" in model or "probook" in model: return "惠普"
if "asus" in model: return "华硕"
return "其他"
def on_scan_complete(self, new_data):
# 更新顶部面板的详细信息
for k, v in new_data.items():
if k in self.info_vars:
self.info_vars[k].set(v)
# 存入列表
self.computers_data.append(new_data)
self.refresh_table()
self.status_var.set("✅ 采集成功!记得保存数据")
# 自动弹窗询问部门
if messagebox.askyesno("采集完成", "是否立即补充【部门】和【使用人】信息?"):
self.tree.selection_set(self.tree.get_children()[-1]) # 选中最后一行
self.add_manual_info()
def open_manual_input_window(self):
"""打开手动录入窗口 (功能复刻)"""
top = tk.Toplevel(self.root)
top.title("手动录入资产信息")
top.geometry("430x1040")
top.attributes("-topmost", True) # 1. 强制窗口显示在所有窗口的最前面
top.grab_set() # 2. 锁定焦点,此时用户无法点击主界面,必须先处理这个窗口
top.focus_set() # 3. 自动将输入焦点移动到这个新窗口
# --- 核心改进:先抓取一次硬件信息作为底稿 ---
try:
base_data = self.collector.get_hardware_info()
except:
base_data = {}
entries = {}
fields = [
("计算机名称", base_data.get("计算机名称", "")),
("ip地址", base_data.get("ip地址", "")),
("MAC地址", base_data.get("MAC地址", "")),
("当前用户", base_data.get("当前用户", "")),
("品牌", base_data.get("品牌", "")),
("型号", base_data.get("型号", "")),
("设备类型", base_data.get("设备类型", "")),
("序列号", base_data.get("序列号", "")),
("系统UUID", base_data.get("系统UUID", "")),
("CPU", base_data.get("CPU", "")),
("内存", base_data.get("内存", "")),
("内存详情", base_data.get("内存详情", "")),
("硬盘", base_data.get("硬盘", "")),
("硬盘详情", base_data.get("硬盘详情", "")),
("显卡", base_data.get("显卡", "")),
("主板信息", base_data.get("主板信息", "")),
("部门", ""),
("现使用人", "")
]
for i, (label_text, default_val) in enumerate(fields):
ttk.Label(top, text=f"{label_text}:").pack(anchor="w", padx=30, pady=(10, 0))
e = ttk.Entry(top, width=40)
e.insert(0, str(default_val)) # 预填数据
e.pack(padx=30, pady=5)
entries[label_text] = e
def submit():
# 整合数据
new_record = base_data.copy() # 以硬件信息为基础
for label_text, entry in entries.items():
new_record[label_text] = entry.get()
# 补全其他信息
new_record["收集时间"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
# 确保硬件字段不会显示“手动录入”
for col in [
"CPU",
"内存",
"内存详情",
"硬盘",
"硬盘详情",
"显卡",
"主板信息",
"操作系统",
"MAC地址",
"ip地址",
]:
if not new_record.get(col) or new_record.get(col) == "未知":
new_record[col] = base_data.get(col, "采集失败")
self.computers_data.append(new_record)
self.refresh_table()
top.destroy()
messagebox.showinfo("成功", "记录已添加至列表")
ttk.Button(top, text="确认提交", command=submit).pack(pady=25)
def refresh_table(self):
self.collector.brands_db.reload()
# 清空
for item in self.tree.get_children():
self.tree.delete(item)
# 重新填充
for data in self.computers_data:
values = [data.get(col, "") for col in self.columns]
self.tree.insert("", "end", values=values)
self.count_var.set(str(len(self.computers_data)))
def save_to_json(self):
if self.handler.save_data(self.computers_data):
self.status_var.set(f"✅ 数据已保存 ({datetime.now().strftime('%H:%M:%S')})")
messagebox.showinfo("保存成功", "数据已成功写入 JSON 文件")
return True
else:
messagebox.showerror("错误", "保存失败")
return False
def export_excel(self):
success, msg = self.handler.export_to_excel(self.computers_data)
if success:
if messagebox.askyesno("导出成功", f"文件位于:\n{msg}\n\n是否立即打开?"):
os.startfile(msg)
else:
messagebox.showerror("导出失败", msg)
def open_data_folder(self):
"""打开数据存储文件夹"""
path = self.handler.data_dir
os.startfile(path)
# --- 右键菜单功能 ---
def show_context_menu(self, event):
item = self.tree.identify_row(event.y)
if item:
self.tree.selection_set(item)
menu = tk.Menu(self.root, tearoff=0)
menu.add_command(label="✏️ 编辑该条记录", command=self.edit_record)
menu.add_command(label="📝 修改部门/人员信息", command=self.add_manual_info)
menu.add_separator()
menu.add_command(label="❌ 删除该条记录", command=self.delete_record)
menu.post(event.x_root, event.y_root)
def add_manual_info(self):
selected = self.tree.selection()
if not selected: return
dept = simpledialog.askstring("补充信息", "请输入部门名称:",parent=self.root)
user = simpledialog.askstring("补充信息", "请输入现使用人:",parent=self.root)
idx = self.tree.index(selected[0])
if dept: self.computers_data[idx]["部门"] = dept
if user: self.computers_data[idx]["现使用人"] = user
self.refresh_table()
def delete_record(self):
selected = self.tree.selection()
if selected:
if messagebox.askyesno("确认", "确定要删除这条记录吗?"):
idx = self.tree.index(selected[0])
del self.computers_data[idx]
self.refresh_table()
def on_row_double_click(self, event):
"""双击表格行,直接进入编辑模式"""
item = self.tree.identify_row(event.y)
if not item:
return
self.tree.selection_set(item)
self.edit_record()
def edit_record(self):
"""编辑当前选中的整条记录(所有字段)"""
selected = self.tree.selection()
if not selected:
messagebox.showinfo("提示", "请先选中一条需要编辑的记录。")
return
idx = self.tree.index(selected[0])
base_data = dict(self.computers_data[idx]) # 复制原始数据,避免直接修改引用
top = tk.Toplevel(self.root)
top.title("编辑资产记录")
# 稍微加高一点,保证字段完整显示
top.geometry("430x820")
top.attributes("-topmost", True)
top.grab_set()
top.focus_set()
entries = {}
# 按列顺序展示可编辑字段
edit_fields = [
"计算机名称",
"ip地址",
"MAC地址",
"当前用户",
"品牌",
"型号",
"CPU",
"内存",
"硬盘",
"显卡",
"主板信息",
"操作系统",
"部门",
"现使用人",
]
for i, key in enumerate(edit_fields):
ttk.Label(top, text=f"{key}:").pack(anchor="w", padx=30, pady=(8, 0))
e = ttk.Entry(top, width=40)
e.insert(0, str(base_data.get(key, "")))
e.pack(padx=30, pady=3)
entries[key] = e
def submit_edit():
# 以原始记录为基础更新
new_record = base_data.copy()
for key, entry in entries.items():
new_record[key] = entry.get()
# 写回列表并刷新
self.computers_data[idx] = new_record
self.refresh_table()
top.destroy()
messagebox.showinfo("成功", "记录已更新")
ttk.Button(top, text="保存修改", command=submit_edit).pack(pady=20)
def show_about(self):
"""弹出关于窗口"""
about_win = tk.Toplevel(self.root)
about_win.title("关于软件")
about_win.geometry("400x250")
about_win.resizable(False, False)
# 强制该窗口置顶,处理完才能回主界面
about_win.grab_set()
# --- 1. 中间的主体文案 (左对齐,正常显示) ---
main_text = (
"🚀 计算机资产管理工具 Pro\n"
"版本:v1.0.0\n\n"
"本工具旨在帮助运维人员快速采集硬件指纹,\n"
"支持一键导出,让资产盘点不再是苦差事.\n"
"“代码是冷的,但解决问题的成就感是热的。”\n"
"SUI缘儿科技:一个爱分享的小小白,你的点赞是我最大的动力"
)
ttk.Label(about_win, text=main_text, padding=20, justify="left").pack(fill="both", expand=True)
# --- 2. 右下角的蓝色文字 (你的感悟和创作信息) ---
# 创建一个容器,专门用来放右下角的字
footer_frame = ttk.Frame(about_win)
footer_frame.pack(side="bottom", fill="x", padx=15, pady=10)
my_words = (
"—— 一个异想天开的低层IT运维工程师 🛠️\n"
"Created by [SUI缘儿科技] @ 2026"
)
# foreground="blue" 设置蓝色,anchor="e" 设置靠右 (East)
personal_label = tk.Label(
footer_frame,
text=my_words,
fg="#005fb8", # 蓝色 (可以微调色号)
font=("微软雅黑", 9, "italic"),
justify="right"
)
personal_label.pack(side="right") # 靠右排列
# --- 批量采集(中心服务器 + 轻量 agent)---
def _detect_lan_ip(self):
"""获取本机局域网 IP,用于局域网模式下分发给其它电脑"""
try:
import socket
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.connect(("8.8.8.8", 80))
ip = s.getsockname()[0]
s.close()
return ip
except Exception:
return "127.0.0.1"
def open_batch_window(self):
"""打开批量采集中心窗口:局域网/互联网双模式、端口可配、预设地址、生成自包含部署包"""
from server import BatchServer, list_reports
top = tk.Toplevel(self.root)
top.title("🌐 批量采集中心")
top.geometry("660x640")
top.transient(self.root)
if not hasattr(self, "batch_server") or self.batch_server is None:
self.batch_server = BatchServer(port=8000)
if not hasattr(self, "batch_mode"):
self.batch_mode = "lan"
if not hasattr(self, "batch_port"):
self.batch_port = "8000"
if not hasattr(self, "saved_addresses"):
self.saved_addresses = self._load_saved_addresses()
lan_ip = self._detect_lan_ip()
mode_var = tk.StringVar(value=self.batch_mode)
port_var = tk.StringVar(value=self.batch_port)
internet_url_var = tk.StringVar(value="")
addr_combo_var = tk.StringVar()
tk.Label(top, text="中心采集服务器", font=("微软雅黑", 13, "bold")).pack(anchor="w", padx=16, pady=(14, 4))
# 模式选择
mode_frame = ttk.LabelFrame(top, text="采集模式")
mode_frame.pack(fill="x", padx=16, pady=(4, 2))
ttk.Radiobutton(mode_frame, text="局域网(服务器与目标机同一网络,零配置)",
variable=mode_var, value="lan", command=lambda: on_mode_or_cfg()).pack(anchor="w", padx=8, pady=2)
ttk.Radiobutton(mode_frame, text="互联网(服务器有公网IP/域名,可跨网络采集)",
variable=mode_var, value="internet", command=lambda: on_mode_or_cfg()).pack(anchor="w", padx=8, pady=2)
# 端口 + 互联网地址
cfg_frame = ttk.Frame(top)
cfg_frame.pack(fill="x", padx=16, pady=4)
tk.Label(cfg_frame, text="服务器端口:").pack(side="left")
port_entry = ttk.Entry(cfg_frame, textvariable=port_var, width=10)
port_entry.pack(side="left", padx=4)
ttk.Button(cfg_frame, text="应用端口", width=9,
command=lambda: (apply_port() and None, refresh_status(), refresh_preview())).pack(side="left", padx=2)
tk.Label(cfg_frame, text=" 公网/服务器地址:").pack(side="left")
url_entry = ttk.Entry(cfg_frame, textvariable=internet_url_var, width=34)
url_entry.pack(side="left", fill="x", expand=True)
# 输入框改动时实时刷新预览(并清除预设选择,因为手动编辑优先)
port_entry.bind("<KeyRelease>", lambda e: (addr_combo_var.set(""), refresh_preview()))
url_entry.bind("<KeyRelease>", lambda e: (addr_combo_var.set(""), refresh_preview()))
# 预设地址(可保存多个、下拉选择)
addr_frame = ttk.Frame(top)
addr_frame.pack(fill="x", padx=16, pady=2)
tk.Label(addr_frame, text="预设地址:").pack(side="left")
addr_combo = ttk.Combobox(addr_frame, textvariable=addr_combo_var, width=40,
values=self.saved_addresses, state="normal")
addr_combo.pack(side="left", padx=4)
ttk.Button(addr_frame, text="应用选中", command=lambda: apply_preset()).pack(side="left", padx=2)
ttk.Button(addr_frame, text="存为预设", command=lambda: save_preset()).pack(side="left", padx=2)
ttk.Button(addr_frame, text="删除", command=lambda: del_preset()).pack(side="left", padx=2)
url_preview_var = tk.StringVar()
tk.Label(top, textvariable=url_preview_var, fg="#005fb8", font=("微软雅黑", 10)).pack(anchor="w", padx=16, pady=2)
# 部署包选项(部门/使用人,可选)
opt_frame = ttk.LabelFrame(top, text="部署包选项(部门/使用人可选)")
opt_frame.pack(fill="x", padx=16, pady=4)
add_meta_var = tk.BooleanVar(value=False)
ttk.Checkbutton(
opt_frame,
text="生成部署包时填写部门/使用人(勾选后弹窗填写)",
variable=add_meta_var,
).pack(side="left", padx=4)
tk.Label(opt_frame, text="(不勾选则部署包不含部门/使用人,可之后在软件里补填)",
fg="#888").pack(side="left", padx=4)
def current_server_url():
port = port_var.get().strip() or "8000"
if mode_var.get() == "lan":
return f"http://{lan_ip}:{port}"
u = internet_url_var.get().strip()
if not u:
return ""
if not u.lower().startswith("http"):
u = "http://" + u
return u.rstrip("/")
def get_target_url():
"""优先使用用户在“预设地址”框里直接输入/选择的地址;否则按上方端口/公网地址推导"""
typed = (addr_combo_var.get() or "").strip()
if typed:
if typed.lower().startswith("http"):
return typed.rstrip("/")
if re.match(r"^\d{1,3}(\.\d{1,3}){3}(:\d+)?$", typed):
return "http://" + typed
return current_server_url()
def refresh_preview():
u = get_target_url()
url_preview_var.set(f"目标机将上报到:{u}" if u else "请填写服务器地址(端口或公网地址)")
def on_mode_or_cfg():
addr_combo_var.set("")
refresh_preview()
def apply_preset():
v = get_target_url()
if not v:
messagebox.showinfo("提示", "请先在预设地址框输入或选择,或上方填写端口/公网地址")
return
if _is_lan_url(v):
mode_var.set("lan")
m = re.search(r":(\d+)", v)
port_var.set(m.group(1) if m else "8000")
internet_url_var.set("")
else:
mode_var.set("internet")
internet_url_var.set(v)
addr_combo_var.set(v)
refresh_preview()
def save_preset():
u = get_target_url()
if not u:
messagebox.showinfo("提示", "请先在预设地址框输入或选择,或上方填写端口/公网地址")
return
if u not in self.saved_addresses:
self.saved_addresses.append(u)
self._save_saved_addresses()
addr_combo["values"] = self.saved_addresses
addr_combo_var.set(u)
messagebox.showinfo("已保存", f"已把以下地址加入预设:\n{u}")
def del_preset():
v = addr_combo_var.get().strip()
if v in self.saved_addresses:
self.saved_addresses.remove(v)
self._save_saved_addresses()
addr_combo["values"] = self.saved_addresses
addr_combo_var.set("")
refresh_preview()
def apply_port():
try:
port = int(port_var.get().strip() or "8000")
except ValueError:
port = 8000
was_running = self.batch_server.is_running()
if was_running:
self.batch_server.stop()
self.batch_server = BatchServer(port=port)
self.batch_port = str(port)
self.batch_mode = mode_var.get()
return was_running
def save_settings():
was_running = apply_port()
if was_running:
ok, msg = self.batch_server.start()
else:
msg = "已保存服务器设置(当前未启动,点『启动服务器』生效)。"
refresh_status()
u = current_server_url()
messagebox.showinfo(
"已保存",
f"服务器设置已保存。\n端口:{self.batch_port}\n"
f"当前服务器地址:{u or '(互联网模式需填写公网地址)'}",
)
# 启停服务器
frm = ttk.Frame(top)
frm.pack(fill="x", padx=16, pady=6)
status_var = tk.StringVar(value="服务器未启动")
tk.Label(frm, textvariable=status_var, fg="#555").pack(side="left")
toggle_btn = ttk.Button(frm, text="▶ 启动服务器")
def refresh_status():
if self.batch_server.is_running():
status_var.set(f"● 运行中:{self.batch_server.url()}(其它机器访问 {current_server_url()})")
toggle_btn.config(text="■ 停止服务器")
else:
status_var.set("服务器未启动")
toggle_btn.config(text="▶ 启动服务器")
def toggle():
was_running = apply_port()
if was_running:
self.batch_server.stop()
else:
ok, msg = self.batch_server.start()
status_var.set(msg)
refresh_status()
toggle_btn.config(command=toggle)
toggle_btn.pack(side="right")
ttk.Button(frm, text="💾 保存设置", command=save_settings).pack(side="right", padx=6)
tip = ("使用方法:\n"
"1) 选模式、填端口(互联网模式还需填服务器公网地址),点【保存设置】。\n"
"2) 点【生成部署包】→ 得到 agent_deploy 文件夹(含编译好的 agent.exe,目标机无需装任何环境)。\n"
"3) 把该文件夹拷到每台目标机,双击 agent.exe 即自动采集并上报。\n"
"提示:常用地址可点『存为预设』,下次下拉选择即可,无需重复输入。\n"
"局域网:同 WiFi/网段即可;互联网:服务器需有公网IP或域名(路由器端口映射或云服务器部署 server.py)。")
tk.Label(top, text=tip, justify="left", wraplength=620, fg="#444",
font=("微软雅黑", 10)).pack(anchor="w", padx=16, pady=6)
bf = ttk.Frame(top)
bf.pack(fill="x", padx=16, pady=6)
ttk.Button(bf, text="🌐 打开仪表盘", style="Primary.TButton",
command=lambda: webbrowser.open(self.batch_server.url())).pack(side="left", padx=4)
ttk.Button(bf, text="📥 导入已上报数据", style="Normal.TButton",
command=lambda: self.import_reports(top)).pack(side="left", padx=4)
ttk.Button(bf, text="🔄 从服务器同步", style="Normal.TButton",
command=lambda: self.sync_from_server(top)).pack(side="left", padx=4)
def on_generate():
url = get_target_url()
if not url:
messagebox.showerror("错误", "请先在上方设置并保存服务器地址(或在预设地址框输入),再生成部署包。")
return
dept = user = ""
if add_meta_var.get():
d = simpledialog.askstring("部门", "请输入部署包默认的【部门】:", parent=top)
if d is None:
return # 用户取消
u = simpledialog.askstring("使用人", "请输入部署包默认的【使用人】:", parent=top)
if u is None:
return
dept, user = d.strip(), u.strip()
# 不勾选 → 部署包不含部门/使用人,且运行 agent 时也不弹窗询问
no_meta = not add_meta_var.get()
self.generate_agent_package(top, url, dept, user, no_meta=no_meta)
ttk.Button(bf, text="📦 一键生成部署包", style="Normal.TButton",
command=on_generate).pack(side="left", padx=4)
cnt_var = tk.StringVar(value="已上报机器数:0")
tk.Label(top, textvariable=cnt_var, fg="#2e7d32", font=("微软雅黑", 11, "bold")).pack(anchor="w", padx=16, pady=6)
def update_cnt():
try:
cnt_var.set(f"已上报机器数:{len(list_reports())}")
except Exception:
cnt_var.set("已上报机器数:?")
refresh_preview()
refresh_status()
update_cnt()
def tick():
update_cnt()
top.after(3000, tick)
top.after(3000, tick)
def import_reports(self, parent=None):
"""把服务器收到的上报数据导入到主资产列表(按计算机名称去重/更新)"""
from server import list_reports
try:
reports = list_reports()
except Exception as e:
messagebox.showerror("错误", str(e))
return
if not reports:
messagebox.showinfo("提示", "暂无上报数据。请先启动服务器,并让其它电脑运行 agent。")
return
existing = {d.get("计算机名称") for d in self.computers_data}
updated = 0
for r in reports:
name = r.get("计算机名称")
if name in existing:
for i, d in enumerate(self.computers_data):
if d.get("计算机名称") == name:
self.computers_data[i] = r
updated += 1
break
else:
self.computers_data.append(r)
existing.add(name)
self.refresh_table()
messagebox.showinfo("完成", f"已汇总 {len(reports)} 台机器的上报数据(其中 {updated} 台为覆盖更新)。")
def sync_from_server(self, parent=None):
"""从远程中心服务器(HTTP)拉取已上报的机器列表,合并进本机资产列表"""
from urllib.request import urlopen, Request
import json as _json
default = self.batch_server.url() if (hasattr(self, "batch_server") and self.batch_server.is_running()) else ""
url = simpledialog.askstring(
"从服务器同步",
"请输入中心服务器地址(含端口):\n例如 http://192.168.1.10:8000 或 http://your.domain:8000",
initialvalue=default, parent=parent or self.root,
)
if not url:
return
api = url.strip().rstrip("/") + "/api/machines"
try:
req = Request(api, headers={"Accept": "application/json"})
with urlopen(req, timeout=15) as resp:
machines = _json.loads(resp.read().decode("utf-8"))
except Exception as e:
messagebox.showerror(
"同步失败",
f"无法连接服务器:\n{e}\n\n请确认:服务器已启动、地址/端口正确、防火墙已放行该端口。",
)
return
if not isinstance(machines, list) or not machines:
messagebox.showinfo("提示", "服务器暂无上报数据。")
return
existing = {d.get("计算机名称") for d in self.computers_data}
added = updated = 0
for m in machines:
name = m.get("计算机名称")
if not name:
continue
if name in existing:
for i, d in enumerate(self.computers_data):