为了版权安全,可以给自己设计的图片设置批量文字水印功能
思路分析
为图片添加文字水印,主要用到PIL库中的相应类,PIL是一个Python图像处理库,它支持多种文件格式,并提供了强大的图像处理和图形处理能力,本例中主要使用PIL库的Image模块、ImageDraw模块、ImageFont模块和ImageEnhance模块,作用分别如下:
- Image模块:该模块中提供了一个与模块同名的Image类,用来对图像进行处理,另外,它还提供从文件加载图像和创建新图像的功能。
- ImageDraw模块:该模块提供简单的2D图形对象,在本例中,使用该模块中提供的draw()方法来绘制水印文字。
- ImageFont模块:该模块中提供了一个与模块同名的ImageFont类,主要用来存储字体,它通常与ImageDraw.Draw.text()方法一起使用。
- ImageEnhance模块:该模块中包含了一些用于图像增强的类,本例中设置水印文字的透明度时用到该模块中Brightness类的enhance()方法
初始图片
首先对水印相关的内容进行设置,并使用Image对象的open()方法和new()方法分别打开原始图片和新建一个存储水印图片的对象;然后使用ImageDraw.Draw对象的text()方法绘制水印文字,通过ImageEnhance模块中Brightness类的enhance()方法设置水印透明度;最后使用save()方法将图像保存为png格式。
代码参考
# -*- coding: utf-8 -*-
"""
@Time : 2021/2/13 18:31
@Auth : Yvon~早安阳光
@File :shuiyin
"""
from PIL import Image, ImageDraw, ImageFont,ImageEnhance
import os
path=input('请输入要添加水印的图片所在路径:')
text=input('请输入水印文字:')
positionflag=int(input('请输入水印位置(0:左上角,1:左下角,2:右上角,3:右下角,4:居中):'))
alphavalue=float(input('请输入水印透明度(范围在0――1之间的1位小数):'))
# 设置所使用的字体
font = ImageFont.truetype(r'C:\Windows\Fonts\SIMLI.TTF', 80)
# 文字水印
def textMark(img):
try:
im = Image.open(img).convert('RGBA') # 打开原始图片,并转换为RGBA色值
newImg = Image.new('RGBA', im.size, (255, 255, 255, 0)) # 存储添加水印后的图片
imagedraw = ImageDraw.Draw(newImg) # 创建绘制对象
imgwidth, imgheight = im.size # 记录图片大小
txtwidth=font.getsize(text)[0] # 获取字体宽度
txtheight = font.getsize(text)[1] # 获取字体高度
# 设置水印文字位置
if positionflag == 0: # 左上角
position=(0,0)
elif positionflag == 1: # 左下角
position=(0,imgheight - txtheight)
elif positionflag == 2: # 右上角
position=(imgwidth - txtwidth,0)
elif positionflag == 3: # 右下角
position=(imgwidth - txtwidth, imgheight - txtheight)
elif positionflag == 4: # 居中
position=(imgwidth/2,imgheight/2)
# 绘制文字
imagedraw.text(position, text, font=font, fill="yellow")
# 设置透明度
alpha = newImg.split()[3]
alpha = ImageEnhance.Brightness(alpha).enhance(alphavalue)
newImg.putalpha(alpha)
Image.alpha_composite(im, newImg).save(img,'png')# 保存图片
except Exception as e:
print(e)
try:
list = os.listdir(path) # 遍历选择的文件夹
for i in range(0, len(list)): # 遍历文件列表
filepath = os.path.join(path, list[i]) # 记录遍历到的文件名
if os.path.isfile(filepath): # 判断是否为文件
filetype = os.path.splitext(filepath)[1] # 获取扩展名
if filetype == '.png': # 判断是否为.png
textMark(filepath) # 批量添加文字水印
print('批量添加水印完成')
except:
print('请输入一个有效路径……')