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

Add test for page list generation and fix backend kwarg

Browse Source
This commit is contained in:
Vinayak Mehta
2021-07-11 22:47:01 +05:30
parent 8650f25331
commit cc820b9e5d
2 changed files with 51 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files
camelot/handlers.py
+21 -18
View File
@@ -47,9 +47,9 @@ class PDFHandler(object):
self.password = password
if sys.version_info[0] < 3:
self.password = self.password.encode("ascii")
self.pages = self._get_pages(self.filepath, pages)
self.pages = self._get_pages(pages)
def _get_pages(self, filepath, pages):
def _get_pages(self, pages):
"""Converts pages string to list of ints.
Parameters
@@ -67,25 +67,28 @@ class PDFHandler(object):
"""
page_numbers = []
if pages == "1":
page_numbers.append({"start": 1, "end": 1})
else:
instream = open(filepath, "rb")
infile = PdfFileReader(instream, strict=False)
if infile.isEncrypted:
infile.decrypt(self.password)
if pages == "all":
page_numbers.append({"start": 1, "end": infile.getNumPages()})
else:
for r in pages.split(","):
if "-" in r:
a, b = r.split("-")
if b == "end":
b = infile.getNumPages()
page_numbers.append({"start": int(a), "end": int(b)})
else:
page_numbers.append({"start": int(r), "end": int(r)})
instream.close()
with open(self.filepath, "rb") as f:
infile = PdfFileReader(f, strict=False)
if infile.isEncrypted:
infile.decrypt(self.password)
if pages == "all":
page_numbers.append({"start": 1, "end": infile.getNumPages()})
else:
for r in pages.split(","):
if "-" in r:
a, b = r.split("-")
if b == "end":
b = infile.getNumPages()
page_numbers.append({"start": int(a), "end": int(b)})
else:
page_numbers.append({"start": int(r), "end": int(r)})
P = []
for p in page_numbers:
P.extend(range(p["start"], p["end"] + 1))