PyQt学习记录03——批量设置水印

0. 目录

PyQt学习记录01——加法计算器 PyQt学习记录02——串口助手

1. 前言

本次主要是为了学习Qt中的 QFileDialog 函数， QFileDialog.getExistingDirectory：用于选择文件夹，返回的是一个文件夹路径。 QFileDialog.getOpenFileName： 用于选择单个文件。返回的是文件的完整路径。

以及通过importlib.util实现跨文件的函数调用，即通过将定义一个对象，动态加载和调用该py文件中的函数（详见第二小节）

希望对你有所帮助。

2. Ui界面

lineEdit 用于记录文件夹路径 lineEdit_2 用于记录水印文件路径 其余元素详见第3小节的程序代码

3. 程序代码 import sys from PyQt5.QtWidgets import QApplication, QMainWindow, QFileDialog from WaterMask import Ui_MainWindow # 导入 WaterMask 的 UI 类 from PyQt5.QtCore import QTimer import importlib.util # 用于动态加载2.py中的函数 # 动态加载 2.py #加载一个名为 "watermark" 的模块（可自由命名），它的文件路径是 "2.py"。 spec = importlib.util.spec_from_file_location("watermark", "2.py") #用于创建一个模块对象watermark_module。 watermark_module = importlib.util.module_from_spec(spec) #这行代码的作用是执行 "2.py" 文件中的代码，并将模块中的函数、类等加载到 watermark_module 对象中。 spec.loader.exec_module(watermark_module) #调用则使用watermark_module.add_watermark_to_images(folder_path, watermark_path) class WatermarkApp(QMainWindow, Ui_MainWindow): def __init__(self): super().__init__() self.setupUi(self) # 设置 UI 界面 # 绑定按钮点击事件 self.pushButton.clicked.connect(self.select_folder) # 打开文件夹路径 self.pushButton_2.clicked.connect(self.select_watermark) # 打开水印路径 self.pushButton_3.clicked.connect(self.process_files) # 开始转换按钮 def select_folder(self): folder_path = QFileDialog.getExistingDirectory(self, "选择文件夹") if folder_path: self.lineEdit.setText(folder_path) def select_watermark(self): watermark_path, _ = QFileDialog.getOpenFileName(self, "选择水印图片", "", "Image Files (*.png *.jpg *.bmp *.jpeg)") if watermark_path: self.lineEdit_2.setText(watermark_path) def process_files(self): folder_path = self.lineEdit.text() watermark_path = self.lineEdit_2.text() if not folder_path or not watermark_path: print("请先选择文件夹和水印图片") return # 按钮按下后直接变为 “转换中” self.pushButton_3.setText("转换中") self.pushButton_3.setEnabled(False) # 延时500ms调用 2.py 中的 add_watermark_to_images 函数，传递回调函数来恢复按钮状态 QTimer.singleShot(500, lambda: self.start_batch_processing(folder_path, watermark_path)) def start_batch_processing(self, folder_path, watermark_path): # 调用 2.py 中的批量处理函数 watermark_module.add_watermark_to_images(folder_path, watermark_path) # 批量处理完成后恢复按钮状态 self.pushButton_3.setText("开始转换") self.pushButton_3.setEnabled(True) if __name__ == "__main__": app = QApplication(sys.argv) window = WatermarkApp() window.show() sys.exit(app.exec_())

需要注意的是，本次学习调用了2.py来处理复杂的水印添加功能，所以展现的的代码中，可以理解为只是一个连接ui界面和2.py中的批处理函数的中间件，即ui界面提供文件路径和开始命令，通过所展现代码传递给2.py。

注意以下仅展现的2.py的基础代码

from PIL import Image import os def add_watermark_to_images(folder_path, watermark_path): try: for filename in os.listdir(folder_path): file_path = os.path.join(folder_path, filename) if file_path.lower().endswith(('png', 'jpg', 'jpeg', 'bmp')): print(f"给 {file_path} 添加水印: {watermark_path}") except Exception as e: print(f"发生错误: {e}")

需要注意的是，这里使用的动态加载方法在使用pyinstaller打包为exe文件时会出现找不到2.py文件的问题，所以你需要将动态加载更改为传统import导入，同时将2.py更改为two.py

import two ''' import importlib.util # 用于动态加载2.py中的函数 # 动态加载 two.py #加载一个名为 "watermark" 的模块（可自由命名），它的文件路径是 "two.py"。 spec = importlib.util.spec_from_file_location("watermark", "two.py") #用于创建一个模块对象watermark_module。 watermark_module = importlib.util.module_from_spec(spec) #这行代码的作用是执行 "two.py" 文件中的代码，并将模块中的函数、类等加载到 watermark_module 对象中。 spec.loader.exec_module(watermark_module) #调用则使用watermark_module.add_watermark_to_images(folder_path, watermark_path) ''' #调用则使用 two.add_watermark_to_images(folder_path, watermark_path) 4. 运行结果

点击’开始转换’后，按钮文本改变为’转换中’，同时在控制台中显示进度，转换完成后按钮恢复