51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
|
# chapter_directory_parser.py
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
import re
|
|||
|
|
|
|||
|
|
def get_chapter_info_from_directory(novel_directory_content: str, chapter_number: int):
|
|||
|
|
"""
|
|||
|
|
从给定的 novel_directory_content 文本中，解析 “第X章” 行，并提取本章的标题和可能的简述。
|
|||
|
|
返回一个 dict: {
|
|||
|
|
"chapter_title": <字符串>,
|
|||
|
|
"chapter_brief": <字符串> (若没有则为空)
|
|||
|
|
}
|
|||
|
|
注意：目录文本示例格式:
|
|||
|
|
第1章 ：潮起
|
|||
|
|
第2章 ：阴影浮现 - 主要角色冲突爆发
|
|||
|
|
...
|
|||
|
|
也可能没有简述，只有一个简单标题。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# 将文本逐行拆分
|
|||
|
|
lines = novel_directory_content.splitlines()
|
|||
|
|
|
|||
|
|
# 章节匹配：形如 “第5章 ：xxx” or “第5章: xxx” or “第5章 xxx”
|
|||
|
|
pattern = re.compile(r'^第\s*(\d+)\s*章\s*[:：]?\s*(.*)$')
|
|||
|
|
|
|||
|
|
for line in lines:
|
|||
|
|
match = pattern.match(line.strip())
|
|||
|
|
if match:
|
|||
|
|
chap_num = int(match.group(1))
|
|||
|
|
if chap_num == chapter_number:
|
|||
|
|
# group(2) 可能是标题及简述的混合
|
|||
|
|
full_title = match.group(2).strip()
|
|||
|
|
# 这里假设用 '-' 进一步区分“标题 - 简述”，也可能用户没写“ - ”
|
|||
|
|
if ' - ' in full_title:
|
|||
|
|
# 根据你的目录格式自由处理
|
|||
|
|
parts = full_title.split(' - ', 1)
|
|||
|
|
return {
|
|||
|
|
"chapter_title": parts[0].strip(),
|
|||
|
|
"chapter_brief": parts[1].strip()
|
|||
|
|
}
|
|||
|
|
else:
|
|||
|
|
return {
|
|||
|
|
"chapter_title": full_title,
|
|||
|
|
"chapter_brief": ""
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 如果没有匹配到，返回默认
|
|||
|
|
return {
|
|||
|
|
"chapter_title": f"第{chapter_number}章",
|
|||
|
|
"chapter_brief": ""
|
|||
|
|
}
|