mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
107 lines
3.8 KiB
Python
107 lines
3.8 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
|
||
|
||
|
||
def get_content(self, file):
|
||
try:
|
||
# 加载 Excel 文件
|
||
workbook = load_workbook(file)
|
||
md_tables = ''
|
||
# 如果未指定 sheet_name，则使用第一个工作表
|
||
for sheetname in workbook.sheetnames:
|
||
sheet = workbook[sheetname] if sheetname else workbook.active
|
||
|
||
# 获取工作表的所有行
|
||
rows = list(sheet.iter_rows(values_only=True))
|
||
if not rows:
|
||
continue
|
||
|
||
# 提取表头和内容
|
||
headers = rows[0]
|
||
data = rows[1:]
|
||
|
||
# 构建 Markdown 表格
|
||
md_table = '| ' + ' | '.join(headers) + ' |\n'
|
||
md_table += '| ' + ' | '.join(['---'] * len(headers)) + ' |\n'
|
||
for row in data:
|
||
md_table += '| ' + ' | '.join(
|
||
[str(cell).replace('\n', '<br>') if cell is not None else '' for cell in row]) + ' |\n'
|
||
|
||
md_tables += md_table + '\n\n'
|
||
return md_tables
|
||
except Exception as e:
|
||
max_kb.error(f'excel split handle error: {e}')
|
||
return f'error: {e}'
|