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

Prevent taking the max of an empty set

pull/187/head
Paul Visscher 2020-08-25 13:40:51 +02:00
parent 705473198f
commit fe13764026
No known key found for this signature in database
GPG Key ID: 5049271900EEA02C
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
camelot/parsers/stream.py
View File

@ -331,7 +331,7 @@ class Stream(BaseParser):
rows_grouped = self._group_rows(self.t_bbox["horizontal"], row_tol=self.row_tol) rows_grouped = self._group_rows(self.t_bbox["horizontal"], row_tol=self.row_tol)
rows = self._join_rows(rows_grouped, text_y_max, text_y_min) rows = self._join_rows(rows_grouped, text_y_max, text_y_min)
elements = [len(r) for r in rows_grouped] elements = [len(r) for r in rows_grouped]
cols = []
if self.columns is not None and self.columns[table_idx] != "": if self.columns is not None and self.columns[table_idx] != "":
# user has to input boundary columns too # user has to input boundary columns too
# take (0, pdf_width) by default # take (0, pdf_width) by default
@ -342,7 +342,7 @@ class Stream(BaseParser):
cols.insert(0, text_x_min) cols.insert(0, text_x_min)
cols.append(text_x_max) cols.append(text_x_max)
cols = [(cols[i], cols[i + 1]) for i in range(0, len(cols) - 1)] cols = [(cols[i], cols[i + 1]) for i in range(0, len(cols) - 1)]
else: elif elements:
# calculate mode of the list of number of elements in # calculate mode of the list of number of elements in
# each row to guess the number of columns # each row to guess the number of columns
ncols = max(set(elements), key=elements.count) ncols = max(set(elements), key=elements.count)
@ -455,6 +455,7 @@ class Stream(BaseParser):
sorted(self.table_bbox.keys(), key=lambda x: x[1], reverse=True) sorted(self.table_bbox.keys(), key=lambda x: x[1], reverse=True)
): ):
cols, rows = self._generate_columns_and_rows(table_idx, tk) cols, rows = self._generate_columns_and_rows(table_idx, tk)
if cols and rows:
table = self._generate_table(table_idx, cols, rows) table = self._generate_table(table_idx, cols, rows)
table._bbox = tk table._bbox = tk
_tables.append(table) _tables.append(table)