import os import tkinter import tkinter.font import tkinter.ttk from datetime import datetime, timedelta from pathlib import Path from typing import Optional import tkcalendar from decrypter.db_decrypt import DatabaseDecrypter from decrypter.image_decrypt import ImageDecrypter from decrypter.video_decrypt import VideoDecrypter from gui.auto_scroll_guide import AutoScrollGuide from gui.auto_scrolls_single_guide import AutoScrollSingleGuide from gui.tool_tip import ToolTip from entity.contact import Contact from exporter.html_exporter import HtmlExporter from gui.listbox_with_search import ListboxWithSearch class Gui: def __init__(self): self.restart_note1 = None self.restart_note2 = None self.auto_scroll_single_guide = None self.auto_scroll_guide = None self.auto_scroll_frame = None self.search_username = None self.open_search_guide_image = None self.open_search_guide = None self.auto_scroll_button_single = None self.auto_scroll_button_single_text = None self.convert_video = None self.convert_video_var = None self.html_exporter_thread = None self.confirm_button_text = None self.succeed_label_2 = None self.succeed_label = None self.auto_scroll_button_text = None self.warning_label = None self.root = None self.waiting_label = None self.listbox = None self.begin_calendar = None self.end_calendar = None self.end_calendar_label = None self.begin_calendar_label = None self.confirm_button = None self.decrypt_progressbar = None self.export_progressbar = None self.next_step_button = None self.decrypter = None self.auto_scroll_button = None self.auto_scrolling_thread = None self.decrypt_note = None self.decrypt_note_text = None self.account_info = None self.video_decrypter = None self.image_decrypter = None self.export_dir_name = None self.exporting = False # 1: 自动滚动数据 2: 解密数据库 3: 导出 self.page_stage = 1 def run_gui(self): self.root = tkinter.Tk() self.root.geometry('650x650') self.root.title('朋友圈导出') self.waiting_label = tkinter.ttk.Label(self.root, text="正在连接微信....", font=("微软雅黑", 16), anchor='center') self.waiting_label.place(relx=0.5, rely=0.05, anchor='center') self.root.mainloop() def wechat_logged_in(self, account_info): self.account_info = account_info self.waiting_label.config(text="微信已登录") self.auto_scroll_button_text = tkinter.StringVar() self.auto_scroll_button_text.set("自动浏览全部朋友圈") self.auto_scroll_button = tkinter.ttk.Button(self.root, textvariable=self.auto_scroll_button_text, command=self.open_auto_scroll_guide) self.auto_scroll_button.place(relx=0.35, rely=0.15, anchor='center') self.auto_scroll_button_single_text = tkinter.StringVar() self.auto_scroll_button_single_text.set("自动浏览单个朋友") self.auto_scroll_button_single = tkinter.ttk.Button(self.root, textvariable=self.auto_scroll_button_single_text, command=self.switch_auto_scroll_single) self.auto_scroll_button_single.place(relx=0.655, rely=0.15, anchor='center') self.next_step_button = tkinter.ttk.Button(self.root, text="下一步", command=self.next_step) self.next_step_button.place(relx=0.65, rely=0.8) def open_auto_scroll_guide(self): if self.auto_scroll_single_guide and self.auto_scroll_single_guide.frame: self.auto_scroll_single_guide.frame.place_forget() self.auto_scroll_guide = AutoScrollGuide(self.root) self.auto_scroll_guide.frame.place(relx=0.5, rely=0.5, anchor='center') def switch_auto_scroll_single(self): if self.auto_scroll_guide and self.auto_scroll_guide.frame: self.auto_scroll_guide.frame.place_forget() self.auto_scroll_single_guide = AutoScrollSingleGuide(self.root) self.auto_scroll_single_guide.frame.place(relx=0.5, rely=0.5, anchor='center') def next_step(self): if self.page_stage == 1: if self.auto_scroll_guide and self.auto_scroll_guide.auto_thread: self.auto_scroll_guide.auto_thread.set_scrolling(False) if self.auto_scroll_single_guide and self.auto_scroll_single_guide.auto_thread: self.auto_scroll_guide.auto_thread.set_scrolling(False) self.auto_scroll_button.place_forget() self.auto_scroll_button_single.place_forget() if self.auto_scroll_guide and self.auto_scroll_guide.frame: self.auto_scroll_guide.frame.place_forget() if self.auto_scroll_single_guide and self.auto_scroll_single_guide.frame: self.auto_scroll_single_guide.frame.place_forget() self.restart_note1 = tkinter.Label(self.root, text="请关闭微信客户端", fg="red") self.restart_note1.place(relx=0.5, rely=0.2, anchor='center') self.restart_note2 = tkinter.Label(self.root, text="然后点击下一步") self.restart_note2.place(relx=0.5, rely=0.3, anchor='center') if self.page_stage == 2: self.restart_note1.place_forget() self.restart_note2.place_forget() self.waiting_label.place_forget() self.decrypter = DatabaseDecrypter(self, self.account_info.get("filePath"), self.account_info.get("key")) self.decrypt_note_text = tkinter.StringVar() self.decrypt_note_text.set("正在复制数据.....") self.decrypt_note = tkinter.Label(self.root, textvariable=self.decrypt_note_text) self.decrypt_note.place(relx=0.5, rely=0.2, anchor='center') self.decrypt_progressbar = tkinter.ttk.Progressbar(self.root) self.decrypt_progressbar.place(relx=0.5, rely=0.3, anchor='center') # 进度值最大值 self.decrypt_progressbar['maximum'] = 100 # 进度值初始值 self.decrypt_progressbar['value'] = 0 # 解密过程禁用下一步按钮 self.next_step_button.config(state=tkinter.DISABLED) self.decrypter.decrypt() if self.page_stage == 3: self.decrypt_note.place_forget() self.decrypt_progressbar.place_forget() self.init_export_page() # 不再有下一步按钮 self.next_step_button.place_forget() # 初始化视频导出器 self.video_decrypter = VideoDecrypter(self, self.account_info.get("filePath")) # 初始化图片导出器 self.image_decrypter = ImageDecrypter(self, self.account_info.get("filePath")) self.page_stage = self.page_stage + 1 def init_export_page(self): from app.DataBase import micro_msg_db contact_datas = micro_msg_db.get_contact() contacts = [] for c in contact_datas: contact = Contact(c[0], c[1], c[2], c[3], c[4], c[5], c[6], c[7], c[8], c[9], c[10], c[11]) contacts.append(contact) def validate_contact(this_contact: Contact): c_type = this_contact.type user_name: str = this_contact.userName # 不是其他号码 is_misc_account = c_type == 1 or c_type == 33 or c_type == 513 # 不是公众号 is_gh_account = user_name.startswith("gh_") # 不是聊天群 is_chatroom = user_name.endswith("@chatroom") # 不是文件传输助手 is_filehelper = this_contact.userName == "filehelper" return (not is_misc_account) and (not is_gh_account) and (not is_chatroom) and (not is_filehelper) filtered = filter(validate_contact, contacts) contacts = list(filtered) self.listbox = ListboxWithSearch(self.root, contacts) self.listbox.frame.place(relx=0.05, rely=0.03) self.begin_calendar_label = tkinter.ttk.Label(text="开始日期") self.begin_calendar_label.place(relx=0.65, rely=0.15) # 默认开始时间是100天前 current_date = datetime.now() half_year_ago = current_date - timedelta(days=100) self.begin_calendar = tkcalendar.DateEntry(master=self.root, locale="zh_CN", year=half_year_ago.year, month=half_year_ago.month, day=half_year_ago.day, maxdate=datetime.now()) self.begin_calendar.place(relx=0.65, rely=0.2) self.end_calendar_label = tkinter.ttk.Label(text="截止日期") self.end_calendar_label.place(relx=0.65, rely=0.25) self.end_calendar = tkcalendar.DateEntry(master=self.root, locale="zh_CN", maxdate=datetime.now()) self.end_calendar.place(relx=0.65, rely=0.3) self.convert_video_var = tkinter.IntVar(value=0) self.convert_video = tkinter.ttk.Checkbutton(self.root, text='视频转码', variable=self.convert_video_var) self.convert_video.place(relx=0.65, rely=0.45) ToolTip(self.convert_video, "视频原始格式为H265,只支持\nChrome浏览器播放，勾选后\n将视频转码为H264,支持大\n部分浏览器，但导出速度变慢") self.confirm_button_text = tkinter.StringVar() self.confirm_button_text.set("开始导出") self.confirm_button = tkinter.ttk.Button(self.root, textvariable=self.confirm_button_text, command=self.confirm_export) self.confirm_button.place(relx=0.65, rely=0.6) # 导出成功的提示 self.succeed_label = tkinter.Label(self.root, text="导出结束") self.succeed_label_2 = tkinter.Label(self.root, text="打开文件夹", fg="#0000FF", cursor="hand2") self.succeed_label_2.bind("", self.open_target_folder) # 进度条 self.export_progressbar = tkinter.ttk.Progressbar(self.root, length=150) def confirm_export(self): if self.html_exporter_thread and not self.html_exporter_thread.stop_flag: self.html_exporter_thread.stop() else: if not self.warning_label: self.warning_label = tkinter.Label(self.root, fg="red") self.warning_label.place(relx=0.65, rely=0.55) self.warning_label.config(text="") contacts = self.listbox.get_contacts() if not contacts: self.warning_label.config(text=f"请选择至少一个联系人") return if self.begin_calendar.get_date() > self.end_calendar.get_date(): self.warning_label.config(text=f"开始时间必须小于截止时间") return self.export_progressbar.place(relx=0.64, rely=0.68) # 进度值最大值 self.export_progressbar['maximum'] = 100 # 进度值初始值 self.export_progressbar['value'] = 0 current_time = datetime.now() self.export_dir_name = current_time.strftime("%Y_%m_%d_%H%M%S") contact_map = {contact.userName: contact for contact in contacts} self.confirm_button_text.set("停止导出") self.succeed_label.place_forget() self.succeed_label_2.place_forget() # 导出线程 self.html_exporter_thread = HtmlExporter(self, self.export_dir_name, contact_map, self.begin_calendar.get_date(), self.end_calendar.get_date(), self.convert_video_var.get()) self.html_exporter_thread.start() def update_decrypt_progressbar(self, progress): self.decrypt_progressbar['value'] = progress self.root.update() def update_export_progressbar(self, progress): self.export_progressbar['value'] = progress self.root.update() def export_succeed(self): self.confirm_button_text.set("开始导出") self.succeed_label.place(relx=0.64, rely=0.75) self.succeed_label_2.place(relx=0.76, rely=0.75) def open_target_folder(self, event): folder_path = Path(f"output/{self.export_dir_name}/") # 转换为绝对路径 absolute_path = folder_path.resolve() os.startfile(absolute_path)