python 给目录下的图片批量加水印

"""
作者：皮蛋编程(https://www.pidancode.com)
创建日期：2022/3/25
功能描述：python 给目录下的图片批量加水印
"""

from PIL import Image
import os


def getlogo(x1, y1):
    # 加载logo文件
    im = Image.open("./" + "pidancode.com.png")
    z1 = int(x1)
    z2 = int(y1)
    in2 = im.resize((z1, z2))
    # in2.show()
    return in2


list = os.listdir("./")
for a in list:
    # print a
    if os.path.isdir(a):
        print("is path---------")
        continue

    path = "./get/"
    if not os.path.isdir(path):
        os.mkdir(path)
    if a.split('.')[1].lower() in ['jpg', 'gif']:
        im = Image.open("./" + a)
        x = im.size[0]
        y = im.size[1]
        x1 = int(x * 0.5)
        y1 = int(y * 0.7)
        logo = getlogo((x - x1), (y - y1))
        box = (x1, y1, x, y)
        im.paste(logo, box)
        im.save(path + a)
        print('================{}'.format(a))