Add extra_kwargs
This commit is contained in:
@@ -8,13 +8,10 @@ from ..utils import get_page_layout, get_text_objects
|
||||
class BaseParser(object):
|
||||
"""Defines a base parser.
|
||||
"""
|
||||
def _generate_layout(self, filename):
|
||||
def _generate_layout(self, filename, extra_kwargs):
|
||||
self.filename = filename
|
||||
self.layout, self.dimensions = get_page_layout(
|
||||
self.filename,
|
||||
char_margin=self.char_margin,
|
||||
line_margin=self.line_margin,
|
||||
word_margin=self.word_margin)
|
||||
self.filename, **self.extra_kwargs)
|
||||
self.horizontal_text = get_text_objects(self.layout, ltype="lh")
|
||||
self.vertical_text = get_text_objects(self.layout, ltype="lv")
|
||||
self.pdf_width, self.pdf_height = self.dimensions
|
||||
|
||||
@@ -348,8 +348,8 @@ class Lattice(BaseParser):
|
||||
|
||||
return table
|
||||
|
||||
def extract_tables(self, filename, suppress_stdout=False):
|
||||
self._generate_layout(filename)
|
||||
def extract_tables(self, filename, suppress_stdout=False, extra_kwargs={}):
|
||||
self._generate_layout(filename, extra_kwargs)
|
||||
if not suppress_stdout:
|
||||
logger.info('Processing {}'.format(os.path.basename(self.rootname)))
|
||||
|
||||
|
||||
@@ -388,8 +388,8 @@ class Stream(BaseParser):
|
||||
|
||||
return table
|
||||
|
||||
def extract_tables(self, filename, suppress_stdout=False):
|
||||
self._generate_layout(filename)
|
||||
def extract_tables(self, filename, suppress_stdout=False, extra_kwargs={}):
|
||||
self._generate_layout(filename, extra_kwargs)
|
||||
if not suppress_stdout:
|
||||
logger.info('Processing {}'.format(os.path.basename(self.rootname)))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user