Briq
/
camelot-py
6
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Log warning when len(header) != len(cols)

pull/2/head
Vinayak Mehta 2016-10-17 18:16:39 +05:30 committed by GitHub
parent adb948d363
commit 2a203a1865
3 changed files with 32 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
camelot/lattice.py
View File

@ -79,10 +79,10 @@ def _fill_spanning(t, fill=None):
t : object
camelot.table.Table
fill : string
{'h', 'v', 'hv'}
Specify to fill spanning cells in horizontal, vertical or both
directions.
fill : list
{'h', 'v'}
Specify to fill spanning cells in horizontal or vertical
direction.
(optional, default: None)
Returns
@ -90,26 +90,19 @@ def _fill_spanning(t, fill=None):
t : object
camelot.table.Table
"""
if fill == "h":
for f in fill:
if f == "h":
for i in range(len(t.cells)):
for j in range(len(t.cells[i])):
if t.cells[i][j].get_text().strip() == '':
if t.cells[i][j].spanning_h:
t.cells[i][j].add_text(t.cells[i][j - 1].get_text())
elif fill == "v":
elif f == "v":
for i in range(len(t.cells)):
for j in range(len(t.cells[i])):
if t.cells[i][j].get_text().strip() == '':
if t.cells[i][j].spanning_v:
t.cells[i][j].add_text(t.cells[i - 1][j].get_text())
elif fill == "hv":
for i in range(len(t.cells)):
for j in range(len(t.cells[i])):
if t.cells[i][j].get_text().strip() == '':
if t.cells[i][j].spanning_h:
t.cells[i][j].add_text(t.cells[i][j - 1].get_text())
elif t.cells[i][j].spanning_v:
t.cells[i][j].add_text(t.cells[i - 1][j].get_text())
return t
@ -131,8 +124,8 @@ class Lattice:
fill : list
List of strings specifying directions to fill spanning cells.
{'h', 'v', 'hv'} to fill spanning cells in horizontal, vertical
or both directions.
{'h', 'v'} to fill spanning cells in horizontal or vertical
direction.
(optional, default: None)
headers : list
@ -256,7 +249,6 @@ class Lattice:
if self.headers is not None:
if len(self.table_area) != len(self.headers):
raise ValueError("Length of headers should be equal to table_area.")
self.headers = [h.split(',') for h in headers]
areas = []
for area in self.table_area:
@ -316,7 +308,9 @@ class Lattice:
rows = [(rows[i], rows[i + 1])
for i in range(0, len(rows) - 1)]
if self.headers is not None and len(self.headers[table_no]) != len(cols):
if self.headers is not None and self.headers[table_no] != [""]:
self.headers[table_no] = self.headers[table_no].split(',')
if len(self.headers[table_no]) != len(cols):
logging.warning("Length of header ({0}) specified for table is not"
" equal to the number of columns ({1}) detected.".format(
len(self.headers[table_no]), len(cols)))

5
camelot/stream.py
View File

@ -325,7 +325,6 @@ class Stream:
if self.headers is not None:
if len(self.table_area) != len(self.headers):
raise ValueError("Length of headers should be equal to table_area.")
self.headers = [h.split(',') for h in headers]
table_bbox = {}
for area in self.table_area:
@ -418,7 +417,9 @@ class Stream:
cols = _add_columns(cols, inner_text, self.ytol[table_no])
cols = _join_columns(cols, text_x_min, text_x_max)
if self.headers is not None and len(self.headers[table_no]) != len(cols):
if self.headers is not None and self.headers[table_no] != [""]:
self.headers[table_no] = self.headers[table_no].split(',')
if len(self.headers[table_no]) != len(cols):
logging.warning("Length of header ({0}) specified for table is not"
" equal to the number of columns ({1}) detected.".format(
len(self.headers[table_no]), len(cols)))

2
camelot/utils.py
View File

@ -605,7 +605,7 @@ def split_textline(table, textline, direction, flag_size=True):
if isinstance(obj, LTChar):
if (col[0] <= (obj.x0 + obj.x1) / 2 <= col[1] and
(obj.y0 + obj.y1) / 2 >= cut[1]):
cut_text.append((cut[0], c, obj.get_text()))
cut_text.append((cut[0], c, obj))
break
elif isinstance(obj, LTAnno):
cut_text.append((cut[0], c, obj))