from PIL import Image, ImageDraw
import os, random
 
water_path = './watermark'  #  水印文件夾路徑
watermark_list = os.listdir(water_path)
 
def do(file_name):
    # 打開原始圖片和水印圖片
    original_image = Image.open(file_name)
    random_mark = random.choice(watermark_list)  # 選擇隨機(jī)水印圖片
    # print(random_mark)
    watermark_image = Image.open(f'{water_path}/{random_mark}')
 
    # 將水印圖片縮放到合適的大小
    watermark_width, watermark_height = watermark_image.size
    base_width, base_height = original_image.size
 
    watermark_image = watermark_image.resize((base_width,int(base_width/watermark_width*watermark_height)))
 
    # 創(chuàng)建一個(gè)和原始圖片相同大小的空白畫布
    canvas = Image.new('RGBA', original_image.size)
 
    # 將原始圖片粘貼到畫布上
    canvas.paste(original_image, (0, 0))
 
    # 將水印圖片粘貼到畫布上
    canvas.paste(watermark_image, (0, 0))
 
    # 保存帶有水印的圖片
    name = file_name.split("\\")[-1]
    canvas.save(f'./output/do_{name}')  # 成品保存路徑
    print(f">>>>  {name} 添加  {random_mark}  水印  成功。")
 
 
for root, dirs, files in os.walk("./original"):  # 需要加水印的文件夾路徑
    for name in files:
        file_name = os.path.join(root, name)
        do(file_name)

from PIL import Image, ImageDraw, ImageFont
 
def add_watermark(input_image_path, output_image_path, watermark_text):
    # 打開原始圖片
    base_image = Image.open(input_image_path).convert("RGBA")
 
    # 創(chuàng)建一個(gè)可以在給定圖像上繪圖的對(duì)象
    draw = ImageDraw.Draw(base_image)
 
    # 設(shè)置字體和大小
    font = ImageFont.truetype("arial.ttf", 15)  # 水印文字的字體，大小，這是英文字體不支持中文，換成自己的字體即可
 
    # 在圖片上添加水印，設(shè)置水印位置和顏色等
    draw.text((20,20), watermark_text, fill=color, font=font)
 
    # 保存帶有水印的圖片
    base_image.save(output_image_path)
 
# 使用示例
add_watermark("../img-test/orig.jpg",  # 需加水印的圖片
              "../img-test/orig--out.png",   # 最終的圖片
              "Watermark Text")  # 水印文字

from PIL import Image
 
# 打開背景圖片和水印圖片
background = Image.open('../img-test/orig.jpg')  # 需加水印的圖片
watermark = Image.open('../img-test/mark.png').convert("RGBA")  # 透明背景的水印圖片
 
 
# 將水印圖片粘貼到背景圖片上
background.paste(watermark, (0, 0),mask=watermark)  # 水印位置(0,0)
 
# 保存合并后的圖片
background.save('../img-test/orig_mark.png')