114 lines
3.4 KiB
Python
114 lines
3.4 KiB
Python
|
"""
|
|||
|
|
PDF 生成服务 - 基于 WeasyPrint + 系统字体
|
|||
|
|
"""
|
|||
|
|
import markdown
|
|||
|
|
from weasyprint import HTML, CSS
|
|||
|
|
from weasyprint.text.fonts import FontConfiguration
|
|||
|
|
import io
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PDFService:
|
|||
|
|
def __init__(self):
|
|||
|
|
self.font_config = FontConfiguration()
|
|||
|
|
|
|||
|
|
def get_css(self):
|
|||
|
|
return """
|
|||
|
|
@page {
|
|||
|
|
margin: 2cm;
|
|||
|
|
@bottom-right {
|
|||
|
|
content: counter(page);
|
|||
|
|
font-size: 9pt;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
/* 全局：macOS 用 Hiragino Sans GB / Heiti SC，Docker 用 WenQuanYi Micro Hei */
|
|||
|
|
* {
|
|||
|
|
font-family: "Hiragino Sans GB", "Heiti SC", "WenQuanYi Micro Hei",
|
|||
|
|
"Arial Unicode MS", sans-serif !important;
|
|||
|
|
}
|
|||
|
|
body {
|
|||
|
|
font-size: 11pt;
|
|||
|
|
line-height: 1.6;
|
|||
|
|
color: #333;
|
|||
|
|
background-color: #fff;
|
|||
|
|
}
|
|||
|
|
h1, h2, h3, h4, h5, h6 {
|
|||
|
|
border-bottom: 1px solid #eee;
|
|||
|
|
padding-bottom: 0.3em;
|
|||
|
|
margin-top: 1.5em;
|
|||
|
|
margin-bottom: 1em;
|
|||
|
|
font-weight: bold;
|
|||
|
|
}
|
|||
|
|
/* 代码块：等宽字体优先，中文回退到对应平台的 CJK 字体 */
|
|||
|
|
pre, code, pre *, code * {
|
|||
|
|
font-family: "Menlo", "DejaVu Sans Mono", "Courier New",
|
|||
|
|
"Hiragino Sans GB", "Heiti SC", "WenQuanYi Micro Hei", monospace !important;
|
|||
|
|
background-color: transparent;
|
|||
|
|
}
|
|||
|
|
pre {
|
|||
|
|
background-color: #f6f8fa !important;
|
|||
|
|
padding: 16px;
|
|||
|
|
border-radius: 6px;
|
|||
|
|
white-space: pre-wrap;
|
|||
|
|
word-break: break-all;
|
|||
|
|
display: block;
|
|||
|
|
}
|
|||
|
|
code {
|
|||
|
|
background-color: rgba(175, 184, 193, 0.2);
|
|||
|
|
padding: 0.2em 0.4em;
|
|||
|
|
border-radius: 6px;
|
|||
|
|
}
|
|||
|
|
table {
|
|||
|
|
border-collapse: collapse;
|
|||
|
|
width: 100%;
|
|||
|
|
margin-bottom: 1em;
|
|||
|
|
}
|
|||
|
|
th, td {
|
|||
|
|
border: 1px solid #dfe2e5;
|
|||
|
|
padding: 6px 13px;
|
|||
|
|
}
|
|||
|
|
tr:nth-child(2n) {
|
|||
|
|
background-color: #f6f8fa;
|
|||
|
|
}
|
|||
|
|
img {
|
|||
|
|
max-width: 100%;
|
|||
|
|
}
|
|||
|
|
blockquote {
|
|||
|
|
border-left: 0.25em solid #dfe2e5;
|
|||
|
|
color: #6a737d;
|
|||
|
|
padding: 0 1em;
|
|||
|
|
margin-left: 0;
|
|||
|
|
}
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
async def md_to_pdf(self, md_content: str, title: str = "Document", base_url: str = None) -> io.BytesIO:
|
|||
|
|
"""将 Markdown 转换为 PDF 字节流"""
|
|||
|
|
html_content = markdown.markdown(
|
|||
|
|
md_content,
|
|||
|
|
extensions=['extra', 'codehilite', 'toc', 'tables']
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
full_html = f"""
|
|||
|
|
<!DOCTYPE html>
|
|||
|
|
<html>
|
|||
|
|
<head>
|
|||
|
|
<meta charset="UTF-8">
|
|||
|
|
<title>{title}</title>
|
|||
|
|
</head>
|
|||
|
|
<body class="markdown-body">
|
|||
|
|
{html_content}
|
|||
|
|
</body>
|
|||
|
|
</html>
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
pdf_buffer = io.BytesIO()
|
|||
|
|
css = CSS(string=self.get_css(), font_config=self.font_config)
|
|||
|
|
HTML(string=full_html, base_url=base_url).write_pdf(
|
|||
|
|
pdf_buffer,
|
|||
|
|
stylesheets=[css],
|
|||
|
|
font_config=self.font_config
|
|||
|
|
)
|
|||
|
|
pdf_buffer.seek(0)
|
|||
|
|
return pdf_buffer
|
|||
|
|
|
|||
|
|
pdf_service = PDFService()
|