add step_on

meme_generator/memes/step_on/__init__.py

@@ -0,0 +1,31 @@
+from pathlib import Path
+from typing import List
+
+from PIL.Image import Image as IMG
+from pil_utils import BuildImage
+
+from meme_generator import add_meme
+from meme_generator.utils import save_gif
+
+img_dir = Path(__file__).parent / "images"
+
+
+def step_on(images: List[BuildImage], texts, args):
+    img = images[0].convert("RGBA").resize((100, 100), keep_ratio=True)
+    frames: List[IMG] = []
+    locs = [
+        (104, 72, 32, 185, -25),
+        (104, 72, 32, 185, -25),
+        (90, 73, 51, 207, 0),
+        (88, 78, 51, 202, 0),
+        (88, 86, 49, 197, 0),
+    ]
+    for i in range(5):
+        frame = BuildImage.open(img_dir / f"{i}.png")
+        w, h, x, y, angle = locs[i]
+        frame.paste(img.resize((w, h)).rotate(angle, expand=True), (x, y), below=True)
+        frames.append(frame.image)
+    return save_gif(frames, 0.07)
+
+
+add_meme("step_on", step_on, min_images=1, max_images=1, keywords=["踩"])