from PIL import Image,ImageDraw,ImageFont
from os import path, listdir
import sys
def search(root, target):
'''
定义搜索文件函数,接收两个参数,
第一个参数是搜索的根目录,
第二个参数是搜索的目标文件类型
'''
#定义一个数组,用于存储找到的文件的路径
fileList=[]
#取出搜索根目录下的文件和文件夹放入items数组
items = listdir(root)
print(items)
#依次检查,如果是目录就继续向下搜索,如果是文件则判断文件类型是否匹配
for item in items:
filepath = path.join(root, item)
#依次向后增加路径
if path.isdir(filepath):
fileList += search(filepath, target)
elif path.isfile(filepath):
if filepath.split('.')[-1] == target:
fileList.append(filepath)
print('[+]', filepath)
return fileList
def main(argv):
path = argv[1]
#从哪个文件夹里找文件
target = argv[2]
#找到文件将其放置哪个文件里去
files = search(path, target)
print(files[0])
for i in range(0,len(files)):
#图片路径,必须是当前目录下
imagefile = files[i]
#打开图片
imageInfo = Image.open(imagefile)
#获取图片尺寸
print(imageInfo.size)
#设置图片水印的字体的字号
fontOne = ImageFont.truetype("C:\Windows\Fonts\AGENCYR.TTF", 36)
fontTwo = ImageFont.truetype("C:\Windows\Fonts\AGENCYR.TTF", 36)
#创建Draw对象,用于之后绘制文字
draw=ImageDraw.Draw(imageInfo)
#设置水印文字的位置(x,y),文本,颜色,字体字号
draw.text((imageInfo.size[0] / 11, imageInfo.size[1] / 20), "我爱学习,学习使我妈快乐", fill=(255, 0, 0), font=fontOne)
draw.text((imageInfo.size[0] / 11, imageInfo.size[1] / 20 + 40), "我妈快乐全家快乐", fill=(20, 150, 200), font=fontTwo)
# 图片预览
imageInfo.show()
#图片保存
imageInfo.save(files[i])
if __name__ == '__main__':
main(sys.argv)
emmm 还有点小问题 明天改完