import tkinter as tk from tkinter import ttk, messagebox, filedialog 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) ttk.Entry(self.custom_frame, textvariable=self.width_var, width=8).grid(row=0, column=1, padx=5) 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) ttk.Entry(self.custom_frame, textvariable=self.height_var, width=8).grid(row=0, column=3, padx=5) # 背景设置 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 if preset_names: self.preset_combo.current(0) def load_selected_preset(self): """加载选中的预设""" selected_preset = self.preset_var.get() if not selected_preset: messagebox.showwarning("警告", "请先选择一个预设") return 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("错误", "加载预设失败") 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() else: self.hide_custom_frame() # 设置默认尺寸 w, h = map(int, self.ratio_var.get().split(":")) self.width_var.set(1920 if w > h else 1080) self.height_var.set(int(self.width_var.get() * h / w)) 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 choose_bg_image(self): """选择背景图""" file_path = filedialog.askopenfilename( filetypes=[ ("图像文件", "*.jpg;*.jpeg;*.png;*.webp;*.bmp"), ("所有文件", "*.*") ] ) if file_path: if self.current_template.load_background(file_path): self.bg_image_var.set("已设置") def clear_bg_image(self): """清除背景图""" self.current_template.clear_background() self.bg_image_var.set("无") def on_ok(self): """确认按钮""" # 如果选择了预设，则应用预设 selected_preset = self.preset_var.get() if selected_preset: if not self.preset_manager.load_preset_to_template(selected_preset, self.current_template): messagebox.showerror("错误", "加载预设失败") return try: if self.ratio_var.get() == "自定义": width = self.width_var.get() height = self.height_var.get() if width <= 0 or height <= 0: messagebox.showerror("错误", "宽度和高度必须为正数") return self.result = { "type": "custom", "width": width, "height": height, "bg_color": self.bg_color_var.get() } else: w, h = map(int, self.ratio_var.get().split(":")) self.result = { "type": "preset", "ratio": (w, h), "width": self.width_var.get(), "bg_color": self.bg_color_var.get() } self.destroy() except Exception as e: messagebox.showerror("错误", f"设置模板失败: {str(e)}") def on_cancel(self): """取消按钮""" self.result = None self.destroy()