Python编写一个图片批量添加文字水印工具(附代码)

pipinstallPyQt5 Pillow

classWorker(QThread):

progress=pyqtSignal(int)# 进度信号

finished=pyqtSignal(str)# 完成信号

defrun(self):

# 遍历文件列表进行处理

fori, file_pathinenumerate(self.files):

self.process_image(file_path)

self.progress.emit(…)# 更新进度条

defprocess_image(self, file_path):

with Image.open(file_path).convert(\”RGBA\”) as img:

# 创建全透明水印层

watermark=Image.new(\’RGBA\’, img.size, (0,0,0,0))

# … (省略字体加载和颜色设置) …

# 创建单个水印小图用于旋转

txt_img=Image.new(\’RGBA\’, (max_dim, max_dim), (0,0,0,0))

txt_draw=ImageDraw.Draw(txt_img)

txt_draw.text((text_x, text_y), text, font=font, fill=fill_color)

# 旋转

ifrotation !=0:

txt_img=txt_img.rotate(rotation, resample=Image.BICUBIC)

# 核心布局逻辑

ifposition==\’平铺 (Tile)\’:

# 双重循环实现全图平铺

step_x=int(w_width+spacing)

step_y=int(w_height+spacing)

foryinrange(0, img.height, step_y):

forxinrange(0, img.width, step_x):

watermark.paste(txt_img, (x, y), txt_img)

else:

# 九宫格定位逻辑

# 根据 \’左\’, \’右\’, \’上\’, \’下\’ 关键字计算坐标

# …

watermark.paste(txt_img, (pos_x, pos_y), txt_img)

# 合成最终图片

out=Image.alpha_composite(img, watermark)

# 缩放比例计算

scale_factor=preview_img.width/original_img_width

# 字体大小也要随之缩放

preview_font_size=int(user_set_font_size*scale_factor)

importsys

importos

fromPyQt5.QtWidgetsimport(QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout,

QLabel, QPushButton, QLineEdit, QFileDialog, QSlider, QSpinBox,

QComboBox, QColorDialog, QProgressBar, QMessageBox, QGroupBox,

QScrollArea, QListWidget)

fromPyQt5.QtCoreimportQt, QThread, pyqtSignal

fromPyQt5.QtGuiimportQPixmap, QImage, QColor, QFont

fromPILimportImage, ImageDraw, ImageFont, ImageEnhance

importmath

classWorker(QThread):

progress=pyqtSignal(int)

finished=pyqtSignal(str)

error=pyqtSignal(str)

def__init__(self, files, output_dir, config):

super().__init__()

self.files=files

self.output_dir=output_dir

self.config=config

self.is_running=True

defrun(self):

total=len(self.files)

success_count=0

ifnotos.path.exists(self.output_dir):

try:

os.makedirs(self.output_dir)

exceptException as e:

self.error.emit(f\”无法创建输出目录: {str(e)}\”)

return

fori, file_pathinenumerate(self.files):

ifnotself.is_running:

break

try:

self.process_image(file_path)

success_count+=1

exceptException as e:

print(f\”Error processing {file_path}: {e}\”)

self.progress.emit(int((i+1)/total*100))

self.finished.emit(f\”处理完成！成功: {success_count}/{total}\”)

defprocess_image(self, file_path):

try:

with Image.open(file_path).convert(\”RGBA\”) as img:

# 创建水印层

watermark=Image.new(\’RGBA\’, img.size, (0,0,0,0))

draw=ImageDraw.Draw(watermark)

text=self.config[\’text\’]

font_size=self.config[\’font_size\’]

opacity=self.config[\’opacity\’]

rotation=self.config[\’rotation\’]

color=self.config[\’color\’]# Tuple (r, g, b)

position=self.config[\’position\’]

spacing=self.config[\’spacing\’]# For tiling

# 加载字体 (使用默认字体，因为系统字体路径复杂，这里简化处理)

try:

# 尝试使用微软雅黑

font=ImageFont.truetype(\”msyh.ttc\”, font_size)

except:

font=ImageFont.load_default()

# default font doesn\’t scale well, but fallback is needed

# If we really want size, we might need a standard font file distributed with app

# Trying basic arial if msyh fails

try:

font=ImageFont.truetype(\”arial.ttf\”, font_size)

except:

pass# Fallback to default

# 计算文本大小

bbox=draw.textbbox((0,0), text, font=font)

text_width=bbox[2]-bbox[0]

text_height=bbox[3]-bbox[1]

# 创建单个水印图片用于旋转

# 留出足够空间以防旋转后被裁剪

max_dim=int(math.sqrt(text_width**2+text_height**2))

txt_img=Image.new(\’RGBA\’, (max_dim, max_dim), (0,0,0,0))

txt_draw=ImageDraw.Draw(txt_img)

# 居中绘制文本

text_x=(max_dim-text_width)//2

text_y=(max_dim-text_height)//2

# 设置颜色和透明度

fill_color=(color[0], color[1], color[2],int(255*opacity))

txt_draw.text((text_x, text_y), text, font=font, fill=fill_color)

# 旋转

ifrotation !=0:

txt_img=txt_img.rotate(rotation, resample=Image.BICUBIC)

# 获取旋转后的实际内容边界（可选，但为了精确布局最好做）

# 这里简单处理，直接使用txt_img

w_width, w_height=txt_img.size

ifposition==\’平铺 (Tile)\’:

# 平铺逻辑

# spacing 是间距倍数或像素

step_x=int(w_width+spacing)

step_y=int(w_height+spacing)

ifstep_x <=0: step_x=w_width+50

ifstep_y <=0: step_y=w_height+50

foryinrange(0, img.height, step_y):

forxinrange(0, img.width, step_x):

watermark.paste(txt_img, (x, y), txt_img)

else:

# 单个位置逻辑

pos_x=0

pos_y=0

margin=20

if\’左\’inposition:

pos_x=margin

elif\’右\’inposition:

pos_x=img.width-w_width-margin

else:# 中 (水平)

pos_x=(img.width-w_width)//2

if\’上\’inposition:

pos_y=margin

elif\’下\’inposition:

pos_y=img.height-w_height-margin

else:# 中 (垂直)

pos_y=(img.height-w_height)//2

watermark.paste(txt_img, (pos_x, pos_y), txt_img)

# 合成

out=Image.alpha_composite(img, watermark)

# 保存

filename=os.path.basename(file_path)

save_path=os.path.join(self.output_dir, filename)

# Convert back to RGB if saving as JPEG, otherwise keep RGBA for PNG

iffilename.lower().endswith((\’.jpg\’,\’.jpeg\’)):

out=out.convert(\’RGB\’)

out.save(save_path, quality=95)

else:

out.save(save_path)

exceptException as e:

print(f\”Processing failed for {file_path}: {e}\”)

raisee

classWatermarkApp(QMainWindow):

def__init__(self):

super().__init__()

self.setWindowTitle(\”批量图片水印工具\”)

self.resize(1000,700)

# Data

self.image_files=[]

self.current_preview_image=None

self.watermark_color=(255,255,255)# Default white

# UI Setup

central_widget=QWidget()

self.setCentralWidget(central_widget)

main_layout=QHBoxLayout(central_widget)

# Left Panel (Settings)

left_panel=QWidget()

left_layout=QVBoxLayout(left_panel)

left_panel.setFixedWidth(350)

main_layout.addWidget(left_panel)

# 1. File Selection

grp_files=QGroupBox(\”文件选择\”)

grp_files_layout=QVBoxLayout()

btn_layout=QHBoxLayout()

self.btn_add_files=QPushButton(\”添加图片\”)

self.btn_add_files.clicked.connect(self.add_files)

self.btn_add_folder=QPushButton(\”添加文件夹\”)

self.btn_add_folder.clicked.connect(self.add_folder)

self.btn_clear_files=QPushButton(\”清空列表\”)

self.btn_clear_files.clicked.connect(self.clear_files)

btn_layout.addWidget(self.btn_add_files)

btn_layout.addWidget(self.btn_add_folder)

btn_layout.addWidget(self.btn_clear_files)

self.list_files=QListWidget()

self.list_files.currentRowChanged.connect(self.update_preview)

grp_files_layout.addLayout(btn_layout)

grp_files_layout.addWidget(self.list_files)

grp_files.setLayout(grp_files_layout)

left_layout.addWidget(grp_files)

# 2. Output Directory

grp_output=QGroupBox(\”输出设置\”)

grp_output_layout=QVBoxLayout()

out_path_layout=QHBoxLayout()

self.edit_output=QLineEdit()

self.edit_output.setPlaceholderText(\”选择输出目录…\”)

self.btn_browse_output=QPushButton(\”浏览\”)

self.btn_browse_output.clicked.connect(self.browse_output)

out_path_layout.addWidget(self.edit_output)

out_path_layout.addWidget(self.btn_browse_output)

grp_output_layout.addLayout(out_path_layout)

grp_output.setLayout(grp_output_layout)

left_layout.addWidget(grp_output)

# 3. Watermark Settings

grp_settings=QGroupBox(\”水印设置\”)

grp_settings_layout=QVBoxLayout()

# Text

self.edit_text=QLineEdit(\”Sample Watermark\”)

self.edit_text.setPlaceholderText(\”输入水印文字\”)

self.edit_text.textChanged.connect(self.update_preview_delayed)

grp_settings_layout.addWidget(QLabel(\”水印文字:\”))

grp_settings_layout.addWidget(self.edit_text)

# Color

color_layout=QHBoxLayout()

self.btn_color=QPushButton(\”选择颜色\”)

self.btn_color.clicked.connect(self.choose_color)

self.lbl_color_preview=QLabel(\” \”)

self.lbl_color_preview.setStyleSheet(\”background-color: white; border: 1px solid black;\”)

self.lbl_color_preview.setFixedWidth(30)

color_layout.addWidget(QLabel(\”颜色:\”))

color_layout.addWidget(self.btn_color)

color_layout.addWidget(self.lbl_color_preview)

color_layout.addStretch()

grp_settings_layout.addLayout(color_layout)

# Font Size

size_layout=QHBoxLayout()

self.spin_size=QSpinBox()

self.spin_size.setRange(10,500)

self.spin_size.setValue(40)

self.spin_size.valueChanged.connect(self.update_preview_delayed)

size_layout.addWidget(QLabel(\”字体大小:\”))

size_layout.addWidget(self.spin_size)

grp_settings_layout.addLayout(size_layout)

# Opacity

opacity_layout=QHBoxLayout()

self.slider_opacity=QSlider(Qt.Horizontal)

self.slider_opacity.setRange(0,100)

self.slider_opacity.setValue(50)

self.slider_opacity.valueChanged.connect(self.update_preview_delayed)

opacity_layout.addWidget(QLabel(\”透明度:\”))

opacity_layout.addWidget(self.slider_opacity)

grp_settings_layout.addLayout(opacity_layout)

# Rotation

rotation_layout=QHBoxLayout()

self.slider_rotation=QSlider(Qt.Horizontal)

self.slider_rotation.setRange(0,360)

self.slider_rotation.setValue(0)

self.slider_rotation.valueChanged.connect(self.update_preview_delayed)

rotation_layout.addWidget(QLabel(\”旋转角度:\”))

rotation_layout.addWidget(self.slider_rotation)

grp_settings_layout.addLayout(rotation_layout)

# Position

pos_layout=QHBoxLayout()

self.combo_pos=QComboBox()

positions=[

\”左上\”,\”中上\”,\”右上\”,

\”左中\”,\”正中\”,\”右中\”,

\”左下\”,\”中下\”,\”右下\”,

\”平铺 (Tile)\”

]

self.combo_pos.addItems(positions)

self.combo_pos.setCurrentText(\”右下\”)

self.combo_pos.currentIndexChanged.connect(self.update_preview_delayed)

pos_layout.addWidget(QLabel(\”位置:\”))

pos_layout.addWidget(self.combo_pos)

grp_settings_layout.addLayout(pos_layout)

# Spacing (only for tile)

spacing_layout=QHBoxLayout()

self.spin_spacing=QSpinBox()

self.spin_spacing.setRange(0,500)

self.spin_spacing.setValue(100)

self.spin_spacing.valueChanged.connect(self.update_preview_delayed)

spacing_layout.addWidget(QLabel(\”间距 (平铺):\”))

spacing_layout.addWidget(self.spin_spacing)

grp_settings_layout.addLayout(spacing_layout)

grp_settings.setLayout(grp_settings_layout)

left_layout.addWidget(grp_settings)

left_layout.addStretch()

# Action Buttons

self.btn_start=QPushButton(\”开始处理\”)

self.btn_start.setMinimumHeight(40)

self.btn_start.setStyleSheet(\”font-weight: bold; font-size: 14px;\”)

self.btn_start.clicked.connect(self.start_processing)

left_layout.addWidget(self.btn_start)

self.progress_bar=QProgressBar()

left_layout.addWidget(self.progress_bar)

# Right Panel (Preview)

right_panel=QWidget()

right_layout=QVBoxLayout(right_panel)

main_layout.addWidget(right_panel)

right_layout.addWidget(QLabel(\”预览 (点击文件列表查看):\”))

self.scroll_area=QScrollArea()

self.scroll_area.setWidgetResizable(True)

self.lbl_preview=QLabel()

self.lbl_preview.setAlignment(Qt.AlignCenter)

self.scroll_area.setWidget(self.lbl_preview)

right_layout.addWidget(self.scroll_area)

# Debounce timer for preview update to avoid lag

self.preview_timer=None

defadd_files(self):

files, _=QFileDialog.getOpenFileNames(self,\”选择图片\”, \”\”, \”Images (*.png*.jpg*.jpeg*.bmp)\”)

iffiles:

self.image_files.extend(files)

self.update_file_list()

ifnotself.edit_output.text():

self.edit_output.setText(os.path.dirname(files[0])+\”/watermarked\”)

defadd_folder(self):

folder=QFileDialog.getExistingDirectory(self,\”选择文件夹\”)

iffolder:

forroot, dirs, filesinos.walk(folder):

forfileinfiles:

iffile.lower().endswith((\’.png\’,\’.jpg\’,\’.jpeg\’,\’.bmp\’)):

self.image_files.append(os.path.join(root,file))

self.update_file_list()

ifnotself.edit_output.text():

self.edit_output.setText(folder+\”/watermarked\”)

defclear_files(self):

self.image_files=[]

self.update_file_list()

self.lbl_preview.clear()

defupdate_file_list(self):

self.list_files.clear()

forfinself.image_files:

self.list_files.addItem(os.path.basename(f))

ifself.image_files:

self.list_files.setCurrentRow(0)

defbrowse_output(self):

folder=QFileDialog.getExistingDirectory(self,\”选择输出目录\”)

iffolder:

self.edit_output.setText(folder)

defchoose_color(self):

color=QColorDialog.getColor()

ifcolor.isValid():

self.watermark_color=(color.red(), color.green(), color.blue())

self.lbl_color_preview.setStyleSheet(f\”background-color: {color.name()}; border: 1px solid black;\”)

self.update_preview()

defupdate_preview_delayed(self):

# In a real app, use a QTimer to debounce.

# For simplicity here, just call update_preview directly,

# but keep method name to indicate intent if we add timer later.

self.update_preview()

defupdate_preview(self):

row=self.list_files.currentRow()

ifrow <0orrow >=len(self.image_files):

return

file_path=self.image_files[row]

# Generate preview

try:

config=self.get_config()

# Use PIL to generate preview

with Image.open(file_path).convert(\”RGBA\”) as img:

# Resize for preview if too large

preview_max_size=800

ifimg.width > preview_max_sizeorimg.height > preview_max_size:

img.thumbnail((preview_max_size, preview_max_size))

# Apply watermark (Reuse logic? For now duplicate simplified logic for preview speed)

watermark=Image.new(\’RGBA\’, img.size, (0,0,0,0))

draw=ImageDraw.Draw(watermark)

font_size=config[\’font_size\’]

# Scale font size relative to preview thumbnail

# Note: config[\’font_size\’] is for the original image?

# Ideally we should scale it down. But font size is usually absolute pixels.

# If we scaled down the image, the font will look HUGE if we don\’t scale it too.

# So we need to know the original image size vs preview size.

# Let\’s read original size first

with Image.open(file_path) as orig_img:

orig_w, orig_h=orig_img.size

scale_factor=img.width/orig_w

scaled_font_size=int(font_size*scale_factor)

ifscaled_font_size <1: scaled_font_size=1

try:

font=ImageFont.truetype(\”msyh.ttc\”, scaled_font_size)

except:

font=ImageFont.load_default()

try:

font=ImageFont.truetype(\”arial.ttf\”, scaled_font_size)

except:

pass

text=config[\’text\’]

bbox=draw.textbbox((0,0), text, font=font)

text_width=bbox[2]-bbox[0]

text_height=bbox[3]-bbox[1]

max_dim=int(math.sqrt(text_width**2+text_height**2))

txt_img=Image.new(\’RGBA\’, (max_dim, max_dim), (0,0,0,0))

txt_draw=ImageDraw.Draw(txt_img)

text_x=(max_dim-text_width)//2

text_y=(max_dim-text_height)//2

color=config[\’color\’]

opacity=config[\’opacity\’]

fill_color=(color[0], color[1], color[2],int(255*opacity))

txt_draw.text((text_x, text_y), text, font=font, fill=fill_color)

ifconfig[\’rotation\’] !=0:

txt_img=txt_img.rotate(config[\’rotation\’], resample=Image.BICUBIC)

w_width, w_height=txt_img.size

ifconfig[\’position\’]==\’平铺 (Tile)\’:

scaled_spacing=int(config[\’spacing\’]*scale_factor)

step_x=int(w_width+scaled_spacing)

step_y=int(w_height+scaled_spacing)

ifstep_x <=0: step_x=w_width+10

ifstep_y <=0: step_y=w_height+10

foryinrange(0, img.height, step_y):

forxinrange(0, img.width, step_x):

watermark.paste(txt_img, (x, y), txt_img)

else:

pos_x=0

pos_y=0

margin=int(20*scale_factor)

position=config[\’position\’]

if\’左\’inposition: pos_x=margin

elif\’右\’inposition: pos_x=img.width-w_width-margin

else: pos_x=(img.width-w_width)//2

if\’上\’inposition: pos_y=margin

elif\’下\’inposition: pos_y=img.height-w_height-margin

else: pos_y=(img.height-w_height)//2

watermark.paste(txt_img, (pos_x, pos_y), txt_img)

out=Image.alpha_composite(img, watermark)

# Convert to QPixmap

ifout.mode==\”RGBA\”:

r, g, b, a=out.split()

out=Image.merge(\”RGBA\”, (b, g, r, a))

elifout.mode==\”RGB\”:

r, g, b=out.split()

out=Image.merge(\”RGB\”, (b, g, r))

im2=out.convert(\”RGBA\”)

data=im2.tobytes(\”raw\”,\”RGBA\”)

qim=QImage(data, out.size[0], out.size[1], QImage.Format_ARGB32)

pixmap=QPixmap.fromImage(qim)

self.lbl_preview.setPixmap(pixmap)

exceptException as e:

print(f\”Preview error: {e}\”)

defget_config(self):

return{

\’text\’:self.edit_text.text(),

\’font_size\’:self.spin_size.value(),

\’opacity\’:self.slider_opacity.value()/100.0,

\’rotation\’:self.slider_rotation.value(),

\’color\’:self.watermark_color,

\’position\’:self.combo_pos.currentText(),

\’spacing\’:self.spin_spacing.value()

}

defstart_processing(self):

ifnotself.image_files:

QMessageBox.warning(self,\”提示\”,\”请先添加图片！\”)

return

output_dir=self.edit_output.text()

ifnotoutput_dir:

QMessageBox.warning(self,\”提示\”,\”请选择输出目录！\”)

return

self.btn_start.setEnabled(False)

self.progress_bar.setValue(0)

config=self.get_config()

self.worker=Worker(self.image_files, output_dir, config)

self.worker.progress.connect(self.progress_bar.setValue)

self.worker.finished.connect(self.processing_finished)

self.worker.error.connect(self.processing_error)

self.worker.start()

defprocessing_finished(self, msg):

self.btn_start.setEnabled(True)

QMessageBox.information(self,\”完成\”, msg)

defprocessing_error(self, msg):

self.btn_start.setEnabled(True)

QMessageBox.critical(self,\”错误\”, msg)

if__name__==\”__main__\”:

app=QApplication(sys.argv)

# 设置全局字体，看起来更现代一点

font=QFont(\”Microsoft YaHei\”,9)

app.setFont(font)

window=WatermarkApp()

window.show()

sys.exit(app.exec_())