Added multi parameter for page level parameters

Author: sverma25

camelot/handlers.py

@@ -36,7 +36,7 @@ class PDFHandler:
 
     """
 
-    def __init__(self, filepath: Union[StrByteType, Path], pages="1", password=None):
+    def __init__(self, filepath: Union[StrByteType, Path], pages="1", password=None, multi=[]):
         if is_url(filepath):
             filepath = download_url(filepath)
         self.filepath: Union[StrByteType, Path] = filepath
@@ -51,6 +51,7 @@ def __init__(self, filepath: Union[StrByteType, Path], pages="1", password=None)
             if sys.version_info[0] < 3:
                 self.password = self.password.encode("ascii")
         self.pages = self._get_pages(pages)
+        self.multi = multi
 
     def _get_pages(self, pages):
         """Converts pages string to list of ints.
@@ -188,6 +189,15 @@ def parse(
                 with mp.get_context("spawn").Pool(processes=cpu_count) as pool:
                     jobs = []
                     for p in self.pages:
+                        p_no = p
+
+                        page_kwargs = kwargs
+                        page_parser = parser
+
+                        if p_no in self.multi:
+                            page_kwargs.update(self.multi[p_no])
+                            page_parser = Lattice(**page_kwargs) if flavor == 'lattice' else Stream(**page_kwargs)
+
                         j = pool.apply_async(
                             self._parse_page,(p, tempdir, parser, suppress_stdout, layout_kwargs)
                         )
@@ -198,6 +208,15 @@ def parse(
                         tables.extend(t)
             else:
                 for p in self.pages:
+                    p_no = p
+
+                    page_kwargs = kwargs
+                    page_parser = parser
+
+                    if p_no in self.multi:
+                        page_kwargs.update(self.multi[p_no])
+                        page_parser = Lattice(**page_kwargs) if flavor == 'lattice' else Stream(**page_kwargs)
+
                     t = self._parse_page(p, tempdir, parser, suppress_stdout, layout_kwargs)
                     tables.extend(t)
 
@@ -224,7 +243,7 @@ def _parse_page(
         -------
         tables : camelot.core.TableList
             List of tables found in PDF.
-        
+
         """
         self._save_page(self.filepath, page, tempdir)
         page_path = os.path.join(tempdir, f"page-{page}.pdf")

camelot/io.py

@@ -17,6 +17,7 @@ def read_pdf(
     suppress_stdout=False,
     parallel=False,
     layout_kwargs=None,
+    multi = {},
     **kwargs
 ):
     """Read PDF and return extracted tables.
@@ -43,6 +44,9 @@ def read_pdf(
     layout_kwargs : dict, optional (default: {})
         A dict of `pdfminer.layout.LAParams
         <https://github.com/euske/pdfminer/blob/master/pdfminer/layout.py#L33>`_ kwargs.
+    multi: dict, optional(default: {})
+        A dict to enter parameters specific only for a page. Key: Page(str) to dict(defined params).
+        Parameters defined in multi overwrite kwargs for that page
     table_areas : list, optional (default: None)
         List of table area strings of the form x1,y1,x2,y2
         where (x1, y1) -> left-top and (x2, y2) -> right-bottom
@@ -120,7 +124,7 @@ def read_pdf(
             warnings.simplefilter("ignore")
 
         validate_input(kwargs, flavor=flavor)
-        p = PDFHandler(filepath, pages=pages, password=password)
+        p = PDFHandler(filepath, pages=pages, password=password, multi=multi)
         kwargs = remove_extra(kwargs, flavor=flavor)
         tables = p.parse(
             flavor=flavor,