Commit 2e109f94 authored by Aurel's avatar Aurel Committed by Kazuhiko Shiozaki

use new pdf lib

parent a561f0f1
...@@ -81,7 +81,11 @@ class PDFDocument(Image): ...@@ -81,7 +81,11 @@ class PDFDocument(Image):
* Watermark is applied at all pages starting watermark_start_page (this * Watermark is applied at all pages starting watermark_start_page (this
index is 0 based) index is 0 based)
""" """
from pyPdf import PdfFileWriter, PdfFileReader try:
from PyPDF2 import PdfFileWriter, PdfFileReader
except ImportError:
pass
else:
if not watermark_data: if not watermark_data:
raise ValueError("watermark_data cannot not be empty") raise ValueError("watermark_data cannot not be empty")
if not self.hasData(): if not self.hasData():
...@@ -301,12 +305,12 @@ class PDFDocument(Image): ...@@ -301,12 +305,12 @@ class PDFDocument(Image):
value = ':'.join(item_list[1:]).strip() value = ':'.join(item_list[1:]).strip()
result[key] = value result[key] = value
# Then we use pyPdf to get extra metadata # Then we use PyPDF2 to get extra metadata
try: try:
from pyPdf import PdfFileReader from PyPDF2 import PdfFileReader
from pyPdf.utils import PdfReadError from PyPDF2.utils import PdfReadError
except ImportError: except ImportError:
# if pyPdf not found, pass # if PyPDF2 not found, pass
pass pass
else: else:
try: try:
...@@ -327,7 +331,7 @@ class PDFDocument(Image): ...@@ -327,7 +331,7 @@ class PDFDocument(Image):
result.setdefault(info_key, info_value) result.setdefault(info_key, info_value)
except PdfReadError: except PdfReadError:
LOG("PDFDocument.getContentInformation", PROBLEM, LOG("PDFDocument.getContentInformation", PROBLEM,
"pyPdf is Unable to read PDF, probably corrupted PDF here : %s" % \ "PyPDF2 is Unable to read PDF, probably corrupted PDF here : %s" % \
(self.getRelativeUrl(),)) (self.getRelativeUrl(),))
finally: finally:
tmp.close() tmp.close()
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment