93 lines
3.9 KiB
Python
93 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from .handlers import PDFHandler
|
|
from .utils import validate_input, remove_extra
|
|
|
|
|
|
def read_pdf(filepath, pages='1', flavor='lattice', **kwargs):
|
|
"""Read PDF and return extracted tables.
|
|
|
|
Note: kwargs annotated with ^ can only be used with flavor='stream'
|
|
and kwargs annotated with * can only be used with flavor='lattice'.
|
|
|
|
Parameters
|
|
----------
|
|
filepath : str
|
|
Path to PDF file.
|
|
pages : str, optional (default: '1')
|
|
Comma-separated page numbers.
|
|
Example: 1,3,4 or 1,4-end.
|
|
flavor : str (default: 'lattice')
|
|
The parsing method to use ('lattice' or 'stream').
|
|
Lattice is used by default.
|
|
table_area : 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
|
|
in PDF coordinate space.
|
|
columns^ : list, optional (default: None)
|
|
List of column x-coordinates strings where the coordinates
|
|
are comma-separated.
|
|
split_text : bool, optional (default: False)
|
|
Split text that spans across multiple cells.
|
|
flag_size : bool, optional (default: False)
|
|
Flag text based on font size. Useful to detect
|
|
super/subscripts. Adds <s></s> around flagged text.
|
|
row_close_tol^ : int, optional (default: 2)
|
|
Tolerance parameter used to combine text vertically,
|
|
to generate rows.
|
|
col_close_tol^ : int, optional (default: 0)
|
|
Tolerance parameter used to combine text horizontally,
|
|
to generate columns.
|
|
process_background* : bool, optional (default: False)
|
|
Process background lines.
|
|
line_size_scaling* : int, optional (default: 15)
|
|
Line size scaling factor. The larger the value the smaller
|
|
the detected lines. Making it very large will lead to text
|
|
being detected as lines.
|
|
copy_text* : list, optional (default: None)
|
|
{'h', 'v'}
|
|
Direction in which text in a spanning cell will be copied
|
|
over.
|
|
shift_text* : list, optional (default: ['l', 't'])
|
|
{'l', 'r', 't', 'b'}
|
|
Direction in which text in a spanning cell will flow.
|
|
line_close_tol* : int, optional (default: 2)
|
|
Tolerance parameter used to merge close vertical and horizontal
|
|
lines.
|
|
joint_close_tol* : int, optional (default: 2)
|
|
Tolerance parameter used to decide whether the detected lines
|
|
and points lie close to each other.
|
|
threshold_blocksize* : int, optional (default: 15)
|
|
Size of a pixel neighborhood that is used to calculate a
|
|
threshold value for the pixel: 3, 5, 7, and so on.
|
|
|
|
For more information, refer `OpenCV's adaptiveThreshold <https://docs.opencv.org/2.4/modules/imgproc/doc/miscellaneous_transformations.html#adaptivethreshold>`_.
|
|
threshold_constant* : int, optional (default: -2)
|
|
Constant subtracted from the mean or weighted mean.
|
|
Normally, it is positive but may be zero or negative as well.
|
|
|
|
For more information, refer `OpenCV's adaptiveThreshold <https://docs.opencv.org/2.4/modules/imgproc/doc/miscellaneous_transformations.html#adaptivethreshold>`_.
|
|
iterations* : int, optional (default: 0)
|
|
Number of times for erosion/dilation is applied.
|
|
|
|
For more information, refer `OpenCV's dilate <https://docs.opencv.org/2.4/modules/imgproc/doc/filtering.html#dilate>`_.
|
|
margins : tuple
|
|
PDFMiner char_margin, line_margin and word_margin.
|
|
|
|
For more information, refer `PDFMiner docs <https://euske.github.io/pdfminer/>`_.
|
|
|
|
Returns
|
|
-------
|
|
tables : camelot.core.TableList
|
|
|
|
"""
|
|
if flavor not in ['lattice', 'stream']:
|
|
raise NotImplementedError("Unknown flavor specified."
|
|
" Use either 'lattice' or 'stream'")
|
|
|
|
validate_input(kwargs, flavor=flavor)
|
|
p = PDFHandler(filepath, pages)
|
|
kwargs = remove_extra(kwargs, flavor=flavor)
|
|
tables = p.parse(flavor=flavor, **kwargs)
|
|
return tables
|