【创意智造组】微信小程序跳一跳游戏——基于二哈2的自动闯关装置 简单

import os
import sys
import tkinter as tk
from tkinter import filedialog, messagebox, ttk

# 尝试导入 Pillow 库
try:
    from PIL import Image
except ImportError:
    print("错误：未安装 Pillow 库。请运行以下命令安装：")
    print("pip install Pillow")
    try:
        root = tk.Tk()
        root.withdraw()
        messagebox.showerror("缺少依赖", "未安装 Pillow 库，请安装后重试。\n\n命令：pip install Pillow")
        root.destroy()
    except:
        pass
    sys.exit(1)


def crop_center_square(image_path: str, output_path: str, skip_existing: bool = True) -> bool:
    """
    将图片按最短边从正中心裁剪为正方形，并保存到指定路径。

    :param image_path: 原始图片路径
    :param output_path: 输出图片路径
    :param skip_existing: 若输出文件已存在，是否跳过（True=跳过并返回False）
    :return: 是否成功裁剪并保存
    """
    if skip_existing and os.path.exists(output_path):
        return False
    try:
        with Image.open(image_path) as img:
            width, height = img.size
            side = min(width, height)
            left = (width - side) // 2
            top = (height - side) // 2
            right = left + side
            bottom = top + side
            cropped = img.crop((left, top, right, bottom))
            cropped.save(output_path)
        return True
    except Exception as e:
        print(f"处理失败：{image_path} - {e}")
        return False


class BatchCropper:
    def __init__(self, root):
        self.root = root
        self.root.title("批量图片正方形裁剪工具")
        self.root.geometry("500x250")
        self.root.resizable(False, False)

        # 说明标签
        info_label = tk.Label(
            root,
            text="选择多张图片，程序将自动按最短边\n从正中心裁剪出正方形区域",
            justify="center",
            font=("微软雅黑", 10)
        )
        info_label.pack(pady=10)

        # 选择文件按钮
        self.select_btn = tk.Button(
            root,
            text="选择多张图片并裁剪",
            command=self.start_batch,
            width=20,
            height=2,
            bg="#4CAF50",
            fg="white",
            font=("微软雅黑", 10)
        )
        self.select_btn.pack(pady=10)

        # 进度条
        self.progress = ttk.Progressbar(root, orient="horizontal", length=400, mode="determinate")
        self.progress.pack(pady=10)

        # 状态标签
        self.status_label = tk.Label(root, text="就绪", fg="gray")
        self.status_label.pack(pady=5)

        # 退出按钮
        exit_btn = tk.Button(root, text="退出", command=root.quit, width=10)
        exit_btn.pack(pady=5)

    def start_batch(self):
        """选择多张图片并开始批量裁剪"""
        file_paths = filedialog.askopenfilenames(
            title="选择多张图片",
            filetypes=[
                ("图像文件", "*.jpg *.jpeg *.png *.bmp *.tiff *.webp"),
                ("所有文件", "*.*")
            ]
        )
        if not file_paths:
            return

        # 禁用按钮，避免重复点击
        self.select_btn.config(state="disabled", text="处理中...")
        self.progress["value"] = 0
        self.progress["maximum"] = len(file_paths)

        # 使用 update_idletasks 保持界面响应，逐张处理（不另开线程，简单可靠）
        success_count = 0
        for i, file_path in enumerate(file_paths, 1):
            # 更新状态标签
            filename = os.path.basename(file_path)
            self.status_label.config(text=f"正在处理：{filename}")
            self.root.update_idletasks()

            # 生成输出路径（原文件夹，添加 _square 后缀）
            dir_name = os.path.dirname(file_path)
            base_name, ext = os.path.splitext(filename)
            output_name = f"{base_name}_square{ext}"
            output_path = os.path.join(dir_name, output_name)

            # 执行裁剪
            if crop_center_square(file_path, output_path, skip_existing=True):
                success_count += 1
            # 更新进度条
            self.progress["value"] = i
            self.root.update_idletasks()

        # 处理完成，恢复按钮并显示结果
        self.select_btn.config(state="normal", text="选择多张图片并裁剪")
        self.status_label.config(text=f"完成！成功：{success_count} / {len(file_paths)}")
        messagebox.showinfo("批量处理完成", f"共处理 {len(file_paths)} 张图片\n成功裁剪 {success_count} 张\n已跳过已存在的正方形图片。")
        self.progress["value"] = 0


def main():
    root = tk.Tk()
    app = BatchCropper(root)
    root.mainloop()


if __name__ == "__main__":
    main()