98 lines
4.0 KiB
Python
98 lines
4.0 KiB
Python
import os
|
||
import sys
|
||
from PIL import Image
|
||
|
||
# 将项目根目录添加到 Python 路径中，以便导入 poster 模块
|
||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from poster.templates.vibrant_template import VibrantTemplate
|
||
from poster.templates.business_template import BusinessTemplate
|
||
|
||
def test_templates():
|
||
"""
|
||
测试 VibrantTemplate 和 BusinessTemplate 的海报生成功能。
|
||
"""
|
||
# 定义资源和输出目录
|
||
image_dir = '/root/autodl-tmp/TCC_RESTRUCT/resource/data/images/天津冒险湾'
|
||
output_dir = 'tests/output'
|
||
os.makedirs(output_dir, exist_ok=True)
|
||
|
||
# --- 1. 测试 VibrantTemplate ---
|
||
print("正在测试 VibrantTemplate...")
|
||
try:
|
||
vibrant_config = {
|
||
'image_path': os.path.join(image_dir, '微信图片_20250703104552.jpg'),
|
||
'content': {
|
||
"title": "探索天津冒险湾",
|
||
"slogan": "奇幻水上乐园，家庭欢乐首选！",
|
||
"price": "199",
|
||
"ticket_type": "夜场票",
|
||
"content_button": "套餐内容",
|
||
"content_items": [
|
||
"体验所有水上滑梯",
|
||
"享受家庭欢乐时光",
|
||
"品尝美味海滨小吃",
|
||
"参与精彩互动表演"
|
||
],
|
||
"remarks": [
|
||
"工作日可直接入园",
|
||
"周末请提前1天预约"
|
||
],
|
||
"tag": "#夏日特惠"
|
||
}
|
||
}
|
||
vibrant_template = VibrantTemplate()
|
||
vibrant_poster = vibrant_template.generate(
|
||
image_path=vibrant_config['image_path'],
|
||
content=vibrant_config['content']
|
||
)
|
||
vibrant_output_path = os.path.join(output_dir, 'vibrant_poster_test.png')
|
||
vibrant_poster.save(vibrant_output_path)
|
||
print(f"VibrantTemplate 测试成功，海报已保存至: {vibrant_output_path}")
|
||
except Exception as e:
|
||
print(f"VibrantTemplate 测试失败: {e}")
|
||
|
||
|
||
# --- 2. 测试 BusinessTemplate ---
|
||
print("\n正在测试 BusinessTemplate...")
|
||
try:
|
||
# 保持与原始文件一致的 "hotel_info" 结构
|
||
hotel_info = {
|
||
"name": "天津冒险湾：商业合作新机遇",
|
||
"feature": "携手共创文旅新篇章",
|
||
"slogan": "打造顶级水上文旅品牌",
|
||
"price": "洽谈",
|
||
"info_list": [
|
||
{'title': '战略合作', 'desc': '与顶级品牌联合，提升影响力。'},
|
||
{'title': '市场推广', 'desc': '覆盖全媒体渠道，精准触达目标客户。'},
|
||
{'title': '活动承办', 'desc': '承接各类大型商业活动，设施齐全。'},
|
||
{'title': '投资前景', 'desc': '高回报率的文旅投资项目，前景广阔。'}
|
||
],
|
||
"footer": [
|
||
"联系我们：partner@adventurebay.com",
|
||
"地址：天津市滨海新区"
|
||
]
|
||
}
|
||
business_config = {
|
||
'top_image_path': os.path.join(image_dir, '微信图片_20250703104600.jpg'),
|
||
'bottom_image_path': os.path.join(image_dir, '微信图片_20250703104605.jpg'),
|
||
'small_image_paths': [os.path.join(image_dir, '微信图片_20250703104609.jpg')],
|
||
'content': hotel_info # 传递完整的 "hotel_info"
|
||
}
|
||
|
||
business_template = BusinessTemplate()
|
||
business_poster = business_template.generate(**business_config)
|
||
|
||
# 将 RGBA 模式转换为 RGB 模式以便保存为 JPG
|
||
if business_poster.mode == 'RGBA':
|
||
business_poster = business_poster.convert('RGB')
|
||
|
||
business_output_path = os.path.join(output_dir, 'business_poster_test.png')
|
||
business_poster.save(business_output_path)
|
||
print(f"BusinessTemplate 测试成功，海报已保存至: {business_output_path}")
|
||
except Exception as e:
|
||
print(f"BusinessTemplate 测试失败: {e}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
test_templates() |