python显示图片-Python批量添加图片水印，就它了！ | 州的先生

点击关注州的先生

编程应用、实战教程，不容错过

在上一篇文章中，我们介绍了使用Python为图片批量添加图像水印的方法，今天，我们继续优化这个小工具，借助QT for Python模块（PyQt5、PySide2）将其制作为一个桌面客户端程序。

一、编写基础界面

首先python显示图片，我们来为这个添加图片水印的桌面客户端程序编写一个基础的图形界面。其包含：

上述的模块我们通过QLineEdit()模块实现输入框、QPushButton()实现按钮、QComboBox()实现下拉框。同时，我们使用网格布局QGridLayout()对模块进行布局。

其最终的图形界面如下图所示：

这个图形界面的创建代码如下图所示：

class MainApp(QtWidgets.QMainWindow):    def __init__(self):        super().__init__()        self.setWindowTitle("一键图片水印") # 窗口标题        self.setFixedSize(350,160) # 设置窗口高和宽
        self.main_widget = QtWidgets.QWidget() # 窗口主控件        self.main_layout = QtWidgets.QGridLayout() # 窗口主控件布局        self.main_widget.setLayout(self.main_layout)
        self.folder_input = QtWidgets.QLineEdit()        self.folder_input.setReadOnly(True)        self.folder_btn = QtWidgets.QPushButton("选择文件夹")
        self.wm_input = QtWidgets.QLineEdit()        self.wm_input.setReadOnly(True)        self.wm_btn = QtWidgets.QPushButton("选择水印图片")
        self.save_input = QtWidgets.QLineEdit()        self.save_input.setReadOnly(True)        self.save_btn = QtWidgets.QPushButton("保存目录")
        self.position_label = QtWidgets.QLabel('水印位置(州的先生zmister.com)：')        self.position_box = QtWidgets.QComboBox()        self.position_box.addItem("左上")        self.position_box.addItem("左下")        self.position_box.addItem("右上")        self.position_box.addItem("右下")        self.position_box.addItem("居中")
        self.submit_btn = QtWidgets.QPushButton("生成图片")
        self.main_layout.addWidget(self.folder_input,0,0,1,1)        self.main_layout.addWidget(self.folder_btn, 0, 1, 1, 1)        self.main_layout.addWidget(self.wm_input, 1, 0, 1, 1)        self.main_layout.addWidget(self.wm_btn, 1, 1, 1, 1)        self.main_layout.addWidget(self.save_input,2,0,1,1)        self.main_layout.addWidget(self.save_btn,2,1,1,1)        self.main_layout.addWidget(self.position_label, 3, 0, 1, 1)        self.main_layout.addWidget(self.position_box, 3, 1, 1, 1)        self.main_layout.addWidget(self.submit_btn, 4, 0, 1, 2)        self.setCentralWidget(self.main_widget) # 设置窗口初始化控件

这样，我们的基础界面已经创建好了，下面我们来对几个按钮的事件进行响应。

二、实现图片选取

在图形界面中，我们设置了4个按钮，其中3个用于目录和文件的选择，一个用于最终批量处理图片。在3个用于目录和文件选择的按钮中python显示图片，其中：

我们通过QFileDialog.getExistingDirectory()来选择文件夹、通过QFileDialog.getOpenFileName()来选择文件。最终，我们在MainApp()中新建3个方法：

# 选择图片文件夹def select_folder(self):    file = str(QtWidgets.QFileDialog.getExistingDirectory(self, "选择文件夹"))    print('文件夹为：',file)    if file != '':        self.folder_input.setText(file)
# 选择水印图片def select_wm_img(self):    file,_ = QtWidgets.QFileDialog.getOpenFileName(        self,'选择水印图片','','Images (*.png *.xpm *.jpg)'    )    print("水印图片为：",file)    if file != '':        self.wm_input.setText(file)
# 选择保存目录def select_save_folder(self):    file = str(QtWidgets.QFileDialog.getExistingDirectory(self, "选择文件夹"))    print('文件夹为：', file)    if file != '':        self.save_input.setText(file)

然后将3个按钮的点击信号连接到这3个方法上：

self.save_btn.clicked.connect(self.select_save_folder)self.wm_btn.clicked.connect(self.select_wm_img)self.folder_btn.clicked.connect(self.select_folder)

这样，我们我们图形界面中的按钮就有实际的功能了，其效果如下动图所示：

这样，我们就可以开始利用选择的东西对图片进行水印操作了。

三、绑定水印添加线程

因为批量对图片添加水印在原图片数量多的情况下，可能会非常耗时。如果我们将其放置在主类里面可能会导致主界面进行的堵塞，所以在此我们通过QThread在子线程中进行批量添加图片水印的任务。其定义如下所示：

class WmThread(QtCore.QThread):    finished_signal = QtCore.pyqtSignal(int) # 使用PySide2模块需要将pyqtSignal改成Signal
    def __init__(self,parent=None,wm_file=None,fpath=None,save_path=None,position=None):        super().__init__(parent)        self.wm_file = wm_file        self.fpath = fpath        self.save_path = save_path        self.position = position        print(self.wm_file,self.fpath)
    # 获取文件夹图片    def get_folder(self,fpath):        try:            img_suffix_list = ['png', 'jpg', 'bmp']            for i in os.listdir(fpath):                if i.split('.')[-1] in img_suffix_list:                    img_path = fpath + '/' + i                    self.img_water_mark(img_file=img_path,wm_file=self.wm_file)        except Exception as e:            print(traceback.print_exc())
    # 图片添加水印    def img_water_mark(self,img_file, wm_file):        try:            img = Image.open(img_file)  # 打开图片            watermark = Image.open(wm_file)  # 打开水印            img_size = img.size            wm_size = watermark.size            # 如果图片大小小于水印大小            if img_size[0] < wm_size[0]:                watermark.resize(tuple(map(lambda x: int(x * 0.5), watermark.size)))            print('图片大小：', img_size,"水印位置：",self.position)            if self.position == '左上':                wm_position = (0,0)            elif self.position == '左下':                wm_position = (0,img_size[1]-wm_size[1])            elif self.position == '右上':                wm_position = (img_size[0]-wm_size[0],0)            elif self.position == '右下':                wm_position = (img_size[0]-wm_size[0],img_size[1]-wm_size[1])            elif self.position == '居中':                wm_position = (img_size[0]//2-wm_size[0]//2,img_size[1]//2-wm_size[1]//2)            layer = Image.new('RGBA', img.size)  # 新建一个图层            layer.paste(watermark, wm_position)  # 将水印图片添加到图层上            mark_img = Image.composite(layer, img, layer)            new_file_name = '/new_'+img_file.split('/')[-1]            if self.save_path == '':                mark_img.save(self.fpath+new_file_name)            else:                mark_img.save(self.save_path + new_file_name)        except Exception as e:            print(traceback.print_exc())
    def run(self):        try:            print("开始处理……")            self.get_folder(fpath=self.fpath)            self.finished_signal.emit(1)        except Exception as e:            print(traceback.print_exc())            self.finished_signal.emit(0)

这样，我们就定义了一个子线程类，批量添加图片水印的实际工作都有这个子线程类来处理。然后我们需要在主类MainApp()中对这个子线程类进行调用：

# 生成水印图片def generate_img(self):    try:        print("获取图片目录及水印文件名")        folder_text = self.folder_input.text()        wm_text = self.wm_input.text()        save_text = self.save_input.text()        position_text = self.position_box.currentText()        print('判断是否为空')        if folder_text != '' and wm_text != '':            self.submit_btn.setEnabled(False)            self.genare_thread = WmThread(                wm_file= wm_text,                fpath=folder_text,                save_path = save_text,                position = position_text            )            print('启动线程')            self.genare_thread.finished_signal.connect(self.thread_resp)            self.genare_thread.start()
    except Exception as e:        print(traceback.print_exc())
def thread_resp(self,value):    if value == 1:        QtWidgets.QMessageBox.information(self,'提示','处理完成！',QtWidgets.QMessageBox.Yes)    if value == 0:        QtWidgets.QMessageBox.information(self,'提示', '处理出错！',QtWidgets.QMessageBox.Yes)    self.submit_btn.setEnabled(True)

最后将generate_img()方法绑定到生成按钮的点击信号上：

self.submit_btn.clicked.connect(self.generate_img)

这样，我们的批量添加图片水印客户端程序就编写完成了。

四、将代码打包为EXE程序

最后，我们借助PyInstaller模块将编写好的代码打包为EXE格式的可执行文件。使用如下命令将其打包为一个文件夹：

pyinstaller -w main.py

最后打包生成了一个最终的文件夹，如下图所示：

最后，我们运行main.exe文件，就会出现之前编写的图形界面，功能亦是相同的。

这样，我们就完成了图片水印批量添加程序的开发和打包。有问题欢迎留言讨论，本程序将会逐步迭代更新并开放免费使用，争取早日实现同类型商业化软件的功能，敬请关注。

正文结束

限时特惠：本站每日持续更新海量展厅资源，一年会员只需29.9元，全站资源免费下载
站长微信：zhanting688

主题授权提示：请在后台主题设置-主题授权-激活主题的正版授权，授权购买：RiTheme官网

声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。

图形界面线程

相关文章