299 lines
11 KiB
Python
299 lines
11 KiB
Python
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
|
|
|
|
# 设置标志以避免在刷新界面时更改尺寸
|
|
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()
|
|
else:
|
|
self.hide_custom_frame()
|
|
# 仅在非预设加载情况下设置默认尺寸
|
|
# 当从预设加载时，保持预设的尺寸不变
|
|
if not (hasattr(self, '_loading_preset') and self._loading_preset):
|
|
# 设置默认尺寸
|
|
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("已设置")
|
|
# 立即刷新画布预览
|
|
self.parent.get_canvas().draw_preview()
|
|
|
|
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 self.preset_manager.load_preset_to_template(selected_preset, self.current_template):
|
|
# 刷新主窗口画布
|
|
self.parent.get_canvas().draw_preview()
|
|
# 设置返回结果，包含预设的所有信息
|
|
self.result = {
|
|
"type": "preset_loaded", # 修改类型以区分预设选择和预设加载
|
|
"ratio": self.current_template.ratio,
|
|
"width": self.current_template.width_px,
|
|
"height": self.current_template.height_px,
|
|
"bg_color": self.current_template.bg_color,
|
|
"bg_image": self.current_template.bg_image
|
|
}
|
|
self.destroy()
|
|
return
|
|
else:
|
|
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(),
|
|
"bg_image": self.current_template.bg_image
|
|
}
|
|
else:
|
|
w, h = map(int, self.ratio_var.get().split(":"))
|
|
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
|
|
}
|
|
self.destroy()
|
|
except Exception as e:
|
|
messagebox.showerror("错误", f"设置模板失败: {str(e)}")
|
|
|
|
def on_cancel(self):
|
|
"""取消按钮"""
|
|
self.result = None
|
|
self.destroy()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|