Python教程：带你使用PyQt手搓一个图片水印小工具

旦莫2024-05-27 15:19

图片水印是在数字图片上叠加一些隐蔽或透明的标识，以证实其版权、来源或其他信息的技术。水印可以是文字、图案、Logo等，用于保护图片的版权并防止未经授权的复制和使用。

步骤1: 安装必要的库

首先，确保你的环境中安装了Python和PyQT。然后，安装必要的库：

pip install PyQT5 pillow

步骤2: 创建PyQT应用程序框架

我们将创建一个基本的PyQT应用程序框架，包括主窗口和布局。

python 复制代码 
import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QVBoxLayout

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("图片水印工具")
        self.setGeometry(100, 100, 800, 600)

        # 创建中心部件
        central_widget = QWidget(self)
        self.setCentralWidget(central_widget)

        # 创建布局
        layout = QVBoxLayout()
        central_widget.setLayout(layout)

        self.show()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    sys.exit(app.exec_())

步骤3: 添加控件

接下来，我们将添加控件，如按钮、文本框和图片显示区域。

python 复制代码 
from PyQt5.QtWidgets import QPushButton, QLabel, QLineEdit

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        # ... 省略之前的代码 ...

        # 添加控件
        self.image_label = QLabel(self)
        layout.addWidget(self.image_label)

        self.watermark_text = QLineEdit(self)
        self.watermark_text.setPlaceholderText("输入水印文本")
        layout.addWidget(self.watermark_text)

        self.load_button = QPushButton("加载图片", self)
        layout.addWidget(self.load_button)

        self.add_watermark_button = QPushButton("添加水印", self)
        layout.addWidget(self.add_watermark_button)

        self.save_button = QPushButton("保存图片", self)
        layout.addWidget(self.save_button)

        # 连接信号与槽
        self.load_button.clicked.connect(self.load_image)
        self.add_watermark_button.clicked.connect(self.add_watermark)
        self.save_button.clicked.connect(self.save_image)

    def load_image(self):
        # 实现图片加载功能
        pass

    def add_watermark(self):
        # 实现添加水印功能
        pass

    def save_image(self):
        # 实现保存图片功能
        pass

步骤4: 实现功能

现在，我们将实现加载图片、添加水印和保存图片的功能。

python 复制代码 
from PIL import Image, ImageDraw, ImageFont

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        # ... 省略之前的代码 ...

    def load_image(self):
        options = QFileDialog.Options()
        file_name, _ = QFileDialog.getOpenFileName(self, "选择图片", "", "Images (*.png *.jpg *.jpeg)", options=options)
        if file_name:
            self.image = Image.open(file_name)
            # 将 Pillow 图像转换为 QImage
            qimage = QImage(self.image.tobytes(), self.image.width, self.image.height, QImage.Format_RGB888)
            # 将 QImage 转换为 QPixmap
            self.image_label.setPixmap(QPixmap.fromImage(qimage))

    def add_watermark(self):
        if not hasattr(self, 'image'):
            return
        draw = ImageDraw.Draw(self.image)
        font = ImageFont.truetype("arial.ttf", 36)
        draw.text((10, 10), self.watermark_text.text(), font=font, fill=(255, 255, 255, 128))
        self.image_label.setPixmap(QPixmap.fromImage(QImage(self.image.tobytes(), self.image.width, self.image.height, QImage.Format_RGB888)))

    def save_image(self):
        if not hasattr(self, 'image'):
            return
        options = QFileDialog.Options()
        file_name, _ = QFileDialog.getSaveFileName(self, "保存图片", "", "Images (*.png *.jpg *.jpeg)", options=options)
        if file_name:
            self.image.save(file_name)

完整代码

将上述所有代码整合到一个文件中，你将得到一个完整的图片增加水印工具。

python 复制代码 
import sys

from PyQt5.QtGui import QPixmap, QImage
from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QVBoxLayout, QPushButton, QLabel, QLineEdit, QFileDialog
from PIL import Image, ImageDraw, ImageFont

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("图片水印工具")
        self.setGeometry(100, 100, 800, 600)

        central_widget = QWidget(self)
        self.setCentralWidget(central_widget)

        layout = QVBoxLayout()
        central_widget.setLayout(layout)

        self.image_label = QLabel(self)
        layout.addWidget(self.image_label)

        self.watermark_text = QLineEdit(self)
        self.watermark_text.setPlaceholderText("输入水印文本")
        layout.addWidget(self.watermark_text)

        self.load_button = QPushButton("加载图片", self)
        layout.addWidget(self.load_button)

        self.add_watermark_button = QPushButton("添加水印", self)
        layout.addWidget(self.add_watermark_button)

        self.save_button = QPushButton("保存图片", self)
        layout.addWidget(self.save_button)

        self.load_button.clicked.connect(self.load_image)
        self.add_watermark_button.clicked.connect(self.add_watermark)
        self.save_button.clicked.connect(self.save_image)

        self.show()

    def load_image(self):
        options = QFileDialog.Options()
        file_name, _ = QFileDialog.getOpenFileName(self, "选择图片", "", "Images (*.png *.jpg *.jpeg)", options=options)
        if file_name:
            self.image = Image.open(file_name)
            # 将 Pillow 图像转换为 QImage
            qimage = QImage(self.image.tobytes(), self.image.width, self.image.height, QImage.Format_RGB888)
            # 将 QImage 转换为 QPixmap
            self.image_label.setPixmap(QPixmap.fromImage(qimage))

    def add_watermark(self):
        if not hasattr(self, 'image'):
            return
        draw = ImageDraw.Draw(self.image)
        font = ImageFont.truetype("arial.ttf", 36)
        draw.text((10, 10), self.watermark_text.text(), font=font, fill=(255, 255, 255, 128))
        self.image_label.setPixmap(QPixmap.fromImage(QImage(self.image.tobytes(), self.image.width, self.image.height, QImage.Format_RGB888)))

    def save_image(self):
        if not hasattr(self, 'image'):
            return
        options = QFileDialog.Options()
        file_name, _ = QFileDialog.getSaveFileName(self, "保存图片", "", "Images (*.png *.jpg *.jpeg)", options=options)
        if file_name:
            self.image.save(file_name)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    sys.exit(app.exec_())

实现效果如下：

总结

本文介绍了一个简易版的图片加水印小工具的开发示例，大致思路如下：

  1. 导入必要的模块和库：导入了sys、QPixmap、QImage、QApplication、QMainWindow、QWidget、QVBoxLayout、QPushButton、QLabel、QLineEdit和QFileDialog等模块，以及Pillow库中的Image、ImageDraw和ImageFont模块。

  2. 创建主窗口类MainWindow：定义了一个继承自QMainWindow的类MainWindow，用于显示程序的主窗口界面和处理用户交互。

  3. 初始化界面：在MainWindow类的初始化方法__init__中，设置了窗口标题、大小，并创建了主窗口的布局和控件，包括一个用于显示图片的标签、一个输入水印文本的文本框、加载图片的按钮、添加水印的按钮和保存图片的按钮。

  4. 绑定按钮点击事件：通过clicked.connect()方法，将加载图片按钮、添加水印按钮和保存图片按钮与对应的处理方法load_image、add_watermark和save_image绑定在一起。

  5. 加载图片：当用户点击加载图片按钮时，使用QFileDialog打开文件对话框选择图片文件，并将选择的图片文件用Pillow库中的Image打开，然后将其转换为QImage格式，并显示在程序界面的图片标签上。

  6. 添加水印：当用户点击添加水印按钮时，首先判断是否已经加载了图片，然后使用Pillow库中的ImageDraw和ImageFont来给图片添加指定的文本水印，并更新显示在界面上的图片标签。

  7. 保存图片：当用户点击保存图片按钮时，首先判断是否已经加载了图片，然后使用QFileDialog打开文件对话框选择保存路径，并将图片保存到指定的路径。

  8. 启动应用程序：在主程序中创建了QApplication实例app，并实例化了MainWindow类的对象window，并通过sys.exit(app.exec_())启动了应用程序的事件循环。

程序的功能还比较简陋，仅供练手参考，后续针对这个程序还可以有更多的优化空间，比如：

  1. 错误处理：在加载图片、添加水印和保存图片的过程中，可以添加一些错误处理，例如捕获文件操作可能出现的异常，并向用户提供友好的错误提示。

  2. 图片显示优化：对于较大的图片，可以在加载时进行适当的缩放处理，以提高程序的响应速度和用户体验。

  3. 界面优化：可以考虑对界面进行美化，增加一些图标、布局调整或者添加状态栏等元素，以提高用户交互体验。

  4. 水印位置和样式：目前的代码中水印的位置固定在(10, 10)处，可以考虑添加更多选项来让用户自定义水印的位置、大小、颜色等样式。

  5. 支持更多图片格式：目前只支持png、jpg和jpeg格式的图片，可以考虑扩展支持更多常见的图片格式。

  6. 多语言支持：可以考虑添加多语言支持，让用户可以选择不同的界面语言，等等。

相关推荐
liu_chunhai几秒前
设计模式（3）builder
java·开发语言·设计模式
Mr.D学长5 分钟前
毕业设计 深度学习社交距离检测系统（源码+论文）
python·毕业设计·毕设
姜学迁9 分钟前
Rust-枚举
开发语言·后端·rust
wdxylb9 分钟前
解决Python使用Selenium 时遇到网页 ＜body＞ 划不动的问题
python
冷白白10 分钟前
【C++】C++对象初探及友元
c语言·开发语言·c++·算法
凌云行者14 分钟前
rust的迭代器方法——collect
开发语言·rust
代码骑士17 分钟前
【一起学NLP】Chapter3-使用神经网络解决问题
python·神经网络·自然语言处理
It'sMyGo17 分钟前
Javascript数组研究09_Array.prototype[Symbol.unscopables]
开发语言·javascript·原型模式
睡觉然后上课28 分钟前
c基础面试题
c语言·开发语言·c++·面试
qing_04060335 分钟前
C++——继承
开发语言·c++·继承
热门推荐
01分享几个惊艳的 Three.js 个人站点02【经验分享】Ubuntu22.04安装微信(linux官方版)032024年高教社杯数学建模国赛C题超详细解题思路分析04组基轨迹建模 GBTM的介绍与实现（Stata 或 R）05安卓系列机型永久去除data分区加密 详细步骤解析06【2024数模国赛赛题思路公开】国赛B题思路丨附可运行代码丨无偿自提07springer 在线投稿编译踩坑08Ubuntu24.04安装中文输入法09【2024高教社杯全国大学生数学建模竞赛】B题 生产过程中的决策问题——解题思路 代码 论文10CTF网络安全大赛简单的web抓包题目：HEADache