From 3708baa96db4938fc48dcb9b8d255a4d4b3e14fa Mon Sep 17 00:00:00 2001
From: 13315423919 <13315423919@qq.com>
Date: Wed, 19 Nov 2025 19:42:59 +0800
Subject: [PATCH] Add File
---
main/f_build.py | 222 ++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 222 insertions(+)
create mode 100644 main/f_build.py
diff --git a/main/f_build.py b/main/f_build.py
new file mode 100644
index 0000000..9e36732
--- /dev/null
+++ b/main/f_build.py
@@ -0,0 +1,222 @@
+"""
+生成文档，包括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="""
+
+
+
+
+
+"""
+
+
+html_template="""
+
+{body}
+
+
+"""
+
+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 = "Hello, World!
这是生成文档的内容，如果出现乱码，请检查字体是否安装！
"
+
+ pdfkit.from_string(html_content, 'output.pdf',options=options)
+
+ print("文件已成功转换并保存至 output.pdf")
+
+
+
+if __name__=="__main__":
+ test_pdf()
\ No newline at end of file