Преглед изворни кода

To avoid assertion while no rows in excel (#197)

### What problem does this PR solve?

_Briefly describe what this PR aims to solve. Include background context
that will help reviewers understand the purpose of the PR._

Issue link:#[[Link the issue
here](https://github.com/infiniflow/ragflow/issues/196)]

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
- [ ] New Feature (non-breaking change which adds functionality)
- [ ] Breaking Change (fix or feature that could cause existing
functionality not to work as expected)
- [ ] Documentation Update
- [ ] Refactoring
- [ ] Performance Improvement
- [ ] Test cases
- [ ] Python SDK impacted, Need to update PyPI
- [ ] Other (please describe):
tags/v0.1.0
KevinHuSh пре 1 година
родитељ
комит
36f2d7b797
No account linked to committer's email address
4 измењених фајлова са 5 додато и 0 уклоњено
  1. 2
    0
      deepdoc/parser/excel_parser.py
  2. 1
    0
      rag/app/manual.py
  3. 1
    0
      rag/app/one.py
  4. 1
    0
      rag/app/table.py

+ 2
- 0
deepdoc/parser/excel_parser.py Прегледај датотеку

@@ -14,6 +14,7 @@ class HuExcelParser:
for sheetname in wb.sheetnames:
ws = wb[sheetname]
rows = list(ws.rows)
if not rows:continue
tb += f"<table><caption>{sheetname}</caption><tr>"
for t in list(rows[0]):
tb += f"<th>{t.value}</th>"
@@ -38,6 +39,7 @@ class HuExcelParser:
for sheetname in wb.sheetnames:
ws = wb[sheetname]
rows = list(ws.rows)
if not rows:continue
ti = list(rows[0])
for r in list(rows[1:]):
l = []

+ 1
- 0
rag/app/manual.py Прегледај датотеку

@@ -109,6 +109,7 @@ def chunk(filename, binary=None, from_page=0, to_page=100000,
sections = [(txt, sec_ids[i], poss)
for i, (txt, _, poss) in enumerate(sections)]
for (img, rows), poss in tbls:
if not rows:continue
sections.append((rows if isinstance(rows, str) else rows[0], -1,
[(p[0] + 1 - from_page, p[1], p[2], p[3], p[4]) for p in poss]))

+ 1
- 0
rag/app/one.py Прегледај датотеку

@@ -44,6 +44,7 @@ class Pdf(PdfParser):
sections = [(b["text"], self.get_position(b, zoomin))
for i, b in enumerate(self.boxes)]
for (img, rows), poss in tbls:
if not rows:continue
sections.append((rows if isinstance(rows, str) else rows[0],
[(p[0] + 1 - from_page, p[1], p[2], p[3], p[4]) for p in poss]))
return [(txt, "") for txt, _ in sorted(sections, key=lambda x: (

+ 1
- 0
rag/app/table.py Прегледај датотеку

@@ -40,6 +40,7 @@ class Excel(ExcelParser):
for sheetname in wb.sheetnames:
ws = wb[sheetname]
rows = list(ws.rows)
if not rows:continue
headers = [cell.value for cell in rows[0]]
missed = set([i for i, h in enumerate(headers) if h is None])
headers = [

Loading…
Откажи
Сачувај