-
Notifications
You must be signed in to change notification settings - Fork 36
/
Copy pathExtract_Word.py
202 lines (177 loc) · 8.35 KB
/
Extract_Word.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
from docx import Document
import io
import os
from docx.shared import Cm
# 项目要用正文下加项目格式，除了章节，不支持其他样式
# 设置最大宽度为14厘米
MAX_WIDTH_CM = 14.0
# Define initial version number
version = [0, 0, 0]
# Get file name function
def get_file_name(version, heading_text):
while version[-1] == 0:
version.pop()
heading_text = ''.join([c for c in heading_text if c.isalnum() or c in (' ', '-', '_')]).strip()
return '.'.join(map(str, version)) + '- ' + heading_text
# Update version number function
def update_version(level):
global version
while len(version) < 3:
version.append(0)
if level == '1':
version[0] += 1
version[1], version[2] = 0, 0
elif level == '2':
version[1] += 1
version[2] = 0
elif level == '3':
version[2] += 1
# Save content as new Word document
def save_content_to_new_doc(content, version, heading_text):
from docx.oxml import OxmlElement
from docx.oxml.ns import qn
file_name = get_file_name(version, heading_text) + ".docx"
new_doc = Document()
for item in content:
if isinstance(item, str):
paragraph = new_doc.add_paragraph()
run = paragraph.add_run(item)
run.font.name = '宋体'
run._element.rPr.rFonts.set(qn('w:eastAsia'), '宋体')
elif isinstance(item, tuple) and item[0] == 'table':
table_data = item[1]
table = new_doc.add_table(rows=len(table_data), cols=len(table_data[0]))
for i, row in enumerate(table_data):
for j, cell_text in enumerate(row):
table.cell(i, j).text = cell_text
for paragraph in table.cell(i, j).paragraphs:
for run in paragraph.runs:
run.font.name = '宋体'
run._element.rPr.rFonts.set(qn('w:eastAsia'), '宋体')
tbl = table._element
tbl_pr = tbl.tblPr
tbl_borders = OxmlElement('w:tblBorders')
for border_name in ['top', 'left', 'bottom', 'right', 'insideH', 'insideV']:
border = OxmlElement(f'w:{border_name}')
border.set(qn('w:val'), 'single')
border.set(qn('w:sz'), '4')
border.set(qn('w:space'), '0')
border.set(qn('w:color'), '000000')
tbl_borders.append(border)
tbl_pr.append(tbl_borders)
elif isinstance(item, tuple) and item[0] == 'image':
image_stream = item[1]
image_name = item[2]
width_cm = item[3]
height_cm = item[4]
if width_cm > MAX_WIDTH_CM:
scale_factor = MAX_WIDTH_CM / width_cm
width_cm = MAX_WIDTH_CM
height_cm = height_cm * scale_factor
new_doc.add_paragraph().add_run().add_picture(image_stream, width=Cm(width_cm), height=Cm(height_cm))
elif isinstance(item, tuple) and item[0] == 'list':
# Add list paragraph to the new document while retaining its list format
paragraph = new_doc.add_paragraph(item[1], style='List Paragraph')
new_doc.save(file_name)
# Function to extract images from runs
def get_image_from_run(run):
drawing_elements = run._element.xpath('.//a:blip')
if drawing_elements:
embed = drawing_elements[0].get('{http://schemas.openxmlformats.org/officeDocument/2006/relationships}embed')
image_part = run.part.related_parts[embed]
image_data = image_part.blob
drawing_element = run._element.xpath('.//wp:extent')
if drawing_element:
cx = int(drawing_element[0].get('cx'))
cy = int(drawing_element[0].get('cy'))
width_cm = cx / 914400 * 2.54
height_cm = cy / 914400 * 2.54
else:
width_cm, height_cm = None, None
image_stream = io.BytesIO(image_data)
image_stream.name = os.path.basename(image_part.partname)
return image_stream, image_stream.name, width_cm, height_cm
return None
# Function to iterate over document elements in order
def iter_block_items(parent):
from docx.oxml.text.paragraph import CT_P
from docx.oxml.table import CT_Tbl
from docx.table import Table
from docx.text.paragraph import Paragraph
from docx.document import Document
parent_elm = parent.element.body if isinstance(parent, Document) else parent._element
for child in parent_elm.iterchildren():
if isinstance(child, CT_P):
yield Paragraph(child, parent)
elif isinstance(child, CT_Tbl):
yield Table(child, parent)
# Extract numbered headings and content from Word document
def extract_numbered_headings_and_content(docx_path):
from docx import Document
from docx.text.paragraph import Paragraph
from docx.table import Table
from docx.oxml.ns import qn
doc = Document(docx_path)
content_between_headings = []
current_level = None
current_heading_text = None
for block in iter_block_items(doc):
if isinstance(block, Paragraph):
para = block
if para.style.name.startswith('Heading'):
if current_level is not None:
save_content_to_new_doc(content_between_headings, version, current_heading_text)
content_between_headings = []
heading_level = para.style.name.split(' ')[-1]
current_level = heading_level
update_version(current_level)
current_heading_text = para.text
#不加入标题进入正文
# content_between_headings.append(para.text)
else:
paragraph_text = ""
for run in para.runs:
image_data = get_image_from_run(run)
if image_data:
image_stream, image_name, width_cm, height_cm = image_data
content_between_headings.append(('image', image_stream, image_name, width_cm, height_cm))
else:
text = run.text.strip()
if text:
paragraph_text += text
# Check if paragraph is a list (ordered or unordered) and copy it with its format
if para.style.name in ['List Paragraph'] or para._element.xpath('.//w:numPr'):
# Set the paragraph text and ensure all runs use the '宋体' font
list_text = para.text
for run in para.runs:
run.font.name = '宋体'
run._element.rPr.rFonts.set(qn('w:eastAsia'), '宋体') # 设置东亚字体为宋体
content_between_headings.append(('list', list_text))
elif paragraph_text:
# Ensure non-list paragraphs also use '宋体'
for run in para.runs:
run.font.name = '宋体'
run._element.rPr.rFonts.set(qn('w:eastAsia'), '宋体')
content_between_headings.append(paragraph_text)
elif isinstance(block, Table):
table = block
table_data = []
for row in table.rows:
row_data = []
for cell in row.cells:
cell_text = ''
for cell_paragraph in cell.paragraphs:
for cell_run in cell_paragraph.runs:
image_stream = get_image_from_run(cell_run)
if image_stream:
pass
else:
cell_text += cell_run.text.strip()
row_data.append(cell_text)
table_data.append(row_data)
content_between_headings.append(('table', table_data))
if content_between_headings:
save_content_to_new_doc(content_between_headings, version, current_heading_text)
# Example: Extract Word document content including images
docx_file = 'Template.docx' # Replace with your document path
extract_numbered_headings_and_content(docx_file)