moved memegen to https://github.com/ewpratten/memegen
This commit is contained in:
parent
8d872a499c
commit
23606d667b
@ -1,5 +0,0 @@
|
||||
# Mini Apps
|
||||
|
||||
This directory contains things that are too complex to be considered a "script", yet too personalized to be useful to anyone else.
|
||||
|
||||
I may occasionally promote things from here to actual public repos someday.
|
||||
@ -1,228 +0,0 @@
|
||||
#! /usr/bin/env python
|
||||
import argparse
|
||||
import sys
|
||||
import logging
|
||||
import json
|
||||
import subprocess
|
||||
from PIL import Image, ImageDraw, ImageFont
|
||||
from typing import List, Dict, Any, Optional
|
||||
from pathlib import Path
|
||||
from dataclasses import dataclass
|
||||
from enum import Enum
|
||||
from datetime import datetime
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
CONFIG_DIR = Path(__file__).parent.parent / "data"
|
||||
DEFAULT_OUTPUT_DIR = Path("~/Pictures/memes").expanduser()
|
||||
|
||||
|
||||
@dataclass
|
||||
class MemeTemplate:
|
||||
image_path: Path
|
||||
config: Dict[str, Any]
|
||||
|
||||
|
||||
class HorizontalAlignment(Enum):
|
||||
LEFT = "left"
|
||||
CENTER = "center"
|
||||
RIGHT = "right"
|
||||
|
||||
|
||||
class VerticalAlignment(Enum):
|
||||
TOP = "top"
|
||||
CENTER = "center"
|
||||
BOTTOM = "bottom"
|
||||
|
||||
|
||||
def discover_templates() -> List[str]:
|
||||
# Find all directories in the templates directory
|
||||
return [p.name for p in (CONFIG_DIR / "templates").glob("*") if p.is_dir()]
|
||||
|
||||
|
||||
def load_template(name: str) -> MemeTemplate:
|
||||
logger.info(f"Loading template: {name}")
|
||||
|
||||
# Find the template directory
|
||||
template_dir = CONFIG_DIR / "templates" / name
|
||||
if not template_dir.exists():
|
||||
logger.error(f"Template {name} does not exist")
|
||||
sys.exit(1)
|
||||
|
||||
return MemeTemplate(
|
||||
image_path=template_dir / "template.png",
|
||||
config=json.loads((template_dir / "config.json").read_text()),
|
||||
)
|
||||
|
||||
|
||||
def calc_width_from_image(width_str: str, image_width: int) -> int:
|
||||
if width_str.endswith("%"):
|
||||
return int(image_width * int(width_str[:-1]) / 100)
|
||||
else:
|
||||
return int(width_str)
|
||||
|
||||
|
||||
def render_text_on_image(image: Image, text: str, zone: str, config: Dict[str, Any]):
|
||||
# NOTE: This must handle text with newlines
|
||||
# Get the zone config
|
||||
zone_config = config["zones"][zone]
|
||||
horizontal_alignment = HorizontalAlignment(zone_config["horizontal_align"])
|
||||
vertical_alignment = VerticalAlignment(zone_config["vertical_align"])
|
||||
text_width = calc_width_from_image(zone_config["width"], image.width)
|
||||
max_line_height = zone_config["max_line_height"]
|
||||
font_path = CONFIG_DIR / "fonts" / config["font"]
|
||||
|
||||
# Create the font
|
||||
font = None
|
||||
font_size = 1
|
||||
while True:
|
||||
font = ImageFont.truetype(str(font_path), font_size)
|
||||
|
||||
# Split the text into lines
|
||||
lines = text.splitlines()
|
||||
bounding_boxes = []
|
||||
for line in lines:
|
||||
bounding_boxes.append(font.getbbox(line))
|
||||
|
||||
# Calculate the height of the text
|
||||
line_height = max([bbox[3] for bbox in bounding_boxes])
|
||||
total_height = sum(
|
||||
[bbox[3] + zone_config["line_spacing"] for bbox in bounding_boxes]
|
||||
)
|
||||
max_width = max([bbox[2] for bbox in bounding_boxes])
|
||||
|
||||
# If we have a max line height, ensure we don't exceed it
|
||||
if max_line_height and line_height > max_line_height:
|
||||
font_size -= 1
|
||||
break
|
||||
|
||||
# Don't exceed the width
|
||||
if max_width > text_width:
|
||||
font_size -= 1
|
||||
break
|
||||
|
||||
# Increment the font size
|
||||
font_size += 1
|
||||
|
||||
# Determine the starting Y position
|
||||
y = zone_config["vertical_offset"]
|
||||
if vertical_alignment == VerticalAlignment.CENTER:
|
||||
y += (image.height - total_height) / 2
|
||||
elif vertical_alignment == VerticalAlignment.BOTTOM:
|
||||
y += image.height - total_height
|
||||
|
||||
# Render each line onto the image
|
||||
draw = ImageDraw.Draw(image)
|
||||
for line in text.splitlines():
|
||||
# Calculate the x position
|
||||
if horizontal_alignment == HorizontalAlignment.LEFT:
|
||||
x = zone_config["horizontal_offset"]
|
||||
elif horizontal_alignment == HorizontalAlignment.CENTER:
|
||||
x = ((image.width - font.getbbox(line)[2]) / 2) + zone_config[
|
||||
"horizontal_offset"
|
||||
]
|
||||
elif horizontal_alignment == HorizontalAlignment.RIGHT:
|
||||
x = (image.width - font.getbbox(line)[2]) + zone_config["horizontal_offset"]
|
||||
else:
|
||||
raise ValueError(f"Invalid horizontal alignment: {horizontal_alignment}")
|
||||
|
||||
# Render the text
|
||||
draw.text(
|
||||
(x, y),
|
||||
line,
|
||||
fill=tuple(config["fill_color"]),
|
||||
stroke_fill=tuple(config["stroke_color"]),
|
||||
stroke_width=config["stroke_width"],
|
||||
font=font,
|
||||
)
|
||||
|
||||
# Increment the y position
|
||||
y += line_height + zone_config["line_spacing"]
|
||||
|
||||
|
||||
def main() -> int:
|
||||
# Handle program arguments
|
||||
ap = argparse.ArgumentParser(prog="memegen", description="Generates memes")
|
||||
ap.add_argument(
|
||||
"template", help="The template to use", choices=discover_templates()
|
||||
)
|
||||
ap.add_argument("--top-text", help="Top text (if applicable)")
|
||||
ap.add_argument("--bottom-text", help="Bottom text (if applicable)")
|
||||
ap.add_argument(
|
||||
"--keep-case", help="Keep the case of the text", action="store_true"
|
||||
)
|
||||
ap.add_argument("--output", "-o", help="Output file path")
|
||||
ap.add_argument(
|
||||
"--no-show", help="Don't show the image after creation", action="store_true"
|
||||
)
|
||||
ap.add_argument(
|
||||
"-v", "--verbose", help="Enable verbose logging", action="store_true"
|
||||
)
|
||||
args = ap.parse_args()
|
||||
|
||||
# Configure logging
|
||||
logging.basicConfig(
|
||||
level=logging.DEBUG if args.verbose else logging.INFO,
|
||||
format="%(levelname)s: %(message)s",
|
||||
)
|
||||
|
||||
# Load the template
|
||||
template = load_template(args.template)
|
||||
template_supports_top_text = "top" in template.config["zones"]
|
||||
template_supports_bottom_text = "bottom" in template.config["zones"]
|
||||
|
||||
# Ensure we have text
|
||||
if args.top_text and not template_supports_top_text:
|
||||
logger.error(f"Template {args.template} does not support top text")
|
||||
sys.exit(1)
|
||||
if args.bottom_text and not template_supports_bottom_text:
|
||||
logger.error(f"Template {args.template} does not support bottom text")
|
||||
sys.exit(1)
|
||||
if not args.top_text and not args.bottom_text:
|
||||
logger.error("No text provided")
|
||||
if not all([template_supports_top_text, template_supports_bottom_text]):
|
||||
required_text = "top" if template_supports_top_text else "bottom"
|
||||
logger.error(
|
||||
f"Template {args.template} requires the --{required_text}-text argument"
|
||||
)
|
||||
sys.exit(1)
|
||||
|
||||
# Transform the text
|
||||
# fmt:off
|
||||
top_text = args.top_text.upper() if args.top_text and (not args.keep_case) else args.top_text
|
||||
bottom_text = args.bottom_text.upper() if args.bottom_text and (not args.keep_case) else args.bottom_text
|
||||
top_text = top_text.replace("\\n", "\n").replace("\\N", "\n") if top_text else None
|
||||
bottom_text = bottom_text.replace("\\n", "\n").replace("\\N", "\n") if bottom_text else None
|
||||
# fmt: on
|
||||
|
||||
# Load the image
|
||||
image = Image.open(template.image_path)
|
||||
|
||||
# Render the text
|
||||
if top_text:
|
||||
render_text_on_image(image, top_text, "top", template.config)
|
||||
if bottom_text:
|
||||
render_text_on_image(image, bottom_text, "bottom", template.config)
|
||||
|
||||
# Build the output path
|
||||
output_path = (
|
||||
Path(args.output)
|
||||
if args.output
|
||||
else (
|
||||
DEFAULT_OUTPUT_DIR
|
||||
/ f"meme-{datetime.now().strftime('%Y-%m-%d-%H-%M-%S')}.{args.template}.png"
|
||||
)
|
||||
)
|
||||
output_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Save the image
|
||||
image.save(output_path)
|
||||
|
||||
# Show the image
|
||||
if not args.no_show:
|
||||
subprocess.run(["xdg-open", str(output_path)])
|
||||
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(main())
|
||||
Binary file not shown.
@ -1,17 +0,0 @@
|
||||
{
|
||||
"font": "impact.ttf",
|
||||
"fill_color": [255, 255, 255],
|
||||
"stroke_color": [0, 0, 0],
|
||||
"stroke_width": 2,
|
||||
"zones": {
|
||||
"bottom": {
|
||||
"horizontal_align": "center",
|
||||
"horizontal_offset": 0,
|
||||
"vertical_align": "bottom",
|
||||
"vertical_offset": -50,
|
||||
"width": "80%",
|
||||
"max_line_height": 50,
|
||||
"line_spacing": 5
|
||||
}
|
||||
}
|
||||
}
|
||||
Binary file not shown.
|
Before Width: | Height: | Size: 414 KiB |
@ -1,34 +0,0 @@
|
||||
{
|
||||
"font": "impact.ttf",
|
||||
"fill_color": [
|
||||
0,
|
||||
0,
|
||||
0
|
||||
],
|
||||
"stroke_color": [
|
||||
0,
|
||||
0,
|
||||
0
|
||||
],
|
||||
"stroke_width": 0,
|
||||
"zones": {
|
||||
"top": {
|
||||
"horizontal_align": "right",
|
||||
"horizontal_offset": -5,
|
||||
"vertical_align": "top",
|
||||
"vertical_offset": 75,
|
||||
"width": "45%",
|
||||
"max_line_height": 50,
|
||||
"line_spacing": 5
|
||||
},
|
||||
"bottom": {
|
||||
"horizontal_align": "right",
|
||||
"horizontal_offset": -5,
|
||||
"vertical_align": "bottom",
|
||||
"vertical_offset": -75,
|
||||
"width": "45%",
|
||||
"max_line_height": 50,
|
||||
"line_spacing": 5
|
||||
}
|
||||
}
|
||||
}
|
||||
Binary file not shown.
|
Before Width: | Height: | Size: 107 KiB |
@ -1,17 +0,0 @@
|
||||
{
|
||||
"font": "impact.ttf",
|
||||
"fill_color": [255, 255, 255],
|
||||
"stroke_color": [0, 0, 0],
|
||||
"stroke_width": 2,
|
||||
"zones": {
|
||||
"top": {
|
||||
"horizontal_align": "center",
|
||||
"horizontal_offset": 0,
|
||||
"vertical_align": "top",
|
||||
"vertical_offset": 5,
|
||||
"width": "80%",
|
||||
"max_line_height": 80,
|
||||
"line_spacing": 5
|
||||
}
|
||||
}
|
||||
}
|
||||
Binary file not shown.
|
Before Width: | Height: | Size: 166 KiB |
Loading…
x
Reference in New Issue
Block a user