""" 作者：太一微信：taiyi1224 邮箱：shuobo1224@qq.com """ import tkinter as tk from tkinter import ttk, messagebox, filedialog from tkinter import TclError import math from template_presets import TemplatePresetManager class TemplateDialog(tk.Toplevel): """模板选择对话框""" def __init__(self, parent, current_template): super().__init__(parent) self.title("模板设置") self.parent = parent self.current_template = current_template self.result = None # 初始化预设管理器 self.preset_manager = TemplatePresetManager() # 确保对话框模态 self.transient(parent) self.grab_set() # 设置布局 frame = ttk.Frame(self, padding=10) frame.pack(fill=tk.BOTH, expand=True) # 预设管理 preset_frame = ttk.LabelFrame(frame, text="模板预设", padding=5) preset_frame.grid(row=0, column=0, columnspan=3, sticky=tk.EW, pady=(0, 10)) ttk.Label(preset_frame, text="选择预设:").grid(row=0, column=0, sticky=tk.W) self.preset_var = tk.StringVar() # 默认空字符串，不选中任何预设 self.preset_combo = ttk.Combobox(preset_frame, textvariable=self.preset_var, width=30) self.preset_combo.grid(row=0, column=1, padx=5) self.update_preset_list() ttk.Button(preset_frame, text="加载", command=self.load_selected_preset).grid(row=0, column=2, padx=5) ttk.Button(preset_frame, text="保存当前为预设", command=self.save_current_as_preset).grid(row=0, column=3, padx=5) ttk.Button(preset_frame, text="删除选中预设", command=self.delete_selected_preset).grid(row=0, column=4, padx=5) # 比例选择 ttk.Label(frame, text="选择比例:").grid(row=1, column=0, sticky=tk.W, pady=5) self.ratio_var = tk.StringVar(value=f"{current_template.ratio[0]}:{current_template.ratio[1]}") ratio_frame = ttk.Frame(frame) ratio_frame.grid(row=1, column=1, sticky=tk.W, pady=5) ratios = ["16:9", "3:4", "9:16", "4:3", "自定义"] for i, ratio in enumerate(ratios): ttk.Radiobutton( ratio_frame, text=ratio, variable=self.ratio_var, value=ratio, command=self.on_ratio_change ).grid(row=0, column=i, padx=5) # 自定义尺寸（默认隐藏） self.custom_frame = ttk.Frame(frame) ttk.Label(self.custom_frame, text="宽度:").grid(row=0, column=0, sticky=tk.W) self.width_var = tk.IntVar(value=current_template.width_px) width_entry = ttk.Entry(self.custom_frame, textvariable=self.width_var, width=8) width_entry.grid(row=0, column=1, padx=5) width_entry.bind("", self.on_custom_size_change) ttk.Label(self.custom_frame, text="高度:").grid(row=0, column=2, sticky=tk.W, padx=5) self.height_var = tk.IntVar(value=current_template.height_px) height_entry = ttk.Entry(self.custom_frame, textvariable=self.height_var, width=8) height_entry.grid(row=0, column=3, padx=5) height_entry.bind("", self.on_custom_size_change) # 背景设置 ttk.Label(frame, text="背景颜色:").grid(row=3, column=0, sticky=tk.W, pady=5) self.bg_color_var = tk.StringVar(value=current_template.bg_color) ttk.Entry(frame, textvariable=self.bg_color_var, width=10).grid(row=3, column=1, sticky=tk.W, pady=5) ttk.Label(frame, text="背景图:").grid(row=4, column=0, sticky=tk.W, pady=5) self.bg_image_var = tk.StringVar(value="无" if current_template.bg_image is None else "已设置") ttk.Label(frame, textvariable=self.bg_image_var).grid(row=4, column=1, sticky=tk.W, pady=5) btn_frame = ttk.Frame(frame) ttk.Button(btn_frame, text="选择背景图", command=self.choose_bg_image).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="清除背景图", command=self.clear_bg_image).pack(side=tk.LEFT, padx=5) btn_frame.grid(row=4, column=2, sticky=tk.W, pady=5) # 确认和取消按钮 btn_frame = ttk.Frame(frame) ttk.Button(btn_frame, text="确定", command=self.on_ok).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="取消", command=self.on_cancel).pack(side=tk.LEFT, padx=5) btn_frame.grid(row=5, column=0, columnspan=3, pady=10) # 检查当前是否为自定义比例 if f"{current_template.ratio[0]}:{current_template.ratio[1]}" not in ratios: self.ratio_var.set("自定义") self.show_custom_frame() else: self.hide_custom_frame() # 调整大小 self.geometry("700x300") self.wait_window(self) def update_preset_list(self): """更新预设列表""" presets = self.preset_manager.get_presets() preset_names = [preset["name"] for preset in presets["presets"]] self.preset_combo['values'] = preset_names # 不自动选中任何预设，让用户手动选择 def load_selected_preset(self): """加载选中的预设""" selected_preset = self.preset_var.get() if not selected_preset: messagebox.showwarning("警告", "请先选择一个预设") return # 设置标志以避免在刷新界面时更改尺寸 self._loading_preset = True if self.preset_manager.load_preset_to_template(selected_preset, self.current_template): # 更新界面显示 self.refresh_from_template() # 刷新主窗口画布 self.parent.get_canvas().draw_preview() messagebox.showinfo("成功", f"已加载预设: {selected_preset}") else: messagebox.showerror("错误", "加载预设失败") # 清除标志 self._loading_preset = False def save_current_as_preset(self): """将当前设置保存为预设""" # 弹出输入框获取预设名称 preset_name = tk.simpledialog.askstring("保存预设", "请输入预设名称:") if not preset_name: return # 保存当前模板为预设 if self.preset_manager.save_current_template_as_preset(self.current_template, preset_name): # 更新预设列表 self.update_preset_list() messagebox.showinfo("成功", f"预设已保存: {preset_name}") else: messagebox.showerror("错误", "保存预设失败") def delete_selected_preset(self): """删除选中的预设""" selected_preset = self.preset_var.get() if not selected_preset: messagebox.showwarning("警告", "请先选择一个预设") return if messagebox.askyesno("确认", f"确定要删除预设 '{selected_preset}' 吗?"): if self.preset_manager.remove_preset(selected_preset): self.update_preset_list() messagebox.showinfo("成功", f"预设已删除: {selected_preset}") else: messagebox.showerror("错误", "删除预设失败") def refresh_from_template(self): """根据当前模板刷新界面""" # 更新比例选择 ratio_str = f"{self.current_template.ratio[0]}:{self.current_template.ratio[1]}" self.ratio_var.set(ratio_str) # 更新尺寸 self.width_var.set(self.current_template.width_px) self.height_var.set(self.current_template.height_px) # 更新背景颜色 self.bg_color_var.set(self.current_template.bg_color) # 更新背景图状态 self.bg_image_var.set("无" if self.current_template.bg_image is None else "已设置") # 处理自定义比例显示 ratios = ["16:9", "3:4", "9:16", "4:3"] if ratio_str in ratios: self.hide_custom_frame() else: self.show_custom_frame() def on_ratio_change(self): if self.ratio_var.get() == "自定义": self.show_custom_frame() # 更新比例显示为当前模板的比例 ratio_str = f"{self.current_template.ratio[0]}:{self.current_template.ratio[1]}" self.ratio_var.set(ratio_str) else: self.hide_custom_frame() if not (hasattr(self, '_loading_preset') and self._loading_preset): try: w, h = map(int, self.ratio_var.get().split(":")) current_width = self.width_var.get() if current_width <= 0: current_width = 100 # 避免宽度为0 new_height = int(current_width * h / w) new_height = max(10, new_height) # 确保最小高度 self.height_var.set(new_height) # 同步更新当前模板的尺寸 self.current_template.width_px = current_width self.current_template.height_px = new_height self.current_template.ratio = (w, h) except (ValueError, ZeroDivisionError): messagebox.showerror("错误", "无效的比例格式") def show_custom_frame(self): """显示自定义尺寸框""" self.custom_frame.grid(row=2, column=0, columnspan=3, sticky=tk.W, pady=5) def hide_custom_frame(self): """隐藏自定义尺寸框""" self.custom_frame.grid_forget() def on_custom_size_change(self, event=None): """自定义尺寸变化时的处理""" if self.ratio_var.get() == "自定义": try: width = self.width_var.get() height = self.height_var.get() if width <= 0 or height <= 0: return # 更新当前模板的尺寸 self.current_template.set_custom_size(width, height) # 更新比例显示 ratio_str = f"{self.current_template.ratio[0]}:{self.current_template.ratio[1]}" self.ratio_var.set(ratio_str) # 刷新主窗口画布 self.parent.get_canvas().draw_preview() except (ValueError, TclError): # 忽略无效输入 pass def choose_bg_image(self): """选择背景图""" file_path = filedialog.askopenfilename( title="选择背景图", filetypes=[("图片文件", "*.png *.jpg *.jpeg *.gif *.bmp")] ) if file_path: if self.current_template.load_background(file_path): self.bg_image_var.set("已设置") # 刷新主窗口画布 self.parent.get_canvas().draw_preview() else: messagebox.showerror("错误", "加载背景图失败") def clear_bg_image(self): """清除背景图""" self.current_template.clear_background() self.bg_image_var.set("无") # 刷新主窗口画布 self.parent.get_canvas().draw_preview() def on_ok(self): """确认按钮""" # 只有在用户主动点击"加载"按钮加载预设时才应用预设 # 不再自动应用选中的预设 try: if self.ratio_var.get() == "自定义": try: width = self.width_var.get() height = self.height_var.get() except tk.TclError: messagebox.showerror("错误", "请输入有效的整数") return if width <= 0 or height <= 0: messagebox.showerror("错误", "宽度和高度必须为正数") return # 更新当前模板 self.current_template.set_custom_size(width, height) # 显式设置ratio属性为自定义比例 gcd = math.gcd(width, height) ratio_w = width // gcd ratio_h = height // gcd self.current_template.ratio = (ratio_w, ratio_h) self.current_template.bg_color = self.bg_color_var.get() self.result = { "type": "custom", "width": width, "height": height, "ratio": (ratio_w, ratio_h), "bg_color": self.bg_color_var.get(), "bg_image": self.current_template.bg_image, "bg_image_path": self.current_template.bg_image_path } else: w, h = map(int, self.ratio_var.get().split(":")) # 更新当前模板 self.current_template.ratio = (w, h) self.current_template.width_px = self.width_var.get() self.current_template.height_px = self.height_var.get() self.current_template.bg_color = self.bg_color_var.get() self.result = { "type": "preset", "ratio": tuple([w, h]), "width": self.width_var.get(), "height": self.height_var.get(), "bg_color": self.bg_color_var.get(), "bg_image": self.current_template.bg_image, "bg_image_path": self.current_template.bg_image_path } self.destroy() except Exception as e: messagebox.showerror("错误", f"设置模板失败: {str(e)}") def on_cancel(self): """取消按钮""" self.result = None self.destroy()