#!/usr/bin/env python3
"""
生成封面图片
"""
from PIL import Image, ImageDraw, ImageFont
import os

# 创建黑色背景图片 (720x1280)
width, height = 720, 1280
img = Image.new('RGB', (width, height), color='black')
draw = ImageDraw.Draw(img)

# 尝试使用系统中文字体
font_paths = [
    "C:/Windows/Fonts/msyh.ttc",      # 微软雅黑
    "C:/Windows/Fonts/simhei.ttf",    # 黑体
    "C:/Windows/Fonts/simsun.ttc",    # 宋体
]

title_font = None
subtitle_font = None

for font_path in font_paths:
    if os.path.exists(font_path):
        try:
            title_font = ImageFont.truetype(font_path, 48)
            subtitle_font = ImageFont.truetype(font_path, 32)
            print(f"使用字体: {font_path}")
            break
        except:
            continue

if title_font is None:
    print("未找到中文字体，使用默认字体")
    title_font = ImageFont.load_default()
    subtitle_font = ImageFont.load_default()

# 标题文字
title = "师徒一场_张雪峰走好_上"
subtitle = "源自：@梁老师的故事汇"

# 计算文字位置（居中）
title_bbox = draw.textbbox((0, 0), title, font=title_font)
title_width = title_bbox[2] - title_bbox[0]
title_x = (width - title_width) // 2
title_y = height // 2 - 80

subtitle_bbox = draw.textbbox((0, 0), subtitle, font=subtitle_font)
subtitle_width = subtitle_bbox[2] - subtitle_bbox[0]
subtitle_x = (width - subtitle_width) // 2
subtitle_y = height // 2 + 50

# 绘制文字
draw.text((title_x, title_y), title, font=title_font, fill='white')
draw.text((subtitle_x, subtitle_y), subtitle, font=subtitle_font, fill='gray')

# 保存图片
output_path = "shitu_cover.jpg"
img.save(output_path, quality=95)
print(f"图片已保存: {output_path}")
print(f"图片大小: {os.path.getsize(output_path)} 字节")