464 lines
13 KiB
Python
464 lines
13 KiB
Python
"""产品推介材料宣传长图固定模板渲染器。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from pathlib import Path
|
||
from typing import Any
|
||
|
||
from app.service.promotion_renderer import STYLE_CONFIG
|
||
|
||
|
||
class PromotionPosterRenderer:
|
||
"""把同一份结构化草稿填入竖版宣传长图模板。"""
|
||
|
||
WIDTH = 1800
|
||
HEIGHT = 2600
|
||
|
||
def render(
|
||
self,
|
||
draft: dict[str, Any],
|
||
output_path: str | Path,
|
||
*,
|
||
photo_path: str | None = None,
|
||
chart_paths: list[str] | None = None,
|
||
) -> str:
|
||
try:
|
||
from PIL import Image, ImageDraw, ImageFont
|
||
except ImportError as exc:
|
||
raise RuntimeError("缺少 Pillow 依赖，无法生成宣传长图") from exc
|
||
|
||
style_code = str(draft.get("style_code") or "balanced_allocation")
|
||
style = STYLE_CONFIG.get(style_code, STYLE_CONFIG["balanced_allocation"])
|
||
colors = {
|
||
"primary": f"#{style['primary']}",
|
||
"secondary": f"#{style['secondary']}",
|
||
"accent": f"#{style['accent']}",
|
||
"ink": f"#{style['ink']}",
|
||
"muted": f"#{style['muted']}",
|
||
"paper": f"#{style['background']}",
|
||
"white": f"#{style['surface']}",
|
||
"line": f"#{style['secondary']}",
|
||
"headline": f"#{style['headline']}",
|
||
}
|
||
fonts = self._fonts(ImageFont)
|
||
canvas = Image.new("RGB", (self.WIDTH, self.HEIGHT), colors["paper"])
|
||
draw = ImageDraw.Draw(canvas)
|
||
chapters = {
|
||
str(item.get("title", "")): item
|
||
for item in draft.get("chapters", [])
|
||
if item.get("title")
|
||
}
|
||
|
||
self._draw_cover(draw, canvas, draft, colors, fonts)
|
||
self._draw_section_header(
|
||
draw,
|
||
"产品定位与投资策略",
|
||
90,
|
||
500,
|
||
colors,
|
||
fonts,
|
||
)
|
||
self._draw_two_columns(
|
||
draw,
|
||
chapters.get("产品基本信息", {}).get("body", ""),
|
||
chapters.get("投资范围、策略与限制", {}).get("body", ""),
|
||
90,
|
||
590,
|
||
colors,
|
||
fonts,
|
||
)
|
||
self._draw_section_header(
|
||
draw,
|
||
"基金经理与投研团队",
|
||
90,
|
||
1030,
|
||
colors,
|
||
fonts,
|
||
)
|
||
self._draw_manager(
|
||
draw,
|
||
canvas,
|
||
chapters.get("管理人及投研团队", {}).get("body", ""),
|
||
photo_path,
|
||
90,
|
||
1115,
|
||
colors,
|
||
fonts,
|
||
)
|
||
self._draw_section_header(
|
||
draw,
|
||
"历史业绩与风险指标",
|
||
90,
|
||
1550,
|
||
colors,
|
||
fonts,
|
||
)
|
||
chart_path = next(
|
||
(
|
||
item
|
||
for item in (chart_paths or [])
|
||
if item and Path(item).exists()
|
||
),
|
||
None,
|
||
)
|
||
if chart_path:
|
||
self._draw_chart(canvas, chart_path, 90, 1635, 1620, 520)
|
||
else:
|
||
self._draw_note(
|
||
draw,
|
||
"当前未提供可展示的业绩曲线附件",
|
||
90,
|
||
1735,
|
||
1620,
|
||
260,
|
||
colors,
|
||
fonts,
|
||
)
|
||
self._draw_metrics(
|
||
draw,
|
||
chapters.get("收益与风险指标", {}).get("body", ""),
|
||
90,
|
||
2185,
|
||
colors,
|
||
fonts,
|
||
)
|
||
self._draw_footer(draw, colors, fonts)
|
||
|
||
output = Path(output_path)
|
||
output.parent.mkdir(parents=True, exist_ok=True)
|
||
canvas.save(output, format="PNG", optimize=True)
|
||
return str(output)
|
||
|
||
@staticmethod
|
||
def _fonts(image_font: Any) -> dict[str, Any]:
|
||
candidates = (
|
||
Path(r"C:\Windows\Fonts\msyh.ttc"),
|
||
Path(r"C:\Windows\Fonts\simhei.ttf"),
|
||
Path("/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc"),
|
||
Path("/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc"),
|
||
)
|
||
font_path = next((item for item in candidates if item.exists()), None)
|
||
if font_path is None:
|
||
return {
|
||
"title": image_font.load_default(),
|
||
"section": image_font.load_default(),
|
||
"body": image_font.load_default(),
|
||
"small": image_font.load_default(),
|
||
}
|
||
return {
|
||
"title": image_font.truetype(str(font_path), 58),
|
||
"section": image_font.truetype(str(font_path), 32),
|
||
"body": image_font.truetype(str(font_path), 24),
|
||
"small": image_font.truetype(str(font_path), 19),
|
||
"metric": image_font.truetype(str(font_path), 29),
|
||
}
|
||
|
||
def _draw_cover(
|
||
self,
|
||
draw: Any,
|
||
canvas: Any,
|
||
draft: dict[str, Any],
|
||
colors: dict[str, str],
|
||
fonts: dict[str, Any],
|
||
) -> None:
|
||
draw.rectangle((0, 0, self.WIDTH, 430), fill=colors["primary"])
|
||
draw.rectangle((0, 430, self.WIDTH, 462), fill=colors["accent"])
|
||
title = str(draft.get("title") or draft.get("subtitle") or "基金产品推介材料")
|
||
self._draw_wrapped(
|
||
draw,
|
||
title,
|
||
90,
|
||
86,
|
||
1550,
|
||
fonts["title"],
|
||
colors["headline"],
|
||
max_lines=2,
|
||
line_gap=14,
|
||
)
|
||
draw.text(
|
||
(95, 300),
|
||
"产品推介材料",
|
||
font=fonts["section"],
|
||
fill=colors["headline"],
|
||
)
|
||
draw.text(
|
||
(95, 350),
|
||
"仅供内部审核和专业投顾使用",
|
||
font=fonts["small"],
|
||
fill=colors["secondary"],
|
||
)
|
||
# 右上角使用抽象数据线和节点，模仿参考图的金融科技视觉。
|
||
points = [(1330, 150), (1410, 188), (1490, 146), (1570, 212), (1680, 170)]
|
||
draw.line(points, fill=colors["secondary"], width=5)
|
||
for x, y in points:
|
||
draw.ellipse((x - 10, y - 10, x + 10, y + 10), fill=colors["accent"])
|
||
draw.line((1320, 272, 1710, 272), fill=colors["secondary"], width=3)
|
||
|
||
def _draw_section_header(
|
||
self,
|
||
draw: Any,
|
||
title: str,
|
||
x: int,
|
||
y: int,
|
||
colors: dict[str, str],
|
||
fonts: dict[str, Any],
|
||
) -> None:
|
||
draw.rectangle((x, y, x + 720, y + 62), fill=colors["accent"])
|
||
draw.polygon(
|
||
((x + 720, y), (x + 760, y + 31), (x + 720, y + 62)),
|
||
fill=colors["accent"],
|
||
)
|
||
draw.text((x + 24, y + 14), title, font=fonts["section"], fill=colors["white"])
|
||
|
||
def _draw_two_columns(
|
||
self,
|
||
draw: Any,
|
||
left_body: str,
|
||
right_body: str,
|
||
x: int,
|
||
y: int,
|
||
colors: dict[str, str],
|
||
fonts: dict[str, Any],
|
||
) -> None:
|
||
column_width = 775
|
||
self._draw_text_panel(
|
||
draw,
|
||
left_body,
|
||
x,
|
||
y,
|
||
column_width,
|
||
365,
|
||
colors,
|
||
fonts,
|
||
)
|
||
self._draw_text_panel(
|
||
draw,
|
||
right_body,
|
||
x + 850,
|
||
y,
|
||
column_width,
|
||
365,
|
||
colors,
|
||
fonts,
|
||
)
|
||
|
||
def _draw_manager(
|
||
self,
|
||
draw: Any,
|
||
canvas: Any,
|
||
body: str,
|
||
photo_path: str | None,
|
||
x: int,
|
||
y: int,
|
||
colors: dict[str, str],
|
||
fonts: dict[str, Any],
|
||
) -> None:
|
||
panel_width = 1620
|
||
panel_height = 355
|
||
draw.rectangle(
|
||
(x, y, x + panel_width, y + panel_height),
|
||
fill=colors["white"],
|
||
outline=colors["line"],
|
||
width=2,
|
||
)
|
||
photo_box = (x + 30, y + 28, x + 330, y + 325)
|
||
if photo_path and Path(photo_path).exists():
|
||
from PIL import Image
|
||
|
||
with Image.open(photo_path) as source:
|
||
portrait = self._contain(source.convert("RGB"), 300, 297)
|
||
canvas.paste(portrait, (photo_box[0], photo_box[1]))
|
||
else:
|
||
draw.rectangle(photo_box, fill=colors["secondary"])
|
||
draw.text(
|
||
(photo_box[0] + 70, photo_box[1] + 125),
|
||
"未上传头像",
|
||
font=fonts["small"],
|
||
fill=colors["muted"],
|
||
)
|
||
draw.line((x + 365, y + 28, x + 365, y + 325), fill=colors["line"], width=2)
|
||
self._draw_wrapped(
|
||
draw,
|
||
body,
|
||
x + 410,
|
||
y + 36,
|
||
1120,
|
||
fonts["body"],
|
||
colors["ink"],
|
||
max_lines=9,
|
||
line_gap=8,
|
||
)
|
||
|
||
def _draw_chart(
|
||
self,
|
||
canvas: Any,
|
||
chart_path: str,
|
||
x: int,
|
||
y: int,
|
||
width: int,
|
||
height: int,
|
||
) -> None:
|
||
from PIL import Image
|
||
|
||
with Image.open(chart_path) as source:
|
||
chart = self._contain(source.convert("RGB"), width, height)
|
||
canvas.paste(chart, (x, y))
|
||
|
||
def _draw_metrics(
|
||
self,
|
||
draw: Any,
|
||
body: str,
|
||
x: int,
|
||
y: int,
|
||
colors: dict[str, str],
|
||
fonts: dict[str, Any],
|
||
) -> None:
|
||
metrics: list[str] = []
|
||
for line in body.splitlines():
|
||
if any(keyword in line for keyword in ("产品收益", "最大回撤", "波动率", "夏普比率")):
|
||
metrics.extend(
|
||
part.strip()
|
||
for part in line.replace("。", "，").split("，")
|
||
if part.strip()
|
||
)
|
||
metrics = metrics[:4]
|
||
if not metrics:
|
||
metrics = ["收益指标待补充", "风险指标待补充"]
|
||
box_width = 375
|
||
for index, text in enumerate(metrics):
|
||
left = x + index * (box_width + 30)
|
||
draw.rectangle(
|
||
(left, y, left + box_width, y + 125),
|
||
fill=colors["secondary"],
|
||
outline=colors["line"],
|
||
width=2,
|
||
)
|
||
draw.rectangle(
|
||
(left, y, left + 12, y + 125),
|
||
fill=colors["accent"],
|
||
)
|
||
self._draw_wrapped(
|
||
draw,
|
||
text,
|
||
left + 20,
|
||
y + 28,
|
||
box_width - 40,
|
||
fonts.get("metric", fonts["body"]),
|
||
colors["primary"],
|
||
max_lines=2,
|
||
)
|
||
|
||
def _draw_text_panel(
|
||
self,
|
||
draw: Any,
|
||
body: str,
|
||
x: int,
|
||
y: int,
|
||
width: int,
|
||
height: int,
|
||
colors: dict[str, str],
|
||
fonts: dict[str, Any],
|
||
) -> None:
|
||
draw.rectangle(
|
||
(x, y, x + width, y + height),
|
||
fill=colors["white"],
|
||
outline=colors["line"],
|
||
width=2,
|
||
)
|
||
draw.rectangle((x, y, x + 16, y + height), fill=colors["accent"])
|
||
self._draw_wrapped(
|
||
draw,
|
||
body or "资料待补充",
|
||
x + 34,
|
||
y + 30,
|
||
width - 68,
|
||
fonts["body"],
|
||
colors["ink"],
|
||
max_lines=11,
|
||
line_gap=8,
|
||
)
|
||
|
||
def _draw_note(
|
||
self,
|
||
draw: Any,
|
||
text: str,
|
||
x: int,
|
||
y: int,
|
||
width: int,
|
||
height: int,
|
||
colors: dict[str, str],
|
||
fonts: dict[str, Any],
|
||
) -> None:
|
||
draw.rectangle(
|
||
(x, y, x + width, y + height),
|
||
fill=colors["secondary"],
|
||
)
|
||
self._draw_wrapped(
|
||
draw,
|
||
text,
|
||
x + 40,
|
||
y + 70,
|
||
width - 80,
|
||
fonts["body"],
|
||
colors["muted"],
|
||
max_lines=3,
|
||
)
|
||
|
||
def _draw_footer(
|
||
self,
|
||
draw: Any,
|
||
colors: dict[str, str],
|
||
fonts: dict[str, Any],
|
||
) -> None:
|
||
y = self.HEIGHT - 115
|
||
draw.rectangle((0, y, self.WIDTH, y + 8), fill=colors["primary"])
|
||
draw.text(
|
||
(90, y + 32),
|
||
"基金的过往业绩并不预示其未来表现，材料使用前须完成合规审核。",
|
||
font=fonts["small"],
|
||
fill=colors["muted"],
|
||
)
|
||
|
||
@staticmethod
|
||
def _contain(image: Any, width: int, height: int) -> Any:
|
||
from PIL import Image
|
||
|
||
ratio = min(width / image.width, height / image.height)
|
||
size = (max(1, int(image.width * ratio)), max(1, int(image.height * ratio)))
|
||
resized = image.resize(size, Image.Resampling.LANCZOS)
|
||
canvas = Image.new("RGB", (width, height), "#FFFFFF")
|
||
canvas.paste(resized, ((width - size[0]) // 2, (height - size[1]) // 2))
|
||
return canvas
|
||
|
||
@staticmethod
|
||
def _draw_wrapped(
|
||
draw: Any,
|
||
text: str,
|
||
x: int,
|
||
y: int,
|
||
width: int,
|
||
font: Any,
|
||
fill: str,
|
||
*,
|
||
max_lines: int,
|
||
line_gap: int = 6,
|
||
) -> int:
|
||
lines: list[str] = []
|
||
for paragraph in str(text).splitlines() or [""]:
|
||
current = ""
|
||
for char in paragraph:
|
||
candidate = current + char
|
||
if draw.textlength(candidate, font=font) > width and current:
|
||
lines.append(current)
|
||
current = char
|
||
else:
|
||
current = candidate
|
||
lines.append(current)
|
||
lines = lines[:max_lines]
|
||
cursor = y
|
||
line_height = int(font.size * 1.35)
|
||
for line in lines:
|
||
draw.text((x, cursor), line, font=font, fill=fill)
|
||
cursor += line_height + line_gap
|
||
return cursor
|