xiaji/FableFlow
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(03_Story): 添加章节文件重命名工具并改进PDF合并排序逻辑

Browse Source 
新增重命名脚本支持按规则批量修改章节文件名
改进merge_md_to_pdf.py中的文件排序逻辑，支持从文件名提取数字排序
添加工作区配置文件和更新后的PDF文档
This commit is contained in:
xiaji
2026-01-12 09:22:44 +08:00
parent c2eb39ff1e
commit 7bd57e7cc2
5 changed files with 115 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
03_Story/merge_md_to_pdf.py
View File

@@ -1,22 +1,37 @@
"""
将当前目录下的所有md文件合并成一个pdf文件
将当前目录下的所有md文件按文件名中的数字排序（从第一章开始），合并成一个pdf文件
"""
import os
import re # 新增：导入正则模块
import markdown
from weasyprint import HTML, CSS
def extract_chapter_number(filename):
"""
正则提取文件名中的数字（核心函数）
:param filename: 文件名（如：第1章.md、10.md、章节2.md）
:return: 提取到的数字（int），无数字返回9999（排最后）
"""
# 正则匹配所有连续数字（\d+ 匹配1个及以上数字）
match = re.search(r'\d+', filename)
if match:
return int(match.group()) # 转成整数，保证排序正确（10>2）
else:
return 9999 # 无数字的文件排最后
# 获取当前目录
current_dir = os.path.dirname(os.path.abspath(__file__))
# 获取所有md文件并按文件名排序
md_files = sorted([f for f in os.listdir(current_dir) if f.endswith('.md')])
# 获取所有md文件 + 按提取的数字排序（核心修正点）
md_files = [f for f in os.listdir(current_dir) if f.endswith('.md')]
md_files = sorted(md_files, key=lambda x: extract_chapter_number(x)) # 按数字排序
print(f"找到 {len(md_files)} 个md文件:")
print(f"找到 {len(md_files)} 个md文件（按章节数字排序）:")
for f in md_files:
print(f" - {f}")
# 合并所有md文件内容
# 合并所有md文件内容（原有逻辑不变）
combined_html = """<!DOCTYPE html>
<html lang=\"zh-CN\">
<head>
@@ -69,7 +84,7 @@ for md_file in md_files:
with open(file_path, 'r', encoding='utf-8') as f:
content = f.read()
# 转换为html
# 转换为html（保留表格、代码块格式）
md = markdown.Markdown(extensions=['tables', 'fenced_code'])
html_content = md.convert(content)
@@ -84,4 +99,4 @@ combined_html += '</body></html>'
output_pdf = os.path.join(current_dir, '小说合集.pdf')
HTML(string=combined_html).write_pdf(output_pdf)
print(f"\nPDF已生成: {output_pdf}")
print(f"\nPDF已生成: {output_pdf}")