diff --git a/config.py b/config.py index 2b3ebfb..c4806c3 100644 --- a/config.py +++ b/config.py @@ -26,6 +26,12 @@ TEMPLATES = { "padding": 20, "label": "Владимир Владимирович Путин самый лучший президент", }, + "2": { + "path": BASE_DIR / "maket" / "2.png", + "anchor": (40, 40), + "box_size": (582, 346), + "label": "Макет 2", + }, } # --- Расписание --- diff --git a/main.py b/main.py index 957f558..2e1dc6e 100644 --- a/main.py +++ b/main.py @@ -262,8 +262,12 @@ def generate_maket(text: str, template_id: str = "1") -> BytesIO: raise FileNotFoundError(f"Шаблон {template_id} не найден") with Image.open(tpl["path"]).convert("RGB") as img: - anchor, padding = tpl["anchor"], tpl["padding"] - max_w, max_h = img.width - anchor[0] - padding, img.height - anchor[1] - padding + anchor = tpl["anchor"] + if "box_size" in tpl: + max_w, max_h = tpl["box_size"] + else: + padding = tpl.get("padding", 0) + max_w, max_h = img.width - anchor[0] - padding, img.height - anchor[1] - padding font, lines, stroke, spacing = fit_text_in_box(text, max_w, max_h) draw = ImageDraw.Draw(img) diff --git a/maket/2.png b/maket/2.png new file mode 100644 index 0000000..550a1e5 Binary files /dev/null and b/maket/2.png differ