mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 18:32:48 +00:00
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
# coding=utf-8
|
|
import io
|
|
import logging
|
|
|
|
from openpyxl import load_workbook
|
|
|
|
from common.handle.base_parse_table_handle import BaseParseTableHandle
|
|
from common.handle.impl.tools import xlsx_embed_cells_images
|
|
|
|
max_kb = logging.getLogger("max_kb")
|
|
|
|
|
|
class XlsxSplitHandle(BaseParseTableHandle):
|
|
def support(self, file, get_buffer):
|
|
file_name: str = file.name.lower()
|
|
if file_name.endswith('.xlsx'):
|
|
return True
|
|
return False
|
|
|
|
def fill_merged_cells(self, sheet, image_dict):
|
|
data = []
|
|
|
|
# 获取第一行作为标题行
|
|
headers = [cell.value for cell in sheet[1]]
|
|
|
|
# 从第二行开始遍历每一行
|
|
for row in sheet.iter_rows(min_row=2, values_only=False):
|
|
row_data = {}
|
|
for col_idx, cell in enumerate(row):
|
|
cell_value = cell.value
|
|
|
|
# 如果单元格为空，并且该单元格在合并单元格内，获取合并单元格的值
|
|
if cell_value is None:
|
|
for merged_range in sheet.merged_cells.ranges:
|
|
if cell.coordinate in merged_range:
|
|
cell_value = sheet[merged_range.min_row][merged_range.min_col - 1].value
|
|
break
|
|
|
|
image = image_dict.get(cell_value, None)
|
|
if image is not None:
|
|
cell_value = f''
|
|
|
|
# 使用标题作为键，单元格的值作为值存入字典
|
|
row_data[headers[col_idx]] = cell_value
|
|
data.append(row_data)
|
|
|
|
return data
|
|
|
|
def handle(self, file, get_buffer, save_image):
|
|
buffer = get_buffer(file)
|
|
try:
|
|
wb = load_workbook(io.BytesIO(buffer))
|
|
try:
|
|
image_dict: dict = xlsx_embed_cells_images(io.BytesIO(buffer))
|
|
save_image([item for item in image_dict.values()])
|
|
except Exception as e:
|
|
image_dict = {}
|
|
result = []
|
|
for sheetname in wb.sheetnames:
|
|
paragraphs = []
|
|
ws = wb[sheetname]
|
|
data = self.fill_merged_cells(ws, image_dict)
|
|
|
|
for row in data:
|
|
row_output = "; ".join([f"{key}: {value}" for key, value in row.items()])
|
|
# print(row_output)
|
|
paragraphs.append({'title': '', 'content': row_output})
|
|
|
|
result.append({'name': sheetname, 'paragraphs': paragraphs})
|
|
|
|
except BaseException as e:
|
|
max_kb.error(f'excel split handle error: {e}')
|
|
return [{'name': file.name, 'paragraphs': []}]
|
|
return result
|