基于Python实现一个图片水印批量添加工具

from PIL import Image, ImageDraw, ImageFont  # 处理图片和添加文字水印
import tkinter as tk  # 做图形界面
from tkinter import filedialog, messagebox, ttk  # 界面里的文件选择、提示框等组件
import os  # 处理文件夹和文件路径

def add_watermark_to_single_image():
    # 1. 让用户选择要处理的单个图片
    img_path = filedialog.askopenfilename(
        title="选一张要加水印的图片",
        filetypes=[("图片文件", "*.jpg;*.png;*.jpeg"), ("所有文件", "*.*")]
    )
    if not img_path:  # 要是用户没选图片，直接退出函数
        return

    # 2. 获取用户设置的水印参数（文字、大小、颜色、透明度）
    watermark_text = entry_text.get().strip()
    if not watermark_text:  # 没填水印文字的话，弹出提示
        messagebox.showwarning("提示", "请先输入水印文字哦！")
        return

    try:
        font_size = int(entry_font_size.get().strip())
        opacity = int(entry_opacity.get().strip())
        # 简单判断参数是否合理，避免出错
        if font_size <= 0 or opacity < 0 or opacity > 100:
            raise ValueError
    except ValueError:
        messagebox.showerror("错误", "字体大小要填正整数，透明度要在0-100之间哦！")
        return

    # 3. 读取图片，处理PNG透明格式（避免透明图片加水印后背景变黑色）
    img = Image.open(img_path).convert("RGBA")
    # 创建一个和图片一样大的透明图层，用来放水印
    watermark_layer = Image.new("RGBA", img.size, (255, 255, 255, 0))
    draw = ImageDraw.Draw(watermark_layer)

    # 4. 加载字体（这里用系统默认字体，Windows和Mac路径不一样，做了兼容）
    try:
        if os.name == "nt":  # Windows系统
            font = ImageFont.truetype("arial.ttf", font_size)
        else:  # Mac或Linux系统
            font = ImageFont.truetype("/Library/Fonts/Arial.ttf", font_size)
    except IOError:
        # 要是没找到Arial字体，就用默认字体，虽然丑点但不影响用
        font = ImageFont.load_default(size=font_size)

    # 5. 计算水印位置（默认放在右下角，距离边缘20像素，也可以自己改这里的数值）
    text_width, text_height = draw.textbbox((0, 0), watermark_text, font=font)[2:]
    img_width, img_height = img.size
    x = img_width - text_width - 20  # 右边距20
    y = img_height - text_height - 20  # 下边距20

    # 6. 添加水印（处理透明度）
    # 把用户输入的0-100透明度，转成PIL需要的0-255范围
    alpha = int(255 * (opacity / 100))
    # 这里默认水印是黑色，想改颜色的话，把(0,0,0,alpha)里的前三个数换成RGB值
    draw.text((x, y), watermark_text, font=font, fill=(0, 0, 0, alpha))

    # 7. 合并图片和水印图层，保存结果
    result = Image.alpha_composite(img, watermark_layer)
    # 处理PNG转JPG的情况（JPG不支持透明，要加白色背景）
    if img_path.lower().endswith(('.jpg', '.jpeg')):
        result = result.convert("RGB")
    
    # 让用户选保存路径
    save_path = filedialog.asksaveasfilename(
        defaultextension=os.path.splitext(img_path)[1],
        filetypes=[("图片文件", "*.jpg;*.png;*.jpeg"), ("所有文件", "*.*")],
        title="保存加水印后的图片"
    )
    if save_path:
        result.save(save_path)
        messagebox.showinfo("成功", f"图片已保存到：\n{save_path}")

def batch_add_watermark():
    # 1. 让用户选要批量处理的文件夹
    folder_path = filedialog.askdirectory(title="选要批量加水印的文件夹")
    if not folder_path:
        return

    # 2. 获取用户设置的水印参数（和单图函数一样，避免重复写代码）
    watermark_text = entry_text.get().strip()
    if not watermark_text:
        messagebox.showwarning("提示", "请先输入水印文字哦！")
        return

    try:
        font_size = int(entry_font_size.get().strip())
        opacity = int(entry_opacity.get().strip())
        if font_size <= 0 or opacity < 0 or opacity > 100:
            raise ValueError
    except ValueError:
        messagebox.showerror("错误", "字体大小要填正整数，透明度要在0-100之间哦！")
        return

    # 3. 遍历文件夹里的所有文件，只处理图片
    # 先定义支持的图片格式，避免处理非图片文件
    supported_formats = ('.jpg', '.jpeg', '.png')
    # 统计成功处理的图片数量
    success_count = 0

    # 加载字体（和单图函数一样，做了系统兼容）
    try:
        if os.name == "nt":
            font = ImageFont.truetype("arial.ttf", font_size)
        else:
            font = ImageFont.truetype("/Library/Fonts/Arial.ttf", font_size)
    except IOError:
        font = ImageFont.load_default(size=font_size)

    # 4. 逐个处理图片
    for filename in os.listdir(folder_path):
        # 只处理支持格式的文件
        if filename.lower().endswith(supported_formats):
            img_path = os.path.join(folder_path, filename)
            try:
                # 读取图片，和单图处理逻辑一致
                img = Image.open(img_path).convert("RGBA")
                watermark_layer = Image.new("RGBA", img.size, (255, 255, 255, 0))
                draw = ImageDraw.Draw(watermark_layer)

                # 计算水印位置（同样默认右下角）
                text_width, text_height = draw.textbbox((0, 0), watermark_text, font=font)[2:]
                img_width, img_height = img.size
                x = img_width - text_width - 20
                y = img_height - text_height - 20

                # 添加水印
                alpha = int(255 * (opacity / 100))
                draw.text((x, y), watermark_text, font=font, fill=(0, 0, 0, alpha))

                # 合并图层，保存图片
                result = Image.alpha_composite(img, watermark_layer)
                if filename.lower().endswith(('.jpg', '.jpeg')):
                    result = result.convert("RGB")
                
                # 保存路径：在原文件夹下加“_watermarked”后缀
                name, ext = os.path.splitext(filename)
                save_path = os.path.join(folder_path, f"{name}_watermarked{ext}")
                result.save(save_path)
                success_count += 1
            except Exception as e:
                # 遇到错误不崩溃，只是提示哪个文件处理失败
                messagebox.showerror("处理失败", f"文件 {filename} 处理出错：\n{str(e)}")

    # 批量处理结束后，提示结果
    messagebox.showinfo("批量处理完成", f"总共处理了 {success_count} 张图片\n结果保存在原文件夹，文件名带“_watermarked”后缀")

def create_gui():
    # 1. 初始化窗口
    root = tk.Tk()
    root.title("Python图片水印工具")
    root.geometry("500x300")  # 窗口大小，也可以自己改
    root.resizable(True, True)  # 允许拉伸窗口

    # 2. 创建标签和输入框（按网格布局，整齐好看）
    # 水印文字设置
    ttk.Label(root, text="水印文字：").grid(row=0, column=0, padx=10, pady=15, sticky="w")
    global entry_text
    entry_text = ttk.Entry(root, width=40)
    entry_text.grid(row=0, column=1, padx=10, pady=15)
    entry_text.insert(0, "我的图片")  # 默认文字，可修改

    # 字体大小设置
    ttk.Label(root, text="字体大小：").grid(row=1, column=0, padx=10, pady=5, sticky="w")
    global entry_font_size
    entry_font_size = ttk.Entry(root, width=40)
    entry_font_size.grid(row=1, column=1, padx=10, pady=5)
    entry_font_size.insert(0, "20")  # 默认20号字，可修改

    # 透明度设置
    ttk.Label(root, text="水印透明度（0-100）：").grid(row=2, column=0, padx=10, pady=5, sticky="w")
    global entry_opacity
    entry_opacity = ttk.Entry(root, width=40)
    entry_opacity.grid(row=2, column=1, padx=10, pady=5)
    entry_opacity.insert(0, "50")  # 默认半透明，可修改

    # 3. 创建功能按钮
    # 单图处理按钮
    btn_single = ttk.Button(root, text="处理单个图片", command=add_watermark_to_single_image)
    btn_single.grid(row=3, column=0, padx=20, pady=20, sticky="ew")

    # 批量处理按钮
    btn_batch = ttk.Button(root, text="批量处理文件夹", command=batch_add_watermark)
    btn_batch.grid(row=3, column=1, padx=20, pady=20, sticky="ew")

    # 4. 运行窗口
    root.mainloop()

if __name__ == "__main__":
    create_gui()

from PIL import Image, ImageDraw, ImageFont
import tkinter as tk
from tkinter import filedialog, messagebox, ttk, colorchooser
import os
import math


class WatermarkTool:
    def __init__(self, root):
        # 初始化主窗口
        self.root = root
        self.root.title("图片水印批量工具")
        self.root.geometry("700x550")
        self.root.resizable(True, True)
        self.root.configure(bg="#f0f0f0")  # 浅灰背景

        # 初始化参数
        self.current_color = (0, 0, 0)  # 默认黑色
        self.fullscreen_mode = False  # 全屏水印模式开关

        # 设置全局样式
        self.setup_styles()

        # 创建界面
        self.create_widgets()

    def setup_styles(self):
        """配置ttk样式，统一美化界面"""
        style = ttk.Style()
        style.theme_use("clam")  # 使用clam主题增强跨平台一致性

        # 标题样式
        style.configure("Title.TLabel", 
                       font=("微软雅黑", 12, "bold"),
                       background="#f0f0f0",
                       foreground="#333333")

        # 标签样式
        style.configure("Label.TLabel",
                       font=("微软雅黑", 10),
                       background="#f0f0f0",
                       foreground="#555555",
                       padding=5)

        # 输入框样式
        style.configure("TEntry",
                       font=("微软雅黑", 10),
                       padding=5,
                       fieldbackground="#ffffff",
                       borderwidth=1,
                       focusthickness=2,
                       focuscolor="#4a90d9")

        # 按钮样式
        style.configure("TButton",
                       font=("微软雅黑", 10, "bold"),
                       padding=8,
                       background="#4a90d9",
                       foreground="#ffffff")
        style.map("TButton",
                 background=[("active", "#357abd"), ("pressed", "#2a6099")])

        # 颜色预览样式（不依赖height，用padding控制大小）
        self.color_style = ttk.Style()
        self.color_style.configure("Color.TLabel", 
                                  background="#000000",  # 默认黑色
                                  borderwidth=1,
                                  relief="solid",
                                  padding=(10, 5))  # 用内边距控制宽高（水平10，垂直5）

        # 分组框标题样式（解决-font错误）
        style.configure("TLabelFrame.Label",
                       font=("微软雅黑", 10, "bold"),
                       foreground="#333333")

    def create_widgets(self):
        """创建界面组件，采用卡片式布局"""
        # 主容器（带边距）
        main_frame = ttk.Frame(self.root, padding=20)
        main_frame.pack(fill=tk.BOTH, expand=True)

        # 参数配置卡片（白色背景，阴影效果）
        config_card = ttk.LabelFrame(main_frame, text="水印参数配置", padding=15)
        config_card.pack(fill=tk.X, pady=(0, 20))

        # 1. 水印文字
        ttk.Label(config_card, text="水印文字：", style="Label.TLabel").grid(
            row=0, column=0, padx=10, pady=10, sticky="w")
        self.entry_text = ttk.Entry(config_card, width=50)
        self.entry_text.grid(row=0, column=1, columnspan=3, padx=10, pady=10, sticky="ew")
        self.entry_text.insert(0, "我的图片")

        # 2. 字体大小 + 透明度（横向排列）
        ttk.Label(config_card, text="字体大小：", style="Label.TLabel").grid(
            row=1, column=0, padx=10, pady=10, sticky="w")
        self.entry_font_size = ttk.Entry(config_card, width=10)
        self.entry_font_size.grid(row=1, column=1, padx=(10, 30), pady=10, sticky="w")
        self.entry_font_size.insert(0, "20")

        ttk.Label(config_card, text="透明度（0-100）：", style="Label.TLabel").grid(
            row=1, column=2, padx=10, pady=10, sticky="w")
        self.entry_opacity = ttk.Entry(config_card, width=10)
        self.entry_opacity.grid(row=1, column=3, padx=10, pady=10, sticky="w")
        self.entry_opacity.insert(0, "50")

        # 3. 文字颜色选择（修复-height错误：用padding控制预览框大小）
        ttk.Label(config_card, text="文字颜色：", style="Label.TLabel").grid(
            row=2, column=0, padx=10, pady=10, sticky="w")
        color_frame = ttk.Frame(config_card)
        color_frame.grid(row=2, column=1, columnspan=3, padx=10, pady=10, sticky="w")
        
        # 移除-height参数，依赖样式中的padding控制大小
        self.color_label = ttk.Label(color_frame, style="Color.TLabel")
        self.color_label.pack(side=tk.LEFT, padx=5)
        
        ttk.Button(color_frame, text="选择颜色", command=self.choose_color).pack(side=tk.LEFT)

        # 4. 水印位置选择
        ttk.Label(config_card, text="水印位置：", style="Label.TLabel").grid(
            row=3, column=0, padx=10, pady=10, sticky="w")
        self.position_var = tk.StringVar(value="右下")
        positions = ["左上", "中上", "右上", "左中", "中心", "右中", "左下", "中下", "右下"]
        self.position_combo = ttk.Combobox(
            config_card, textvariable=self.position_var, values=positions, width=10, state="readonly"
        )
        self.position_combo.grid(row=3, column=1, padx=10, pady=10, sticky="w")

        # 5. 重复分布（行数+列数）
        ttk.Label(config_card, text="重复分布：", style="Label.TLabel").grid(
            row=4, column=0, padx=10, pady=10, sticky="w")
        
        ttk.Label(config_card, text="行数：", style="Label.TLabel").grid(
            row=4, column=1, padx=(10, 5), pady=10, sticky="e")
        self.entry_rows = ttk.Entry(config_card, width=5)
        self.entry_rows.grid(row=4, column=2, padx=(0, 20), pady=10, sticky="w")
        self.entry_rows.insert(0, "1")

        ttk.Label(config_card, text="列数：", style="Label.TLabel").grid(
            row=4, column=2, padx=(20, 5), pady=10, sticky="e")
        self.entry_cols = ttk.Entry(config_card, width=5)
        self.entry_cols.grid(row=4, column=3, padx=10, pady=10, sticky="w")
        self.entry_cols.insert(0, "1")

        # 6. 全屏水印模式（新增功能）
        fullscreen_frame = ttk.Frame(config_card)
        fullscreen_frame.grid(row=5, column=0, columnspan=4, padx=10, pady=10, sticky="w")
        
        self.fullscreen_check = ttk.Checkbutton(
            fullscreen_frame, text="全屏水印模式（忽略位置和行列设置）",
            command=self.toggle_fullscreen_mode
        )
        self.fullscreen_check.pack(side=tk.LEFT)

        # 7. 全屏水印参数（仅在全屏模式下启用）
        self.fullscreen_params_frame = ttk.Frame(config_card)
        self.fullscreen_params_frame.grid(row=6, column=0, columnspan=4, padx=10, pady=5, sticky="w")
        self.disable_frame(self.fullscreen_params_frame)  # 默认禁用（自定义禁用方法）

        ttk.Label(self.fullscreen_params_frame, text="水印旋转角度：", style="Label.TLabel").pack(side=tk.LEFT, padx=5)
        self.entry_rotation = ttk.Entry(self.fullscreen_params_frame, width=5)
        self.entry_rotation.pack(side=tk.LEFT, padx=5)
        self.entry_rotation.insert(0, "30")  # 默认旋转30度

        ttk.Label(self.fullscreen_params_frame, text="水平间距（像素）：", style="Label.TLabel").pack(side=tk.LEFT, padx=5)
        self.entry_h_spacing = ttk.Entry(self.fullscreen_params_frame, width=5)
        self.entry_h_spacing.pack(side=tk.LEFT, padx=5)
        self.entry_h_spacing.insert(0, "100")

        ttk.Label(self.fullscreen_params_frame, text="垂直间距（像素）：", style="Label.TLabel").pack(side=tk.LEFT, padx=5)
        self.entry_v_spacing = ttk.Entry(self.fullscreen_params_frame, width=5)
        self.entry_v_spacing.pack(side=tk.LEFT, padx=5)
        self.entry_v_spacing.insert(0, "80")

        # 按钮区域（底部居中）
        btn_frame = ttk.Frame(main_frame)
        btn_frame.pack(fill=tk.X, pady=10)

        ttk.Button(
            btn_frame, text="处理单个图片", command=self.process_single_image, width=20
        ).pack(side=tk.LEFT, padx=(0, 30), pady=10, anchor=tk.CENTER)

        ttk.Button(
            btn_frame, text="批量处理文件夹", command=self.batch_process_images, width=20
        ).pack(side=tk.LEFT, padx=30, pady=10, anchor=tk.CENTER)

        # 状态标签（底部显示提示信息）
        self.status_label = ttk.Label(
            main_frame, text="请配置参数后选择图片处理", style="Label.TLabel", anchor=tk.CENTER
        )
        self.status_label.pack(fill=tk.X, pady=10)

        # 让列自适应宽度
        config_card.columnconfigure(1, weight=1)
        config_card.columnconfigure(3, weight=1)

    def disable_frame(self, frame):
        """自定义禁用框架内所有组件的方法（替代state=disabled）"""
        for widget in frame.winfo_children():
            widget.config(state="disabled")

    def enable_frame(self, frame):
        """自定义启用框架内所有组件的方法"""
        for widget in frame.winfo_children():
            widget.config(state="normal")

    def toggle_fullscreen_mode(self):
        """切换全屏水印模式，启用/禁用相关参数"""
        self.fullscreen_mode = not self.fullscreen_mode
        if self.fullscreen_mode:
            self.enable_frame(self.fullscreen_params_frame)
            self.update_status("已启用全屏水印模式")
        else:
            self.disable_frame(self.fullscreen_params_frame)
            self.update_status("已禁用全屏水印模式")

    def choose_color(self):
        """选择水印颜色并更新预览"""
        color = colorchooser.askcolor(title="选择水印颜色")[0]
        if color:
            self.current_color = (int(color[0]), int(color[1]), int(color[2]))
            hex_color = f"#{int(color[0]):02x}{int(color[1]):02x}{int(color[2]):02x}"
            self.color_style.configure("Color.TLabel", background=hex_color)
            self.update_status(f"已选择颜色：{hex_color}")

    def get_watermark_params(self):
        """获取并校验水印参数"""
        watermark_text = self.entry_text.get().strip()
        if not watermark_text:
            messagebox.showwarning("提示", "请输入水印文字")
            return None

        try:
            font_size = int(self.entry_font_size.get().strip())
            opacity = int(self.entry_opacity.get().strip())
            rows = int(self.entry_rows.get().strip())
            cols = int(self.entry_cols.get().strip())
            
            if not (font_size > 0 and 0 <= opacity <= 100 and rows > 0 and cols > 0):
                raise ValueError
        except ValueError:
            messagebox.showerror("参数错误", "请检查参数：\n- 字体大小：正整数\n- 透明度：0-100\n- 行数/列数：正整数")
            return None

        # 全屏模式参数校验
        fullscreen_params = None
        if self.fullscreen_mode:
            try:
                rotation = int(self.entry_rotation.get().strip())
                h_spacing = int(self.entry_h_spacing.get().strip())
                v_spacing = int(self.entry_v_spacing.get().strip())
                if not (h_spacing > 0 and v_spacing > 0):
                    raise ValueError
                fullscreen_params = (rotation, h_spacing, v_spacing)
            except ValueError:
                messagebox.showerror("参数错误", "全屏模式参数需为正整数")
                return None

        return (
            watermark_text, font_size, opacity,
            self.current_color, self.position_var.get(),
            rows, cols, fullscreen_params
        )

    def load_font(self, size):
        """加载字体（兼容中文）"""
        try:
            if os.name == "nt":  # Windows系统
                return ImageFont.truetype("simsun.ttc", size)  # 宋体支持中文
            else:  # Mac/Linux
                return ImageFont.truetype("/System/Library/Fonts/PingFang.ttc", size)  # 苹方字体
        except IOError:
            self.update_status("未找到指定字体，使用默认字体")
            return ImageFont.load_default(size=size)

    def calculate_position(self, position, img_w, img_h, text_w, text_h):
        """计算水印位置坐标"""
        margin = 20  # 边缘距离
        pos_map = {
            "左上": (margin, margin),
            "中上": ((img_w - text_w) // 2, margin),
            "右上": (img_w - text_w - margin, margin),
            "左中": (margin, (img_h - text_h) // 2),
            "中心": ((img_w - text_w) // 2, (img_h - text_h) // 2),
            "右中": (img_w - text_w - margin, (img_h - text_h) // 2),
            "左下": (margin, img_h - text_h - margin),
            "中下": ((img_w - text_w) // 2, img_h - text_h - margin),
            "右下": (img_w - text_w - margin, img_h - text_h - margin)
        }
        return pos_map.get(position, (margin, margin))  # 默认左上

    def draw_fullscreen_watermark(self, draw, watermark_text, font, img_width, img_height, fill_color):
        """绘制全屏水印（优化旋转逻辑）"""
        # 获取全屏模式参数
        params = self.get_watermark_params()
        if not params:
            return
        fullscreen_params = params[-1]
        rotation, h_spacing, v_spacing = fullscreen_params
        
        # 计算文字基础尺寸
        text_bbox = draw.textbbox((0, 0), watermark_text, font=font)
        text_width = text_bbox[2] - text_bbox[0]
        text_height = text_bbox[3] - text_bbox[1]
        
        # 计算网格覆盖范围（避免边缘空白）
        grid_width = img_width + text_width * 2
        grid_height = img_height + text_height * 2
        
        # 遍历网格绘制旋转水印
        for x in range(-text_width, grid_width, h_spacing):
            for y in range(-text_height, grid_height, v_spacing):
                # 创建临时图层（尺寸足够容纳旋转后的文字）
                temp_size = max(text_width, text_height) * 2
                temp_layer = Image.new("RGBA", (temp_size, temp_size), (255, 255, 255, 0))
                temp_draw = ImageDraw.Draw(temp_layer)
                
                # 在临时图层中心绘制文字
                temp_draw.text(
                    (temp_size//2 - text_width//2, temp_size//2 - text_height//2),
                    watermark_text,
                    font=font,
                    fill=fill_color
                )
                
                # 旋转临时图层（expand=True确保完整显示）
                rotated_layer = temp_layer.rotate(rotation, expand=True)
                rotated_w, rotated_h = rotated_layer.size
                
                # 计算最终绘制位置（文字中心对齐网格点）
                draw_x = x - rotated_w // 2
                draw_y = y - rotated_h // 2
                
                # 绘制旋转后的水印（用paste替代bitmap，避免透明度问题）
                draw._image.paste(rotated_layer, (draw_x, draw_y), rotated_layer)

    def process_single_image(self):
        """处理单个图片"""
        img_path = filedialog.askopenfilename(
            title="选择需要加水印的图片",
            filetypes=[("图片文件", "*.jpg;*.png;*.jpeg"), ("所有文件", "*.*")]
        )
        if not img_path:
            self.update_status("已取消选择图片")
            return

        # 获取参数
        params = self.get_watermark_params()
        if not params:
            return
        watermark_text, font_size, opacity, color, position, rows, cols, fullscreen_params = params

        try:
            self.update_status(f"正在处理图片：{os.path.basename(img_path)}")
            
            # 处理图片
            img = Image.open(img_path).convert("RGBA")
            watermark_layer = Image.new("RGBA", img.size, (255, 255, 255, 0))
            draw = ImageDraw.Draw(watermark_layer)
            font = self.load_font(font_size)
            alpha = int(255 * (opacity / 100))
            fill_color = (*color, alpha)

            # 全屏水印模式
            if self.fullscreen_mode and fullscreen_params:
                self.draw_fullscreen_watermark(draw, watermark_text, font, img.width, img.height, fill_color)
            else:
                # 普通模式：计算文字尺寸和分布
                text_bbox = draw.textbbox((0, 0), watermark_text, font=font)
                text_width = text_bbox[2] - text_bbox[0]
                text_height = text_bbox[3] - text_bbox[1]
                x_step = math.floor(img.width / (cols - 1)) if cols > 1 else 0
                y_step = math.floor(img.height / (rows - 1)) if rows > 1 else 0

                # 绘制水印
                for i in range(rows):
                    for j in range(cols):
                        base_x, base_y = self.calculate_position(
                            position, img.width, img.height, text_width, text_height
                        )
                        x = base_x + (j * x_step) if cols > 1 else base_x
                        y = base_y + (i * y_step) if rows > 1 else base_y
                        x = max(0, min(x, img.width - text_width))
                        y = max(0, min(y, img.height - text_height))
                        draw.text((x, y), watermark_text, font=font, fill=fill_color)

            # 合并保存
            result = Image.alpha_composite(img, watermark_layer)
            if img_path.lower().endswith(('.jpg', '.jpeg')):
                result = result.convert("RGB")

            save_path = filedialog.asksaveasfilename(
                defaultextension=os.path.splitext(img_path)[1],
                filetypes=[("图片文件", "*.jpg;*.png;*.jpeg"), ("所有文件", "*.*")],
                title="保存加水印后的图片"
            )
            if save_path:
                result.save(save_path)
                messagebox.showinfo("成功", f"图片已保存到：\n{save_path}")
                self.update_status(f"处理完成：{os.path.basename(save_path)}")

        except Exception as e:
            messagebox.showerror("处理失败", f"错误信息：\n{str(e)}")
            self.update_status("处理失败，请检查图片是否正常")

    def batch_process_images(self):
        """批量处理文件夹图片"""
        folder_path = filedialog.askdirectory(title="选择批量处理的文件夹")
        if not folder_path:
            self.update_status("已取消选择文件夹")
            return

        params = self.get_watermark_params()
        if not params:
            return
        watermark_text, font_size, opacity, color, position, rows, cols, fullscreen_params = params

        supported_formats = ('.jpg', '.jpeg', '.png')
        success_count = 0
        font = self.load_font(font_size)
        alpha = int(255 * (opacity / 100))
        fill_color = (*color, alpha)

        self.update_status(f"开始批量处理：{os.path.basename(folder_path)}")

        for filename in os.listdir(folder_path):
            if filename.lower().endswith(supported_formats):
                img_path = os.path.join(folder_path, filename)
                try:
                    img = Image.open(img_path).convert("RGBA")
                    watermark_layer = Image.new("RGBA", img.size, (255, 255, 255, 0))
                    draw = ImageDraw.Draw(watermark_layer)

                    # 全屏水印模式
                    if self.fullscreen_mode and fullscreen_params:
                        self.draw_fullscreen_watermark(draw, watermark_text, font, img.width, img.height, fill_color)
                    else:
                        # 普通模式绘制
                        text_bbox = draw.textbbox((0, 0), watermark_text, font=font)
                        text_width = text_bbox[2] - text_bbox[0]
                        text_height = text_bbox[3] - text_bbox[1]
                        x_step = math.floor(img.width / (cols - 1)) if cols > 1 else 0
                        y_step = math.floor(img.height / (rows - 1)) if rows > 1 else 0

                        for i in range(rows):
                            for j in range(cols):
                                base_x, base_y = self.calculate_position(
                                    position, img.width, img.height, text_width, text_height
                                )
                                x = base_x + (j * x_step) if cols > 1 else base_x
                                y = base_y + (i * y_step) if rows > 1 else base_y
                                x = max(0, min(x, img.width - text_width))
                                y = max(0, min(y, img.height - text_height))
                                draw.text((x, y), watermark_text, font=font, fill=fill_color)

                    # 保存处理结果
                    result = Image.alpha_composite(img, watermark_layer)
                    if filename.lower().endswith(('.jpg', '.jpeg')):
                        result = result.convert("RGB")

                    name, ext = os.path.splitext(filename)
                    save_path = os.path.join(folder_path, f"{name}_watermarked{ext}")
                    result.save(save_path)
                    success_count += 1
                    self.update_status(f"已处理：{filename}（{success_count}张）")

                except Exception as e:
                    messagebox.showerror("单个文件失败", f"文件 {filename} 处理出错：\n{str(e)}")

        messagebox.showinfo("批量完成", f"批量处理结束！\n共成功处理 {success_count} 张图片")
        self.update_status(f"批量处理完成，成功 {success_count} 张")

    def update_status(self, text):
        """更新底部状态提示"""
        self.status_label.config(text=text)
        self.root.update_idletasks()  # 立即刷新界面


if __name__ == "__main__":
    root = tk.Tk()
    app = WatermarkTool(root)
    root.mainloop()