81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
import tkinter as tk
|
|
from entity.contact import Contact
|
|
|
|
|
|
class ListboxWithSearch:
|
|
|
|
def __init__(self, root, contacts: list[Contact]):
|
|
|
|
# key index(在控件里的编号) value Contact
|
|
self.index_contact_map = {}
|
|
|
|
self.frame = tk.LabelFrame(root, text="请选择导出联系人")
|
|
self.tool_frame = tk.Frame(self.frame)
|
|
self.tool_frame.pack()
|
|
|
|
self.search_label = tk.Label(self.tool_frame, text="搜索：")
|
|
self.search_label.pack(side='left')
|
|
|
|
self.re = tk.Entry(self.tool_frame, width=10)
|
|
self.re.pack(side='left')
|
|
self.re.bind("<KeyRelease>", self.filter)
|
|
|
|
self.select_all_button = tk.Button(self.tool_frame, text="全选", command=self.select_all)
|
|
self.select_all_button.pack(side='left', padx="10")
|
|
|
|
self.invert_select_button = tk.Button(self.tool_frame, text="反选", command=self.invert_select)
|
|
self.invert_select_button.pack(side='left', padx="2")
|
|
|
|
self.scrollbar = tk.Scrollbar(self.frame)
|
|
self.scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
|
|
|
|
self.lb = tk.Listbox(self.frame, selectmode='multiple', height=20, width=30, exportselection=False,
|
|
yscrollcommand=self.scrollbar.set)
|
|
self.scrollbar.config(command=self.lb.yview)
|
|
|
|
self.lb.bind('<<ListboxSelect>>', self.on_select)
|
|
|
|
self.lb.pack()
|
|
self.contacts = contacts
|
|
|
|
|
|
for index, contact in enumerate(contacts):
|
|
text = f'{contact.nickName}({contact.remark})' if contact.remark else f'{contact.nickName}'
|
|
self.lb.insert(index, text)
|
|
self.index_contact_map[index] = contact
|
|
|
|
def select_all(self, event=None):
|
|
for index in self.index_contact_map.keys():
|
|
self.lb.select_set(index)
|
|
self.on_select()
|
|
|
|
def on_select(self, event=None):
|
|
selection = self.lb.curselection()
|
|
self.frame.config(text=f"已选择{len(selection)}个联系人")
|
|
|
|
def invert_select(self, event=None):
|
|
"""反选"""
|
|
selected = self.lb.curselection()
|
|
for index in self.index_contact_map.keys():
|
|
if index in selected:
|
|
self.lb.selection_clear(index)
|
|
else:
|
|
self.lb.select_set(index)
|
|
self.on_select()
|
|
|
|
def get_contacts(self, event=None):
|
|
contacts = []
|
|
selected = self.lb.curselection()
|
|
for index in selected:
|
|
contacts.append(self.index_contact_map.get(index))
|
|
return contacts
|
|
|
|
def filter(self, event=None):
|
|
p = self.re.get()
|
|
if p:
|
|
for index, contact in self.index_contact_map.items():
|
|
text = f'{contact.nickName}({contact.remark})' if contact.remark else f'{contact.nickName}'
|
|
if p in text:
|
|
self.lb.yview(index)
|
|
break
|