66 lines
2.6 KiB
Python
66 lines
2.6 KiB
Python
|
import io
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from PIL import Image, ImageDraw, ImageFont
|
||
|
|
|
||
|
|
def apply_watermark(input_img_path, output_img_path):
|
||
|
|
base = Image.open(input_img_path).convert("RGBA")
|
||
|
|
bw, bh = base.size
|
||
|
|
|
||
|
|
base_dir = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
logo_path = os.path.join(base_dir, "public", "assets", "brand", "shanghai_karate_logo.png")
|
||
|
|
qr_path = os.path.join(base_dir, "public", "assets", "brand", "dojo_qr.png")
|
||
|
|
|
||
|
|
margin = int(bw * 0.035)
|
||
|
|
|
||
|
|
# 1. School Logo without background (Top-Left Corner)
|
||
|
|
if os.path.exists(logo_path):
|
||
|
|
logo = Image.open(logo_path).convert("RGBA")
|
||
|
|
target_logo_w = int(bw * 0.28)
|
||
|
|
target_logo_h = int(logo.height * (target_logo_w / logo.width))
|
||
|
|
logo_resized = logo.resize((target_logo_w, target_logo_h), Image.Resampling.LANCZOS)
|
||
|
|
base.paste(logo_resized, (margin, margin), logo_resized)
|
||
|
|
|
||
|
|
# 2. WeChat-Scannable QR Code (Bottom-Right Corner)
|
||
|
|
if os.path.exists(qr_path):
|
||
|
|
qr = Image.open(qr_path).convert("RGBA")
|
||
|
|
target_qr_w = int(bw * 0.17)
|
||
|
|
qr_resized = qr.resize((target_qr_w, target_qr_w), Image.Resampling.LANCZOS)
|
||
|
|
qr_pad = 5
|
||
|
|
# Clean white backing ensures WeChat scans reliably even on dark or textured photos
|
||
|
|
qr_bg = Image.new("RGBA", (target_qr_w + qr_pad * 2, target_qr_w + qr_pad * 2), (255, 255, 255, 250))
|
||
|
|
qr_pos_x = bw - (target_qr_w + qr_pad * 2) - margin
|
||
|
|
qr_pos_y = bh - (target_qr_w + qr_pad * 2) - int(margin * 1.5)
|
||
|
|
|
||
|
|
base.paste(qr_bg, (qr_pos_x, qr_pos_y), qr_bg)
|
||
|
|
base.paste(qr_resized, (qr_pos_x + qr_pad, qr_pos_y + qr_pad), qr_resized)
|
||
|
|
|
||
|
|
# 3. Bottom Text: '上海健志空手道馆 · Shanghai Karate Club · Est. 2008'
|
||
|
|
draw = ImageDraw.Draw(base)
|
||
|
|
font_path = os.path.join(base_dir, "public", "fonts", "ZCOOLKuaiLe.ttf")
|
||
|
|
font_size = max(14, int(bw * 0.026))
|
||
|
|
try:
|
||
|
|
font = ImageFont.truetype(font_path, font_size)
|
||
|
|
except Exception:
|
||
|
|
font = ImageFont.load_default()
|
||
|
|
|
||
|
|
text = "上海健志空手道馆 · Shanghai Karate Club · Est. 2008"
|
||
|
|
bbox = draw.textbbox((0, 0), text, font=font)
|
||
|
|
th = bbox[3] - bbox[1]
|
||
|
|
tx = margin
|
||
|
|
ty = bh - int(margin * 1.2) - th
|
||
|
|
|
||
|
|
# Subtle drop shadow + high-contrast text
|
||
|
|
draw.text((tx + 1, ty + 1), text, font=font, fill=(0, 0, 0, 220))
|
||
|
|
draw.text((tx, ty), text, font=font, fill=(255, 240, 200, 255))
|
||
|
|
|
||
|
|
base.convert("RGB").save(output_img_path, format="JPEG", quality=94)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if len(sys.argv) >= 3:
|
||
|
|
apply_watermark(sys.argv[1], sys.argv[2])
|
||
|
|
else:
|
||
|
|
apply_watermark(
|
||
|
|
r"public/test_style1_v2.jpg",
|
||
|
|
r"public/test_watermarked_result.jpg"
|
||
|
|
)
|