222 lines
4.7 KiB
Python
222 lines
4.7 KiB
Python
"""
|
||
生成文档，包括word和pdf两种类型
|
||
"""
|
||
|
||
import pypandoc
|
||
import pdfkit
|
||
|
||
from init import gcfg
|
||
from utils import regular_filename
|
||
|
||
#pdf文档的配置选项
|
||
options = {
|
||
'enable-local-file-access': None, # 允许访问本地文件
|
||
'page-size': 'A4',
|
||
'margin-top': '0.5in',
|
||
'margin-right': '0.5in',
|
||
'margin-bottom': '0.5in',
|
||
'margin-left': '0.5in',
|
||
'encoding': "UTF-8",
|
||
'custom-header': [
|
||
('Accept-Encoding', 'gzip')
|
||
],
|
||
'no-outline': None,
|
||
'footer-center': '[page]',
|
||
'footer-font-size': '8',
|
||
'dpi': '300',
|
||
'disable-smart-shrinking': ''
|
||
}
|
||
|
||
header_html="""
|
||
<!DOCTYPE html>
|
||
<html>
|
||
<head>
|
||
<style>
|
||
table {
|
||
border-collapse: collapse;
|
||
}
|
||
td,th {
|
||
border: 1px solid #ccc;
|
||
min-width: 50px;
|
||
height: 20px;
|
||
}
|
||
th {
|
||
background-color: #f1f1f1;
|
||
}
|
||
|
||
blockquote {
|
||
border-left: 8px solid #d0e5f2;
|
||
padding: 10px 10px;
|
||
margin: 10px 0;
|
||
background-color: #f1f1f1;
|
||
}
|
||
|
||
code {
|
||
font-family: monospace;
|
||
background-color: #eee;
|
||
padding: 3px;
|
||
border-radius: 3px;
|
||
}
|
||
pre>code {
|
||
display: block;
|
||
padding: 10px;
|
||
}
|
||
|
||
/* 基础样式 */
|
||
h1, h2, h3, h4, h5 {
|
||
font-weight: bold;
|
||
line-height: 1.2;
|
||
margin-top: 1.5em;
|
||
margin-bottom: 0.5em;
|
||
}
|
||
|
||
/* 特定级别样式 */
|
||
h1 {
|
||
font-size: 2.5em; /* 或者使用百分比如 250% */
|
||
color: #333;
|
||
border-bottom: 1px solid #ddd;
|
||
}
|
||
|
||
h2 {
|
||
font-size: 2.5em;
|
||
color: #444;
|
||
}
|
||
|
||
h3 {
|
||
font-size: 2em;
|
||
color: #000;
|
||
}
|
||
|
||
h4 {
|
||
font-size: 1.75em;
|
||
color: #000;
|
||
}
|
||
|
||
h5 {
|
||
font-size: 1.5em;
|
||
color: #000;
|
||
}
|
||
|
||
ul {
|
||
list-style-type: disc; /* 默认样式为实心圆 */
|
||
margin: 0;
|
||
padding-left: 20px; /* 根据需要调整缩进 */
|
||
}
|
||
|
||
ul li {
|
||
margin-bottom: 0.5em; /* 可选：增加项目之间的间距 */
|
||
}
|
||
|
||
ol {
|
||
list-style-type: decimal; /* 默认样式为数字1, 2, 3... */
|
||
margin: 0;
|
||
padding-left: 20px; /* 根据需要调整缩进 */
|
||
}
|
||
|
||
ol li {
|
||
margin-left: 0.5em;
|
||
margin-bottom: 0.5em; /* 可选：增加项目之间的间距 */
|
||
}
|
||
|
||
body {
|
||
font-size: 1.5em;
|
||
}
|
||
|
||
</style>
|
||
</head>
|
||
"""
|
||
|
||
|
||
html_template="""
|
||
<body>
|
||
{body}
|
||
</body>
|
||
</html>
|
||
"""
|
||
|
||
def gen_word(name,content):
|
||
name = regular_filename(name)
|
||
path =f'{gcfg["fs"]["path"]}/pub/{name}.docx'
|
||
fname=f'pub/{name}.docx'
|
||
content = content.replace('src="/api/',f'src="{gcfg["fs"]["path"]}/')
|
||
html_content = header_html+html_template.format(body=content)
|
||
output = pypandoc.convert_text(html_content, 'docx', format='html', outputfile=path)
|
||
return fname
|
||
|
||
|
||
def gen_pdf(name,content):
|
||
name = regular_filename(name)
|
||
path =f'{gcfg["fs"]["path"]}/pub/{name}.pdf'
|
||
fname=f'pub/{name}.pdf'
|
||
content = content.replace('src="/api/',f'src="files://{gcfg["fs"]["path"]}/')
|
||
html_content = header_html+html_template.format(body=content)
|
||
pdfkit.from_string(html_content, path,options=options)
|
||
return fname
|
||
|
||
|
||
def gen_ppt(name,content):
|
||
name = regular_filename(name)
|
||
path =f'{gcfg["fs"]["path"]}/pub/{name}.pptx'
|
||
fname=f'pub/{name}.pptx'
|
||
from make_ppt2 import create_unified_ppt
|
||
from gen_img_servie import create_ppt_images
|
||
color = create_ppt_images()
|
||
style={"title_style":2,"main_color":color,"style":0}
|
||
create_unified_ppt(content,path,style)
|
||
return fname
|
||
|
||
#颜色转换
|
||
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 gen_ppt_by_ai(name,content,style):
|
||
name = regular_filename(name)
|
||
path =f'{gcfg["fs"]["path"]}/pub/{name}.pptx'
|
||
fname=f'pub/{name}.pptx'
|
||
|
||
from make_ppt2 import create_unified_ppt
|
||
from gen_img_servie import create_ppt_images
|
||
|
||
style["main_color"] = hex_to_rgb(style["main_color"])
|
||
create_ppt_images(style["main_color"]) #宝石蓝
|
||
|
||
create_unified_ppt(content,path,style)
|
||
return fname
|
||
|
||
#生成海报
|
||
def gen_poster(name,content):
|
||
name = regular_filename(name)
|
||
path =f'{gcfg["fs"]["path"]}/pub/{name}.png'
|
||
fname=f'pub/{name}.png'
|
||
|
||
from make_poster import create_poster
|
||
create_poster(content,path)
|
||
return fname
|
||
|
||
#生成海报,根据样式生成
|
||
def gen_poster_style(name,content,style):
|
||
name = regular_filename(name)
|
||
path =f'{gcfg["fs"]["path"]}/pub/{name}.png'
|
||
fname=f'pub/{name}.png'
|
||
|
||
from make_poster import create_poster_style
|
||
create_poster_style(content,path,style)
|
||
return fname
|
||
|
||
def test_pdf():
|
||
html_content = "<h1>Hello, World!</h1><p>这是生成文档的内容，如果出现乱码，请检查字体是否安装！</p>"
|
||
|
||
pdfkit.from_string(html_content, 'output.pdf',options=options)
|
||
|
||
print("文件已成功转换并保存至 output.pdf")
|
||
|
||
|
||
|
||
if __name__=="__main__":
|
||
test_pdf() |