Python 将文件名写入图片水印

import os
from time import sleep
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
a=998 #初始化图片编号
def search(path):
    for x in os.listdir(path): #遍历目录下的所有文件和文件夹
        filePath = (path+'\\'+x)
        if os.path.isfile(filePath) and x.endswith('.jpg'): #判断文件x是否文件且它的后缀是否是.jpg
            yield x   #生成器
def ger(x):
    print(x)
    dir = r'D:\Downloads\工作照\25-26\阳小中计工'#保存位置
    filePath = os.path.join(dir,x) #保存文件名
    font = ImageFont.truetype('simsun.ttc',size=80) #添加字体文件和字体大小
    im=Image.open(filePath)
    draw = ImageDraw.Draw(im)  #构建画图对象
    draw.text((120,550,im.size[0]-100,im.size[1]-185), x[:x.index('.')], fill='red',font=font)#添加水印
    dirSave = r'D:\Downloads\工作照\25-26\new'#保存位置
    filename = os.path.join(dirSave,'25-26区间-'+str(a)+'.jpg') #保存文件名
    sleep(1)  #睡眠时间
    im.save(filename) #保存文件
for y in search(r'D:\Downloads\工作照\25-26\阳小中计工'):
    a=a+1  #每次开始保存文件，a就加一
    ger(y) #添加水印并保存图片