优化图片引文排版：自动适配最大字号并居中的技术求助

一、现有代码问题排查 #

• 文本换行计算误差：wrap_text用字符'A'的宽度估算每行最大字符数，不同字符（如中文、宽字符）宽度差异大，导致实际换行后宽度超出限制
• 高度计算逻辑不一致：get_font_size用line_height +10计算总高度，但create_quote_image逐行累加高度再减10，溢出判断不准
• 作者文本未纳入整体校验：仅计算引文高度溢出，未考虑作者行高度，可能导致整体内容超出图片范围
• 字号遍历效率低：从最小到最大逐个尝试，大字号范围下速度慢

二、简化实现思路：指定区域内的最大字号适配 #

核心逻辑：

• 先确定文本区域的可用宽高（扣除边距）
• 用二分法替代遍历，快速定位最大适配字号（效率远高于逐个尝试）
• 统一文本尺寸计算：使用Pillow 9.1+的getsize_multiline直接获取多行文本实际宽高，避免手动累加误差
• 整体校验：将引文和作者的总高度一起纳入判断，确保整体内容不超出区域

三、现成库推荐 #

• Pillow进阶用法：Pillow 9.1+的textbbox和getsize_multiline可精准计算文本尺寸，配合二分法即可实现需求，无需额外依赖
• textwrap+Pillow组合：Python标准库textwrap负责换行，Pillow负责尺寸计算，完全满足基础排版需求
• 复杂排版需求可选用**reportlab**：支持精细的对齐、行间距控制，但学习成本稍高，适合专业场景

四、修改后的代码实现 #

import textwrap
from PIL import Image, ImageDraw, ImageFont

# 全局配置
SPACING = 20  # 引文与作者间距
LEFT_RIGHT_PADDING = 20  # 左右边距
TOP_BOTTOM_PADDING = 40  # 上下边距
IMAGE_SIZE = (1024, 1024)

def get_optimal_font_size(quote, author, font_path, author_font_scale=0.85, min_size=42, max_size=96):
    img_width, img_height = IMAGE_SIZE
    avail_width = img_width - 2 * LEFT_RIGHT_PADDING
    avail_height = img_height - 2 * TOP_BOTTOM_PADDING - SPACING  # 预留作者位置和间距

    # 二分法快速定位最大适配字号
    left, right = min_size, max_size
    best_size = min_size

    while left <= right:
        mid = (left + right) // 2
        quote_font = ImageFont.truetype(font_path, mid)
        # 计算换行后的引文尺寸
        wrapped_quote = textwrap.fill(quote, width=avail_width // quote_font.getsize('A')[0])
        quote_width, quote_height = quote_font.getsize_multiline(wrapped_quote)

        # 计算作者文本尺寸
        author_size = int(mid * author_font_scale)
        author_font = ImageFont.truetype(font_path, author_size)
        wrapped_author = textwrap.fill(author, width=avail_width // author_font.getsize('A')[0])
        author_width, author_height = author_font.getsize_multiline(wrapped_author)

        # 校验是否在可用区域内
        if quote_width <= avail_width and (quote_height + SPACING + author_height) <= avail_height:
            best_size = mid
            left = mid + 1  # 尝试更大字号
        else:
            right = mid - 1  # 字号过大，尝试更小尺寸

    return best_size

def draw_centered_text(draw, text, font, avail_width, y_offset, fill='white'):
    wrapped_text = textwrap.fill(text, width=avail_width // font.getsize('A')[0])
    text_width, text_height = font.getsize_multiline(wrapped_text)
    x = (IMAGE_SIZE[0] - text_width) // 2
    draw.text((x, y_offset), wrapped_text, font=font, fill=fill, align='center')
    return y_offset + text_height + SPACING

def create_quote_image(author, quote, font_path, img_name):
    img = Image.new('RGB', IMAGE_SIZE, color='black')
    draw = ImageDraw.Draw(img)

    # 获取最优字号
    quote_size = get_optimal_font_size(quote, author, font_path)
    author_size = int(quote_size * 0.85)

    # 绘制引文
    quote_font = ImageFont.truetype(font_path, quote_size)
    avail_width = IMAGE_SIZE[0] - 2 * LEFT_RIGHT_PADDING
    current_y = TOP_BOTTOM_PADDING
    current_y = draw_centered_text(draw, quote, quote_font, avail_width, current_y)

    # 绘制作者
    author_font = ImageFont.truetype(font_path, author_size)
    draw_centered_text(draw, author, author_font, avail_width, current_y - SPACING)  # 抵消多余间距

    img.save(img_name, 'JPEG')

# 示例调用
# create_quote_image("史蒂夫·乔布斯", "Stay hungry, stay foolish.", "path/to/your/font.ttf", "quote.jpg")

五、效果说明 #

• 短引文自动使用接近max_size的最大字号，长引文自动缩小到区域可容纳的最大尺寸
• 引文与作者均居中对齐，整体布局严格控制在图片边距内
• 二分法相比原遍历方法效率提升数倍，大字号范围下优势更明显

内容的提问来源于stack exchange，提问作者BaGRoS