mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 18:32:48 +00:00
57b15a8a7f
--story=1016154 --user=刘瑞斌 【知识库】-支持上传表格类型文档（Excel/CSV）按行分段 https://www.tapd.cn/57709429/s/1567910
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
# coding=utf-8
|
|
import io
|
|
import logging
|
|
|
|
from openpyxl import load_workbook
|
|
|
|
from common.handle.base_parse_table_handle import BaseParseTableHandle
|
|
|
|
max_kb = logging.getLogger("max_kb")
|
|
|
|
|
|
class ExcelSplitHandle(BaseParseTableHandle):
|
|
def support(self, file, get_buffer):
|
|
file_name: str = file.name.lower()
|
|
if file_name.endswith('.xls') or file_name.endswith('.xlsx'):
|
|
return True
|
|
return False
|
|
|
|
def handle(self, file, get_buffer):
|
|
buffer = get_buffer(file)
|
|
try:
|
|
wb = load_workbook(io.BytesIO(buffer))
|
|
result = []
|
|
for sheetname in wb.sheetnames:
|
|
paragraphs = []
|
|
ws = wb[sheetname]
|
|
rows = list(ws.rows)
|
|
if not rows: continue
|
|
ti = list(rows[0])
|
|
for r in list(rows[1:]):
|
|
title = []
|
|
l = []
|
|
for i, c in enumerate(r):
|
|
if not c.value:
|
|
continue
|
|
t = str(ti[i].value) if i < len(ti) else ""
|
|
title.append(t)
|
|
t += (": " if t else "") + str(c.value)
|
|
l.append(t)
|
|
l = "; ".join(l)
|
|
if sheetname.lower().find("sheet") < 0:
|
|
l += " ——" + sheetname
|
|
paragraphs.append({'title': '', 'content': l})
|
|
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
|