|
|
import os
|
|
|
|
|
|
import docx
|
|
|
|
|
|
|
|
|
def read_word_content(docx_path):
|
|
|
res = ""
|
|
|
idx = 0
|
|
|
"""遍历Word文档的每个段落，输出文字或图片标识"""
|
|
|
try:
|
|
|
doc = docx.Document(docx_path)
|
|
|
|
|
|
for paragraph in doc.paragraphs:
|
|
|
has_image = False
|
|
|
# 检查段落中是否有图片
|
|
|
for run in paragraph.runs:
|
|
|
for element in run._element:
|
|
|
if element.tag.endswith('drawing'):
|
|
|
# 找到图片元素
|
|
|
has_image = True
|
|
|
break
|
|
|
if has_image:
|
|
|
break
|
|
|
|
|
|
if has_image:
|
|
|
idx = idx + 1
|
|
|
res = res + "\n" + "【图片" + str(idx) + "】"
|
|
|
elif paragraph.text.strip():
|
|
|
res = res + "\n" + paragraph.text.strip()
|
|
|
return res
|
|
|
except Exception as e:
|
|
|
print(f"处理Word文档时出错: {str(e)}")
|
|
|
|
|
|
|
|
|
def split_into_blocks(text):
|
|
|
"""按行遍历文本，发现'问题X'或'话题X'时开始分割，只移除前缀但保留整行内容"""
|
|
|
blocks = []
|
|
|
current_block = []
|
|
|
in_block = False
|
|
|
|
|
|
for line in text.splitlines():
|
|
|
if line.startswith(('问题', '话题')) and any(c.isdigit() for c in line[:5]):
|
|
|
if in_block:
|
|
|
blocks.append('\n'.join(current_block))
|
|
|
current_block = []
|
|
|
in_block = True
|
|
|
# 循环移除问题和话题前缀后的数字
|
|
|
while line and (line.startswith(('问题', '话题')) or (line and line and line[0].isdigit())):
|
|
|
if line.startswith(('问题', '话题')):
|
|
|
line = line[2:] if len(line) > 2 else line
|
|
|
elif line and line[0].isdigit():
|
|
|
line = line[1:] if len(line) > 1 else line
|
|
|
line = line.strip()
|
|
|
if in_block and line: # 只添加非空行
|
|
|
current_block.append(line)
|
|
|
|
|
|
if current_block:
|
|
|
blocks.append('\n'.join(current_block))
|
|
|
|
|
|
return [(i + 1, block) for i, block in enumerate(blocks)]
|
|
|
|
|
|
|
|
|
# 保留原有的save_to_txt函数
|
|
|
def save_to_txt(content, file_path, mode='w'):
|
|
|
"""将内容保存到文本文件"""
|
|
|
try:
|
|
|
with open(file_path, mode, encoding='utf-8') as f:
|
|
|
f.write(content)
|
|
|
return True
|
|
|
except Exception as e:
|
|
|
print(f"保存文件{file_path}时出错: {str(e)}")
|
|
|
return False
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
word_document_path = "d:\\dsWork\\dsProject\\dsRag\\static\\Test\\带图的WORD文档_MATH_3.docx"
|
|
|
output_dir="d:\\dsWork\\dsProject\\dsRag\\static\\Test\\"
|
|
|
res = read_word_content(word_document_path)
|
|
|
chunks = split_into_blocks(res)
|
|
|
for x in chunks:
|
|
|
print("===段落开始：===")
|
|
|
firstLine = x[1].split("\n")[0].strip()
|
|
|
content = x[1][len(firstLine):].strip()
|
|
|
print("firstLine=" + firstLine)
|
|
|
print("content=" + content)
|
|
|
print("===段落结束：===\n")
|
|
|
|
|
|
saved_count=0
|
|
|
for chunk_num, chunk in chunks:
|
|
|
chunk = chunk.strip() # 确保去除空白字符
|
|
|
output_file = os.path.join(output_dir, f"{chunk_num}.txt")
|
|
|
if save_to_txt(chunk, output_file, mode='w'):
|
|
|
saved_count += 1
|
|
|
|
|
|
print(f"处理完成，共保存{saved_count}个文件到目录: {output_dir}")
|
