1378 lines
45 KiB
Python
1378 lines
45 KiB
Python
from PIL import Image, ImageDraw, ImageFont
|
||
import os
|
||
from parse_html import parse_html_to_ppt,parse_html_to_poster
|
||
|
||
|
||
|
||
# 加载字体 - 使用系统路径
|
||
font_path_1 = "/usr/share/fonts/opentype/noto/NotoSansCJK-Thin.ttc"
|
||
font_path_2 = "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc"
|
||
font_path_3 = "/usr/share/fonts/opentype/noto/NotoSansCJK-Black.ttc"
|
||
font_path_4 = "/usr/share/fonts/opentype/noto/NotoSansCJK-Bold.ttc"
|
||
try:
|
||
title_font = ImageFont.truetype(font_path_3, 100) # 粗体
|
||
subtitle_font = ImageFont.truetype(font_path_1, 60) # 细
|
||
end_font = ImageFont.truetype(font_path_1, 50) # 细
|
||
|
||
section_font = ImageFont.truetype(font_path_2, 100) # 粗体
|
||
section_title_font = ImageFont.truetype(font_path_4, 40) # 粗体
|
||
section_subtitle_font = ImageFont.truetype(font_path_4, 60) # 粗体
|
||
content_bold_font = ImageFont.truetype(font_path_4, 35) # 粗体
|
||
keyword_font = ImageFont.truetype(font_path_2, 40) # 中等
|
||
|
||
content_font = ImageFont.truetype(font_path_2, 35) # 中等
|
||
footer_font = ImageFont.truetype(font_path_1, 35) # 常规
|
||
except:
|
||
raise Exception("缺少思源黑体字体文件")
|
||
|
||
|
||
#颜色组
|
||
card_bg_color = None
|
||
|
||
#颜色转换
|
||
def hex_to_rgb(hex_color):
|
||
# 去掉可能的 '#' 字符
|
||
hex_color = hex_color.lstrip('#')
|
||
|
||
# 使用 int 和 base=16 将十六进制字符串转换为整数，然后用 ',' 分隔这些值，并创建一个元组
|
||
rgb = tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
|
||
|
||
return rgb
|
||
|
||
#获取相近色
|
||
def get_similar_hsv_color(rgb, dh=0.05, ds=0.1, dv=0.1):
|
||
import colorsys
|
||
r,g,b = rgb
|
||
h, s, v = colorsys.rgb_to_hsv(r/255., g/255., b/255.)
|
||
|
||
# 调整色调、饱和度和亮度
|
||
h = (h + dh) % 1 # 循环
|
||
s = max(0, min(1, s + ds))
|
||
v = max(0, min(1, v + dv))
|
||
|
||
r_new, g_new, b_new = [int(x*255) for x in colorsys.hsv_to_rgb(h, s, v)]
|
||
return (r_new, g_new, b_new)
|
||
|
||
|
||
import re
|
||
|
||
def is_english_alpha(s):
|
||
return bool(re.fullmatch(r'[A-Za-z]+', s))
|
||
|
||
#处理多行文本
|
||
def draw_multiline_text(draw, text, position, max_width, font, fill, line_spacing=10, only_calc=False):
|
||
"""
|
||
在指定区域内绘制自动换行的文本
|
||
|
||
参数:
|
||
draw - ImageDraw对象
|
||
text - 要绘制的文本
|
||
position - 起始位置 (x, y)
|
||
max_width - 最大行宽
|
||
font - 字体对象
|
||
fill - 文本颜色
|
||
line_spacing - 行间距 (默认10)
|
||
"""
|
||
x, y = position
|
||
lines = []
|
||
|
||
|
||
# 如果文本包含换行符，先按换行符分割
|
||
paragraphs = text.split('\n')
|
||
|
||
for paragraph in paragraphs:
|
||
words = []
|
||
# 中英文混合处理：将每个字符视为独立的"词"
|
||
for char in paragraph:
|
||
if char !='\n' and char !='\r' and char !="\xa0" and not char.isspace():
|
||
words.append(char)
|
||
# # 添加空格字符作为可能的断点
|
||
# if char.isspace():
|
||
# words.append('')
|
||
|
||
space_width = draw.textlength(' ', font=font)
|
||
current_line = []
|
||
current_width = 0
|
||
|
||
for i,word in enumerate(words):
|
||
word_width = draw.textlength(word, font=font)
|
||
#中文超宽符号的处理
|
||
if word in ['“','”','《','》','『','』','·']:
|
||
#print("中文",word,word_width)
|
||
word_width += word_width
|
||
elif word.islower() or word.isdigit(): #小写字母和数字要窄一点
|
||
#print("字母数字",word,word_width)
|
||
word_width -= word_width//5
|
||
elif word=="-":
|
||
word_width = word_width//2
|
||
|
||
# 如果当前行宽度加上新词宽度超过最大宽度
|
||
if int(current_width) + int(word_width) > int(max_width) and current_line:
|
||
#print(''.join(current_line),int(current_width),int(word_width),int(max_width),word)
|
||
#如果只剩下最后一个字，就放置在末尾
|
||
if len(words) -i ==1:
|
||
current_line.append(word)
|
||
break
|
||
# 将当前行添加到行列表
|
||
lines.append(''.join(current_line))
|
||
#下一行计数
|
||
current_line = [word]
|
||
current_width = word_width
|
||
else:
|
||
current_line.append(word)
|
||
current_width += word_width
|
||
|
||
# 添加最后一行
|
||
if current_line:
|
||
lines.append(''.join(current_line))
|
||
|
||
# 计算行高
|
||
_, _, _, line_height = font.getbbox("A")
|
||
|
||
# 绘制每一行文本
|
||
for line in lines:
|
||
if not only_calc:
|
||
draw.text((x, y), line, fill=fill, font=font)
|
||
y += line_height + line_spacing
|
||
|
||
return y
|
||
|
||
#绘制图片
|
||
def draw_image(path_data,image_width):
|
||
# 如果有 data:image/...;base64, 前缀，需要去掉
|
||
if path_data.startswith('data:image'):
|
||
base64_str = path_data.split(',', 1)[1] # 取逗号后面的部分
|
||
|
||
import base64
|
||
from io import BytesIO
|
||
# 解码 Base64 为二进制
|
||
image_data = base64.b64decode(base64_str)
|
||
|
||
# 用 BytesIO 包装成“文件对象”
|
||
image_file = BytesIO(image_data)
|
||
content_img = Image.open(image_file)
|
||
else:
|
||
content_img = Image.open(path_data)
|
||
|
||
# 获取原始尺寸
|
||
original_width, original_height = content_img.size
|
||
|
||
# 计算高度的比例
|
||
width_ratio = image_width / original_width
|
||
target_height = int(original_height * width_ratio)
|
||
|
||
# 调整图像大小
|
||
resized_img = content_img.resize((image_width, target_height), Image.Resampling.LANCZOS)
|
||
return resized_img,target_height
|
||
|
||
|
||
#绘制表格
|
||
def draw_table(draw,data,table_top,table_left,table_width,header_font,data_font,text_color,header_color,tbc):
|
||
if data is None: return table_top
|
||
cell_height = 65 # 默认单元格高度
|
||
|
||
headers=data[0]
|
||
data=data[1:] #去掉第一行的空行
|
||
|
||
rows = len(data)
|
||
|
||
cols = len(headers)
|
||
#table_height = cell_height * rows
|
||
cell_width = table_width // cols
|
||
|
||
# 绘制表头圆角矩形背景
|
||
draw.rounded_rectangle(
|
||
(table_left-10, table_top, table_left+table_width+10, table_top+cell_height+15),
|
||
radius=40,
|
||
fill=header_color, # 深蓝色背景
|
||
outline=header_color
|
||
)
|
||
|
||
# # 绘制表头分割线
|
||
for i in range(1, cols):
|
||
x = table_left + i * cell_width
|
||
# 绘制垂直线条，稍微短一点以避开圆角
|
||
draw.line(
|
||
[(x, table_top + 10), (x, table_top + cell_height)],
|
||
fill=(255, 255, 255, 180), # 半透明白色
|
||
width=2
|
||
)
|
||
|
||
|
||
for col in range(cols):
|
||
text = headers[col]
|
||
# 使用textbbox计算文本尺寸
|
||
text_bbox = draw.textbbox((0, 0), text, font=header_font)
|
||
text_width = text_bbox[2] - text_bbox[0]
|
||
text_height = text_bbox[3] - text_bbox[1]
|
||
|
||
x = table_left + col * cell_width + (cell_width - text_width) // 2
|
||
y = table_top + (cell_height - text_height) // 2
|
||
draw.text((x, y), text, fill=(255, 255, 230), font=header_font)
|
||
|
||
#微调一下数据表格的大小和位置，下移15
|
||
table_top +=15
|
||
table_width = table_width - 40
|
||
table_left +=20
|
||
cell_width = table_width // cols
|
||
|
||
# 绘制数据行的起始位置
|
||
y = table_top + cell_height
|
||
|
||
for row in range(rows):
|
||
|
||
#计算最大行高
|
||
max_y = y + cell_height
|
||
for col in range(cols):
|
||
text = data[row][col]
|
||
y1 = draw_multiline_text(draw, text, (table_left,y), cell_width-20, data_font, text_color, line_spacing=10, only_calc=True)
|
||
if y1>max_y:
|
||
max_y=y1
|
||
|
||
row_height = max_y - y
|
||
|
||
for col in range(cols):
|
||
# 计算单元格位置
|
||
x1 = table_left + col * cell_width
|
||
#y1 = table_top + (row + 1) * cell_height
|
||
y1 = y
|
||
x2 = x1 + cell_width
|
||
y2 = y1 + row_height
|
||
|
||
# 交替行颜色
|
||
if row % 2 == 0:
|
||
#bg_color = (248, 250, 253) # 浅色行
|
||
bg_color = (tbc[0],tbc[1],tbc[2],100)
|
||
else:
|
||
#bg_color = (240, 245, 251) # 更浅的蓝色行
|
||
bg_color = (tbc[0],tbc[1],tbc[2],200)
|
||
|
||
# 绘制单元格背景
|
||
draw.rectangle([x1, y1, x2, y2], fill=bg_color)
|
||
|
||
# 添加数据
|
||
text = data[row][col]
|
||
|
||
draw_multiline_text(draw, text, (x1+15,y+10), cell_width-15, data_font, text_color, line_spacing=10, only_calc=False)
|
||
|
||
# # 使用textbbox计算文本尺寸
|
||
# text_bbox = draw.textbbox((0, 0), text, font=data_font)
|
||
# text_width = text_bbox[2] - text_bbox[0]
|
||
# text_height = text_bbox[3] - text_bbox[1]
|
||
|
||
# x = x1 + 15
|
||
# # 默认y位置
|
||
# y = y1 + (cell_height - text_height) // 2
|
||
# draw.text((x, y), text, fill=text_color, font=data_font)
|
||
|
||
#一行的横线
|
||
y = max_y+10
|
||
draw.line(
|
||
[(table_left, y), (table_left+table_width, y)],
|
||
fill=(225, 232, 243),
|
||
width=1
|
||
)
|
||
y +=1
|
||
|
||
#表格高度
|
||
table_height = y - table_top
|
||
|
||
# 固定高度的横线
|
||
# for i in range(1, rows):
|
||
# y = table_top + i * cell_height
|
||
# draw.line(
|
||
# [(table_left, y), (table_left+table_width, y)],
|
||
# fill=(225, 232, 243),
|
||
# width=1
|
||
# )
|
||
|
||
#竖线
|
||
for i in range(1, cols):
|
||
x = table_left + i * cell_width
|
||
draw.line(
|
||
[(x, table_top + cell_height), (x, table_top+table_height)],
|
||
fill=(225, 232, 243),
|
||
width=1
|
||
)
|
||
|
||
#表格边线，三条，, # 深蓝色背景
|
||
draw.line(
|
||
[(table_left, table_top + cell_height-5), (table_left, table_top+table_height)],
|
||
fill=header_color,
|
||
width=2
|
||
)
|
||
draw.line(
|
||
[(table_left+table_width, table_top + cell_height-5), (table_left+table_width, table_top+table_height)],
|
||
fill=header_color,
|
||
width=2
|
||
)
|
||
draw.line(
|
||
[(table_left, table_top + table_height), (table_left+table_width, table_top+table_height)],
|
||
fill=header_color,
|
||
width=2
|
||
)
|
||
return table_top+table_height+20
|
||
|
||
|
||
# 辅助函数：获取文本尺寸
|
||
def get_text_size(draw,text, font):
|
||
bbox = draw.textbbox((0, 0), text, font=font)
|
||
return bbox[2] - bbox[0], bbox[3] - bbox[1]
|
||
|
||
|
||
|
||
#绘制头部样式1： 大logo+自动居中布局的标题
|
||
def draw_header_1(img,draw,poster_data,width,title_color,subtitle_color):
|
||
# 1. 顶部标题区域,高度
|
||
title = poster_data["主标"]
|
||
subtitle = poster_data["副标"]
|
||
|
||
#主标题和副标题的宽度
|
||
title_w, title_h = get_text_size(draw,title, title_font)
|
||
subtitle_w, subtitle_h = get_text_size(draw,subtitle, subtitle_font)
|
||
|
||
|
||
# 复制logo
|
||
logo_path = poster_data["logo_path"] # logo图片路径
|
||
logo = Image.open(logo_path)
|
||
logo = logo.resize((400, 150), Image.Resampling.LANCZOS) # 调整logo大小
|
||
|
||
#标题很长，需要改变布局
|
||
if title_w > width-450 or subtitle_w > width-450:
|
||
#logo和标题上下布局
|
||
|
||
# 这里位置是以左上角为起点的坐标(x, y)
|
||
position = ((width - 400) // 2, 10) # 上面居中的位置
|
||
|
||
# 将logo复制到当前图像上
|
||
# 使用Image.paste方法，注意如果你的logo有透明度，需要使用mask参数
|
||
img.paste(logo, position, logo.convert('RGBA'))
|
||
|
||
#主标
|
||
draw.text(((width - title_w) // 2, 180), title, fill=title_color, font=title_font)
|
||
|
||
draw.text(((width - subtitle_w) // 2, 180 + title_h + 30),
|
||
subtitle, fill=subtitle_color, font=subtitle_font)
|
||
|
||
y = 180 + title_h + 30 + subtitle_h +60
|
||
else:
|
||
#logo和标题左右布局
|
||
draw.text(((width - title_w) // 2+200, 50), title, fill=title_color, font=title_font)
|
||
|
||
# 绘制副标题
|
||
subtitle_w, subtitle_h = get_text_size(draw,subtitle, subtitle_font)
|
||
draw.text(((width - subtitle_w) // 2+200, 50 + title_h + 30),
|
||
subtitle, fill=subtitle_color, font=subtitle_font)
|
||
|
||
# 这里位置是以左上角为起点的坐标(x, y)
|
||
position = (10, 10) # 左上角的位置
|
||
|
||
# 将logo复制到当前图像上
|
||
# 使用Image.paste方法，注意如果你的logo有透明度，需要使用mask参数
|
||
img.paste(logo, position, logo.convert('RGBA'))
|
||
|
||
#固定值
|
||
y=300
|
||
|
||
# 装饰元素 - 金色圆点
|
||
for i in range(5):
|
||
x = width * (0.1 + 0.2 * i)
|
||
size = 10 if i % 2 == 0 else 15
|
||
draw.ellipse([(x, y),
|
||
(x+size, y + size)],
|
||
fill=title_color)
|
||
|
||
return y
|
||
|
||
|
||
|
||
#绘制头部样式2： 小logo+ 标题居左
|
||
def draw_header_2(img,draw,poster_data,width,title_color,subtitle_color):
|
||
# 1. 顶部标题区域,高度
|
||
title = poster_data["主标"]
|
||
subtitle = poster_data["副标"]
|
||
|
||
#主标题和副标题的宽度
|
||
title_w, title_h = get_text_size(draw,title, title_font)
|
||
subtitle_w, subtitle_h = get_text_size(draw,subtitle, subtitle_font)
|
||
|
||
# 复制logo
|
||
image_width = 250 #总宽度的4分之一
|
||
logo_path = poster_data["logo_path"] # logo图片路径
|
||
logo = Image.open(logo_path)
|
||
|
||
# 获取原始尺寸
|
||
original_width, original_height = logo.size
|
||
|
||
# 计算高度的比例
|
||
width_ratio = 250 / original_width
|
||
target_height = int(original_height * width_ratio)
|
||
|
||
# 调整图像大小
|
||
logo = logo.resize((image_width, target_height), Image.Resampling.LANCZOS)
|
||
#logo和标题上下布局
|
||
|
||
# 这里位置是以左上角为起点的坐标(x, y)
|
||
position = ((width - image_width)-50, 10) # 右边的位置
|
||
|
||
# 将logo复制到当前图像上
|
||
# 使用Image.paste方法，注意如果你的logo有透明度，需要使用mask参数
|
||
img.paste(logo, position, logo.convert('RGBA'))
|
||
|
||
y = target_height +20
|
||
|
||
#主标
|
||
#draw.text((50, y), title, fill=title_color, font=title_font)
|
||
|
||
y = draw_multiline_text(
|
||
draw,
|
||
title,
|
||
(50, y),
|
||
width-100,
|
||
title_font,
|
||
title_color,
|
||
line_spacing=15,
|
||
only_calc = False
|
||
)
|
||
|
||
draw.text((50, y + 30),
|
||
subtitle, fill=subtitle_color, font=subtitle_font)
|
||
|
||
y += 30 + subtitle_h +60
|
||
|
||
draw.line(
|
||
[(0, y), (width, y)],
|
||
|
||
fill = title_color,
|
||
width=1
|
||
)
|
||
|
||
return y
|
||
|
||
|
||
#绘制头部样式2： 无logo+ 标题居左+ 日期
|
||
def draw_header_3(img,draw,poster_data,width,title_color,subtitle_color):
|
||
# 1. 顶部标题区域,高度
|
||
title = poster_data["主标"]
|
||
subtitle = poster_data["副标"]
|
||
|
||
|
||
from datetime import date
|
||
|
||
# 获取当前日期
|
||
current_date = date.today()
|
||
# 如果需要将日期格式化为特定格式的字符串，可以使用 strftime 方法
|
||
formatted_date = current_date.strftime("%Y-%m-%d")
|
||
print("格式化后的日期:", formatted_date)
|
||
|
||
#主标题和副标题的宽度
|
||
date_w, date_h = get_text_size(draw,formatted_date, keyword_font)
|
||
title_w, title_h = get_text_size(draw,title, title_font)
|
||
subtitle_w, subtitle_h = get_text_size(draw,subtitle, subtitle_font)
|
||
|
||
#日期
|
||
y = 20
|
||
|
||
#主标日期
|
||
draw.text(((width-date_w)-40, y), formatted_date, fill=subtitle_color, font=keyword_font)
|
||
|
||
y = 100
|
||
#主标
|
||
#draw.text((50, y), title, fill=title_color, font=title_font)
|
||
|
||
y = draw_multiline_text(
|
||
draw,
|
||
title,
|
||
(50, y),
|
||
width-100,
|
||
title_font,
|
||
title_color,
|
||
line_spacing=15,
|
||
only_calc = False
|
||
)
|
||
|
||
|
||
# 绘制矩形,副标题题点缀
|
||
draw.rectangle([(50-10, y+15), (50+subtitle_w+10, y+subtitle_h+25)],
|
||
fill=None,
|
||
outline=subtitle_color)
|
||
|
||
draw.text((50, y),
|
||
subtitle, fill=subtitle_color, font=subtitle_font)
|
||
|
||
y += subtitle_h +60
|
||
|
||
draw.line(
|
||
[(0, y), (width, y)],
|
||
fill = title_color,
|
||
width=4
|
||
)
|
||
|
||
return y
|
||
|
||
|
||
"""
|
||
数字卡片，卡片没有标题，有数字，通过后面增加一个区域做标题汇总
|
||
"""
|
||
def draw_content_card_1(img,draw,poster_data,width,y_start,section_colors,card_bg_color,font_color):
|
||
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
|
||
section_y_start = y_start
|
||
|
||
y = section_y_start #内容的绝对值是
|
||
|
||
#内容
|
||
for i,slide in enumerate(poster_data["内容列表"]):
|
||
|
||
# 计算位置,卡片的位置
|
||
y_pos = y
|
||
|
||
#卡片宽度
|
||
card_width = width * 0.9
|
||
card_x = int((width - card_width) / 2)
|
||
|
||
#y从标题颜色后面开始测算
|
||
y = y_pos + 50
|
||
|
||
#文字内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= True)
|
||
|
||
# 卡片高度,保障一个最小值300的空间
|
||
if y - y_pos < 150:
|
||
y += 100
|
||
|
||
#卡片高度
|
||
card_height = y - y_pos+30
|
||
# 创建圆角卡片
|
||
card = Image.new('RGBA', (int(card_width), int(card_height)), (0, 0, 0, 0))
|
||
card_draw = ImageDraw.Draw(card)
|
||
|
||
# 圆角矩形填充
|
||
card_draw.rounded_rectangle([(0, 0), (card_width, card_height)],
|
||
radius=40, fill=card_bg_color)
|
||
|
||
# 抹去上面的圆角
|
||
bar_height = 40
|
||
card_draw.rectangle([(0, 0), (card_width, bar_height)],
|
||
fill=card_bg_color)
|
||
|
||
#添加色条
|
||
bar_height = 20
|
||
card_draw.rectangle([(0, 0), (card_width, bar_height)],
|
||
fill=section_colors[i])
|
||
|
||
# 添加装饰元素
|
||
card_draw.ellipse([(card_width - 80, card_height - 80),
|
||
(card_width - 20, card_height - 20)],
|
||
fill=(255, 255, 255, 30))
|
||
|
||
# 粘贴卡片到海报
|
||
img.paste(card, (int(card_x), int(y_pos)), card)
|
||
|
||
# 数字标识
|
||
num_text = str(i+1)
|
||
num_w, num_h = get_text_size(draw,num_text, section_font)
|
||
draw.text((card_x + 25, y_pos + 60),
|
||
num_text, fill=section_colors[i], font=section_font)
|
||
|
||
#真正绘制内容
|
||
y = y_pos + 50
|
||
|
||
#文字内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= False)
|
||
|
||
|
||
# # 装饰元素 - 小圆点
|
||
# for j in range(3):
|
||
# draw.ellipse([(card_x + card_width - 180 + j*40, y_pos + card_height/2 - 5),
|
||
# (card_x + card_width - 160 + j*40, y_pos + card_height/2 + 15)],
|
||
# fill=section_colors[i])
|
||
|
||
#下一个
|
||
y = y_pos + card_height+60
|
||
|
||
# 3. 底部关键字区域，和内容列表的主标对应
|
||
keywords = [content["主标"] for content in poster_data["内容列表"]]
|
||
keyword_y = y+10
|
||
|
||
# 关键字容器
|
||
container_height = 180
|
||
container = Image.new('RGBA', (int(width*0.90), container_height), (0, 0, 0, 0))
|
||
container_draw = ImageDraw.Draw(container)
|
||
container_draw.rounded_rectangle([(0, 0), (width*0.90, container_height)],
|
||
radius=30, fill=card_bg_color)
|
||
|
||
# 粘贴容器
|
||
img.paste(container, (int(width*0.05), int(keyword_y)), container)
|
||
|
||
keyword_w = int(width*0.90) // len(keywords)
|
||
|
||
# 绘制关键字
|
||
for idx, kw in enumerate(keywords):
|
||
# 计算位置
|
||
x_pos = keyword_w //2 + keyword_w*idx + int(width*0.05)
|
||
|
||
# 绘制图标背景
|
||
icon_size = 50
|
||
draw.ellipse([(x_pos - icon_size//2, keyword_y + 40),
|
||
(x_pos + icon_size//2, keyword_y + 40 + icon_size)],
|
||
fill=section_colors[idx])
|
||
|
||
# 绘制关键字文本
|
||
kw_w, kw_h = get_text_size(draw,kw, keyword_font)
|
||
draw.text((x_pos - kw_w//2, keyword_y + 100),
|
||
kw, fill=font_color, font=keyword_font)
|
||
|
||
return y+180
|
||
|
||
|
||
"""
|
||
带标题卡片，直接在卡片前放置标题
|
||
"""
|
||
def draw_content_card_2(img,draw,poster_data,width,y_start,section_colors,card_bg_color,font_color):
|
||
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
|
||
section_y_start = y_start
|
||
|
||
#卡片宽度
|
||
card_width = width * 0.9
|
||
card_x = int((width - card_width) / 2)
|
||
|
||
#内容
|
||
y = section_y_start #内容的绝对值是
|
||
for i,slide in enumerate(poster_data["内容列表"]):
|
||
|
||
# 计算位置,卡片的位置
|
||
y_pos = y
|
||
|
||
#y从 主标
|
||
y = y_pos + 80 + 30
|
||
|
||
#文字内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= True)
|
||
|
||
# 卡片高度,保障一个最小值300的空间
|
||
if y - y_pos < 150:
|
||
y += 100
|
||
|
||
#卡片高度
|
||
card_height = y - y_pos
|
||
|
||
# 绘制主标,标题
|
||
draw.rounded_rectangle([(card_x+card_width*0.055, y_pos), (card_x+card_width*0.95, y_pos+80)],
|
||
radius=40, fill=section_colors[i])
|
||
|
||
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
|
||
draw.text((card_x + (card_width-num_w)//2, y_pos+5),
|
||
slide["主标"], fill=(255,255,255), font=section_title_font)
|
||
|
||
# 创建圆角卡片
|
||
card = Image.new('RGBA', (int(card_width), int(card_height)), (0, 0, 0, 0))
|
||
card_draw = ImageDraw.Draw(card)
|
||
|
||
# 圆角矩形填充
|
||
card_draw.rounded_rectangle([(0, 0), (card_width, card_height)],
|
||
radius=40, fill=card_bg_color)
|
||
|
||
# 抹去上面的圆角
|
||
bar_height = 50
|
||
card_draw.rectangle([(0, 0), (card_width, bar_height)],
|
||
fill=card_bg_color)
|
||
|
||
# 添加装饰元素
|
||
card_draw.ellipse([(card_width - 80, card_height - 80),
|
||
(card_width - 20, card_height - 20)],
|
||
fill=(255, 255, 255, 30))
|
||
|
||
# 粘贴卡片到海报
|
||
img.paste(card, (int(card_x), int(y_pos+100)), card)
|
||
|
||
#真正绘制内容
|
||
y = y_pos + 80 + 30
|
||
|
||
#文字内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= False)
|
||
|
||
|
||
|
||
#下一个
|
||
y = y_pos +80+30+card_height+60
|
||
|
||
return y
|
||
|
||
|
||
|
||
|
||
"""
|
||
彩色标签，直接在卡片前放置标题
|
||
"""
|
||
def draw_content_card_5(img,draw,poster_data,width,y_start,section_colors,card_bg_color,font_color):
|
||
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
|
||
section_y_start = y_start
|
||
|
||
#卡片宽度
|
||
card_width = width * 0.9
|
||
card_x = int((width - card_width) / 2)
|
||
|
||
#内容
|
||
y = section_y_start #内容的绝对值是
|
||
for i,slide in enumerate(poster_data["内容列表"]):
|
||
|
||
# 计算位置,卡片的位置
|
||
y_pos = y
|
||
|
||
#y从 主标开始，80是主标高度
|
||
y = y_pos +80 + 40
|
||
|
||
#文字内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= True)
|
||
|
||
# 卡片高度,保障一个最小值300的空间
|
||
if y - y_pos < 150:
|
||
y += 100
|
||
|
||
#卡片高度
|
||
card_height = y - y_pos -80
|
||
|
||
# 绘制主标,标题
|
||
# draw.rounded_rectangle([(card_x+card_width*0.055, y_pos), (card_x+card_width*0.95, y_pos+80)],
|
||
# radius=40, fill=section_colors[i])
|
||
|
||
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
|
||
draw.text((card_x + (card_width-num_w)//2, y_pos+5),
|
||
slide["主标"], fill=section_colors[i], font=section_title_font)
|
||
|
||
# 创建圆角卡片
|
||
card = Image.new('RGBA', (int(card_width), int(card_height)), (0, 0, 0, 0))
|
||
card_draw = ImageDraw.Draw(card)
|
||
|
||
# 圆角矩形填充
|
||
card_draw.rounded_rectangle([(0, 0), (card_width, card_height)],
|
||
radius=40,
|
||
fill=card_bg_color)
|
||
|
||
#抹去左边的圆角
|
||
bar_width = 40
|
||
card_draw.rectangle([(0, 0), (bar_width, card_height)],
|
||
fill=card_bg_color)
|
||
|
||
# 左边的彩带
|
||
bar_width = 20
|
||
card_draw.rectangle([(0, 0), (bar_width, card_height)],
|
||
fill=section_colors[i])
|
||
|
||
# 添加装饰元素
|
||
card_draw.ellipse([(card_width - 80, 10),
|
||
(card_width - 20, 10+60)],
|
||
#fill=(255, 255, 255, 30)
|
||
fill=section_colors[i]
|
||
)
|
||
|
||
# 粘贴卡片到海报
|
||
img.paste(card, (int(card_x), int(y_pos+100)), card)
|
||
|
||
#真正绘制内容
|
||
y = y_pos +80 + 40
|
||
|
||
#文字内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= False)
|
||
|
||
#下一个
|
||
y = y_pos +80+30+card_height+60
|
||
|
||
return y
|
||
|
||
"""
|
||
小圆角卡片，标题放置在卡片内部
|
||
"""
|
||
def draw_content_card_6(img,draw,poster_data,width,y_start,section_colors,card_bg_color,font_color):
|
||
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
|
||
section_y_start = y_start
|
||
|
||
#卡片宽度
|
||
card_width = width * 0.9
|
||
card_x = int((width - card_width) / 2)
|
||
|
||
#内容
|
||
y = section_y_start #内容的绝对值是
|
||
for i,slide in enumerate(poster_data["内容列表"]):
|
||
|
||
# 计算位置,卡片的位置
|
||
y_pos = y
|
||
|
||
#y, 80是主标高度
|
||
y = y_pos +80 +30
|
||
|
||
#文字内容 （含副标题）
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= True)
|
||
|
||
|
||
# 卡片高度,保障一个最小值300的空间
|
||
if y - y_pos < 150:
|
||
y += 100
|
||
|
||
#卡片高度
|
||
card_height = y - y_pos
|
||
|
||
# 创建圆角卡片
|
||
card = Image.new('RGBA', (int(card_width), int(card_height)), (0, 0, 0, 0))
|
||
card_draw = ImageDraw.Draw(card)
|
||
|
||
# 小圆角矩形填充
|
||
card_draw.rounded_rectangle([(0, 0), (card_width, card_height)],
|
||
radius=15,
|
||
fill=card_bg_color,
|
||
outline= section_colors[i])
|
||
|
||
# 粘贴卡片到海报
|
||
img.paste(card, (int(card_x), int(y_pos)), card)
|
||
|
||
#真正绘制内容
|
||
y = y_pos
|
||
|
||
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
|
||
draw.rectangle([(card_x + 15, y+20+10), (card_x + 15+10, y+20+10+num_h)],
|
||
fill= section_colors[i])
|
||
draw.text((card_x + 30, y+20),
|
||
slide["主标"], fill=section_colors[i], font=section_title_font)
|
||
|
||
y += num_h + 50
|
||
|
||
#文字内容(含副标题)
|
||
if "content" in slide:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,card_bg_color,font_color,section_colors,i,y,only_calc= False)
|
||
|
||
#下一个
|
||
y = y_pos + card_height+60
|
||
|
||
return y
|
||
|
||
|
||
#绘制内容列表
|
||
def draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= True):
|
||
ol=1
|
||
for j,content in enumerate(slide["content"]):
|
||
#文字
|
||
if "text" in content:
|
||
num_w = 40
|
||
if len(slide["content"]) ==1:
|
||
num_w = 0
|
||
|
||
font = content_font
|
||
if content["level"]==2:
|
||
num_w = 0
|
||
font = content_bold_font
|
||
|
||
if "bullet" in content:#二级标题
|
||
num_w= 15
|
||
if not only_calc:
|
||
|
||
# draw.rectangle([(card_x +110, y+15), (card_x +110+20, y+15+20)],
|
||
# fill=section_colors[i],
|
||
# #outline=font_color,
|
||
# #width=4,
|
||
# )
|
||
# 圆形
|
||
# draw.ellipse([(card_x +110, y+15), (card_x +110+25, y+15+25)],
|
||
# fill=font_color,
|
||
# outline=section_colors[i],
|
||
# width = 5,
|
||
# )
|
||
triangle_points = [
|
||
(card_x +90+15, y+10), # 顶部顶点
|
||
(card_x +90, y+10+30), # 左下顶点
|
||
(card_x +90+30, y+10+30) # 右下顶点
|
||
]
|
||
draw.polygon(triangle_points,
|
||
fill=section_colors[i],
|
||
#outline=,
|
||
#width = 5,
|
||
)
|
||
if "square" in content:
|
||
num_w= 30
|
||
if not only_calc:
|
||
# 方形填充
|
||
draw.rectangle([(card_x +110, y+15), (card_x +110+20, y+15+20)],
|
||
fill=section_colors[i],
|
||
#outline=font_color,
|
||
#width=4,
|
||
)
|
||
#顺序号
|
||
if "number" in content:
|
||
num_w= 50
|
||
if not only_calc:
|
||
# 圆点填充
|
||
draw.ellipse([(card_x +110, y+5), (card_x +110+40, y+5+40)],
|
||
fill=section_colors[i],
|
||
#outline=font_color
|
||
)
|
||
#数字
|
||
draw.text((card_x +110+10, y-2),
|
||
str(ol), fill=card_bg_color, font=content_font)
|
||
ol +=1
|
||
|
||
|
||
if content["level"]==1:
|
||
#副标题
|
||
y = draw_multiline_text(
|
||
draw,
|
||
content["text"],
|
||
(card_x +90, y),
|
||
card_width-120,
|
||
section_title_font,
|
||
section_colors[i],
|
||
line_spacing=15,
|
||
only_calc = only_calc
|
||
)
|
||
else:
|
||
|
||
y = draw_multiline_text(
|
||
draw,
|
||
content["text"],
|
||
(card_x +110+num_w, y),
|
||
card_width-120-num_w,
|
||
font,
|
||
font_color,
|
||
line_spacing=15,
|
||
only_calc = only_calc
|
||
)
|
||
y +=20
|
||
elif "image" in content:
|
||
image = content
|
||
image_width= int(card_width)-100
|
||
|
||
resized_img,target_height= draw_image(image["path"],image_width)
|
||
draw.text((card_x + 100, y),
|
||
image["caption"], fill=font_color, font=footer_font)
|
||
img.paste(resized_img, (card_x+50,y+50), resized_img.convert('RGBA'))
|
||
y += target_height+65
|
||
|
||
elif "table" in content:
|
||
y = draw_table(draw,content["data"],
|
||
y,card_x+40,card_width-80,
|
||
keyword_font,content_font,
|
||
font_color,section_colors[i],
|
||
bg_color
|
||
)
|
||
y +=20
|
||
|
||
return y
|
||
|
||
|
||
|
||
"""
|
||
无卡片布局
|
||
"""
|
||
def draw_content_3(img,draw,poster_data,width,y_start,section_colors,title_color,bg_color,font_color):
|
||
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
|
||
section_y_start = y_start
|
||
|
||
#卡片宽度
|
||
card_width = width * 0.9
|
||
card_x = int((width - card_width) / 2)
|
||
|
||
#内容
|
||
y = section_y_start #内容的绝对值是
|
||
for i,slide in enumerate(poster_data["内容列表"]):
|
||
|
||
# 计算位置,卡片的位置
|
||
y_pos = y
|
||
|
||
#y从 主标
|
||
y = y_pos + 80 + 30
|
||
|
||
#内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= True)
|
||
|
||
# 卡片高度,保障一个最小值300的空间
|
||
if y - y_pos < 150:
|
||
y += 100
|
||
|
||
#卡片高度
|
||
card_height = y - y_pos
|
||
|
||
# 绘制主标,标题点缀
|
||
draw.rectangle([(card_x, y_pos), (card_x+10, y_pos+70)],
|
||
fill=section_colors[i])
|
||
|
||
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
|
||
draw.text((card_x+20, y_pos+5),
|
||
slide["主标"], fill=title_color, font=section_title_font)
|
||
|
||
#真正绘制内容
|
||
y = y_pos + 80 + 30
|
||
|
||
#文字内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= False)
|
||
|
||
#下一个
|
||
y = y_pos +80+30+card_height+20
|
||
|
||
return y
|
||
|
||
|
||
"""
|
||
无卡片，网格布局
|
||
"""
|
||
def draw_content_4(img,draw,poster_data,width,y_start,section_colors,title_color,bg_color,font_color):
|
||
# 2. 内容区域 -多个内容块,流式布局按各自的内容大小来排
|
||
section_y_start = y_start
|
||
|
||
#卡片宽度
|
||
card_width = width * 0.9
|
||
card_x = int((width - card_width) / 2)
|
||
|
||
#内容
|
||
y = section_y_start #内容的绝对值是
|
||
for i,slide in enumerate(poster_data["内容列表"]):
|
||
|
||
# 计算位置,卡片的位置
|
||
y_pos = y
|
||
|
||
#y从 主标
|
||
y = y_pos + 80 + 30
|
||
|
||
#文字内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= True)
|
||
|
||
# 卡片高度,保障一个最小值300的空间
|
||
if y - y_pos < 150:
|
||
y += 100
|
||
|
||
#卡片高度
|
||
card_height = y - y_pos
|
||
|
||
#主标题
|
||
num_w, num_h = get_text_size(draw,slide["主标"], section_title_font)
|
||
draw.text((card_x+20, y_pos+5),
|
||
slide["主标"], fill=section_colors[i], font=section_title_font)
|
||
|
||
# 绘制主标,标题下方点缀
|
||
draw.rectangle([(card_x+20, y_pos+30+num_h), (card_x+20+num_w, y_pos+30+num_h+10)],
|
||
fill=section_colors[i])
|
||
|
||
#真正绘制内容
|
||
y = y_pos + 80 + 30
|
||
|
||
#文字内容
|
||
if slide["content"]:
|
||
y = draw_content_list(img,draw,slide,card_x,card_width,bg_color,font_color,section_colors,i,y,only_calc= False)
|
||
|
||
#下一个
|
||
y = y_pos +80+30+card_height+20
|
||
|
||
return y
|
||
|
||
#总结的card
|
||
def draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color,style="round"):
|
||
card_width = int(width*0.90)
|
||
card_x = int(width*0.05)
|
||
|
||
container_height = draw_multiline_text(
|
||
draw,
|
||
poster_data["结语"],
|
||
(card_x+40, y),
|
||
card_width-40,
|
||
end_font,
|
||
title_color,
|
||
line_spacing=15,
|
||
only_calc=True
|
||
)-y+20
|
||
|
||
container = Image.new('RGBA', (card_width, container_height), (0, 0, 0, 0))
|
||
container_draw = ImageDraw.Draw(container)
|
||
if style=="round":
|
||
container_draw.rounded_rectangle([(0, 0), (card_width, container_height)],
|
||
radius=20, fill=card_bg_color)
|
||
elif style=="rect":
|
||
container_draw.rectangle([(0, 0), (card_width, container_height)],
|
||
fill=card_bg_color)
|
||
|
||
# 粘贴容器
|
||
img.paste(container, (card_x, y), container)
|
||
|
||
y +=10
|
||
y = draw_multiline_text(
|
||
draw,
|
||
poster_data["结语"],
|
||
(card_x+40, y),
|
||
card_width-40,
|
||
end_font,
|
||
subtitle_color,
|
||
line_spacing=15
|
||
)
|
||
|
||
return y
|
||
|
||
|
||
#尾部样式1： 渐变圆点
|
||
def draw_footer_1(img,draw,poster_data,width,y,title_color,subtitle_color):
|
||
# 装饰元素 - 渐变圆点
|
||
footer_dots_y = y
|
||
|
||
#手机一整屏
|
||
if footer_dots_y <1980:#内容少的直接到底部
|
||
footer_dots_y = 1980
|
||
|
||
#13个点，一个点大，一个点小
|
||
for i in range(13):
|
||
x = width * (0.1 + 0.066 * i)
|
||
size = 10 + i % 4
|
||
color = subtitle_color
|
||
draw.ellipse([(x, footer_dots_y), (x+size, footer_dots_y+size)],
|
||
fill=color)
|
||
|
||
#装饰元素 -线
|
||
footer_line_y = footer_dots_y+20
|
||
draw.line([(width*0.1, footer_line_y), (width*0.9, footer_line_y)],
|
||
fill=title_color, width=2)
|
||
|
||
#海报尾部文字
|
||
y = footer_line_y+30
|
||
for i,footer in enumerate(poster_data["报尾"]):
|
||
if len(footer)<30:
|
||
#单行，居中对齐
|
||
footer = footer.replace("."," · ")
|
||
footer_w, footer_h = get_text_size(draw,footer, footer_font)
|
||
draw.text(((width - footer_w) // 2, y),
|
||
footer, fill=subtitle_color, font=footer_font)
|
||
y += footer_h+10
|
||
else:
|
||
#多行，居左对齐
|
||
y = draw_multiline_text(
|
||
draw,
|
||
footer,
|
||
(40, y),
|
||
width-40,
|
||
footer_font,
|
||
subtitle_color,
|
||
line_spacing=15
|
||
)
|
||
|
||
|
||
#海报尾部图片
|
||
image_width = width //2 #海报的一半
|
||
y = y+30
|
||
|
||
for i,image in enumerate(poster_data["报尾图片"],1):
|
||
# 调整图像大小
|
||
resized_img,target_height =draw_image(image["path"],image_width)
|
||
|
||
#图片文字
|
||
if image["caption"] and image["caption"][-3:] not in ["png","jpg","peg"]:
|
||
draw.text(((width-image_width)//2+10, y+i*10),
|
||
image["caption"], fill=subtitle_color, font=footer_font)
|
||
y += i*10+60
|
||
|
||
#图片居中
|
||
img.paste(resized_img, ((width-image_width)//2,y), resized_img.convert('RGBA'))
|
||
y += target_height+10
|
||
|
||
|
||
return y+60
|
||
|
||
#手机适合的宣传图
|
||
def create_modern_mobile_poster(poster_data,poster_style={}):
|
||
# 海报尺寸 (手机竖屏9:16比例)
|
||
width, height = 1080, 20000
|
||
|
||
accent_color = (100, 180, 180) # 青蓝色点缀
|
||
|
||
# 主色调：深蓝背景 + 金色点缀
|
||
bg_color = (15, 30, 64) # 深蓝背景
|
||
title_color = (255, 215, 0) # 金色标题
|
||
subtitle_color = (220, 220, 240) #亚白
|
||
card_bg_color = (30, 50, 80, 220) #浅蓝
|
||
section_colors = [
|
||
(255,77,90), #粉红
|
||
(70, 130, 180), # 蓝色
|
||
(100, 180, 160), # 青色
|
||
(180, 120, 70), # 橙色
|
||
(184, 66, 55), # 砖红
|
||
]
|
||
font_color = (240, 240, 240) #接近白色
|
||
|
||
if poster_style:
|
||
try:
|
||
bg_color = hex_to_rgb(poster_style["bg_color"])
|
||
title_color = hex_to_rgb(poster_style["title_color"])
|
||
subtitle_color = hex_to_rgb(poster_style["subtitle_color"])
|
||
card_bg_color = hex_to_rgb(poster_style["card_bg_color"])
|
||
section_colors = [ hex_to_rgb(color) for color in poster_style["section_colors"]]*5
|
||
font_color = hex_to_rgb(poster_style["font_color"])
|
||
except:
|
||
pass
|
||
|
||
# 创建画布
|
||
img = Image.new('RGBA', (width, height),bg_color)
|
||
draw = ImageDraw.Draw(img)
|
||
|
||
if poster_style["style"]==4:
|
||
#网格背景
|
||
size=50
|
||
bg_color1 = get_similar_hsv_color(bg_color)
|
||
#横线
|
||
for i in range(0,height//size):
|
||
y = size//2 + i * size
|
||
draw.line(
|
||
[(0, y), (width, y)],
|
||
#fill=(int(0.299*subtitle_color[0]),int(0.587*subtitle_color[1]),int(0.114*subtitle_color[2])),
|
||
fill = bg_color1,
|
||
width=1
|
||
)
|
||
#竖线
|
||
for i in range(0, width//size):
|
||
x = size//2 + i * size
|
||
draw.line(
|
||
[(x, 0), (x, height)],
|
||
#fill=(subtitle_color[0],subtitle_color[1],subtitle_color[2],0),
|
||
fill = bg_color1,
|
||
width=1
|
||
)
|
||
|
||
|
||
# 1. 绘制头部
|
||
if poster_style["header_style"]==2:
|
||
y = draw_header_2(img,draw,poster_data,width,title_color,subtitle_color)
|
||
elif poster_style["header_style"]==3:
|
||
y = draw_header_3(img,draw,poster_data,width,title_color,subtitle_color)
|
||
else:
|
||
y = draw_header_1(img,draw,poster_data,width,title_color,subtitle_color)
|
||
|
||
|
||
|
||
# 2. 绘制中间内容
|
||
if poster_style["style"] ==1: #多彩卡片
|
||
# 2. 绘制内容
|
||
y +=60
|
||
y = draw_content_card_1(img,draw,poster_data,width,y,section_colors,card_bg_color,font_color)
|
||
|
||
# 3.结语
|
||
y += 50
|
||
if poster_data["结语"]:
|
||
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color)
|
||
|
||
elif poster_style["style"] ==2: #标题卡片
|
||
# 2.结语
|
||
y += 40
|
||
if poster_data["结语"]:
|
||
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color)
|
||
|
||
# 3. 绘制内容
|
||
y +=50
|
||
y = draw_content_card_2(img,draw,poster_data,width,y,section_colors,card_bg_color,font_color)
|
||
|
||
elif poster_style["style"] ==3: #无卡片
|
||
# 2.结语
|
||
if poster_data["结语"]:
|
||
y += 40
|
||
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color,"rect")
|
||
|
||
# 3. 绘制内容
|
||
y +=50
|
||
y = draw_content_3(img,draw,poster_data,width,y,section_colors,title_color,bg_color,font_color)
|
||
|
||
elif poster_style["style"] ==4: # 网格背景
|
||
# 2.结语
|
||
if poster_data["结语"]:
|
||
y += 40
|
||
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color,"rect")
|
||
|
||
# 3. 绘制内容
|
||
y +=50
|
||
y = draw_content_4(img,draw,poster_data,width,y,section_colors,title_color,bg_color,font_color)
|
||
|
||
elif poster_style["style"] ==5: #彩色标签卡片
|
||
# 2.绘制内容
|
||
y += 40
|
||
y = draw_content_card_5(img,draw,poster_data,width,y,section_colors,card_bg_color,font_color)
|
||
|
||
# 3. 结语
|
||
y +=50
|
||
if poster_data["结语"]:
|
||
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color)
|
||
elif poster_style["style"] ==6: #小圆角卡片
|
||
# 2. 结语
|
||
if poster_data["结语"]:
|
||
y += 40
|
||
y = draw_submit_card(img,draw,poster_data,width,y,title_color,card_bg_color,subtitle_color)
|
||
|
||
# 3. 绘制内容
|
||
y +=50
|
||
y = draw_content_card_6(img,draw,poster_data,width,y,section_colors,card_bg_color,font_color)
|
||
|
||
|
||
|
||
# 4. 底部装饰、版权信息
|
||
y +=60
|
||
if poster_data["报尾"]:
|
||
y = draw_footer_1(img,draw,poster_data,width,y,title_color,subtitle_color)
|
||
|
||
#裁切图片
|
||
if y < height:
|
||
new_img = Image.new('RGB', (width, y), bg_color)
|
||
new_img.paste(img, (0, 0))
|
||
return new_img
|
||
else:
|
||
return img
|
||
|
||
|
||
#入口函数，创建海报
|
||
def create_poster(content,path):
|
||
#百科内容转化为结构
|
||
poster_data = parse_html_to_poster(content)
|
||
|
||
from fun_calls import gen_poster_color
|
||
#生成主题颜色
|
||
|
||
try:
|
||
poster_style = gen_poster_color(poster_data)
|
||
except Exception as e:
|
||
print("主题生成出错",e)
|
||
poster_style = { "style": 6, "header_style": 3, "bg_color": "#F0F0F0", "title_color": "#003366", "subtitle_color": "#FFA500", "font_color": "#FFFFFF", "card_bg_color": "#D3D3D3", "section_colors": ["#FF0000", "#00FF00", "#0000FF", "#FFFF00", "#FF00FF"] }
|
||
|
||
poster_style["style"] = 6
|
||
poster_style["header_style"] = 3
|
||
|
||
# #测试使用
|
||
# poster_style={
|
||
# "style": 6,
|
||
# "header_style":3,
|
||
# "bg_color": "#FFFFFF",
|
||
# "title_color": "#333333",
|
||
# "subtitle_color": "#003366",
|
||
# "font_color": "#000000",
|
||
# "card_bg_color": "#F0F8FF",
|
||
# "section_colors": ["#FF6B6B", "#4ECDC4", "#45B7D1", "#96CEB4", "#FFEEAD"],
|
||
|
||
# # "bg_color": "#000080",
|
||
# # "title_color": "#FFFFFF",
|
||
# # "subtitle_color": "#CCCCCC",
|
||
# # "font_color": "#333333",
|
||
# # "card_bg_color": "#ADD8E6",
|
||
# # "section_colors": ["#FFFFFF", "#ADD8E6", "#87CEEB", "#B0C4DE", "#D3D3D3"]
|
||
# }
|
||
|
||
|
||
# 生成并保存海报
|
||
poster = create_modern_mobile_poster(poster_data,poster_style)
|
||
poster.save(path)
|
||
print('海报已生成: modern_poster.png')
|
||
|
||
#入口函数，创建海报
|
||
def create_poster_style(content,path,poster_style):
|
||
#百科内容转化为结构
|
||
poster_data = parse_html_to_poster(content)
|
||
|
||
# 生成并保存海报
|
||
poster = create_modern_mobile_poster(poster_data,poster_style)
|
||
poster.save(path)
|
||
print('海报已生成: modern_poster.png') |