pdfrw is now disabled by default if it's not installed (closes: #74)

src/img2pdf.py

@@ -34,6 +34,14 @@ import logging
 import struct
 import platform
 
+with_pdfrw = False
+try:
+    import pdfrw
+
+    with_pdfrw = True
+except ImportError:
+    pass
+
 PY3 = sys.version_info[0] >= 3
 
 __version__ = "0.3.4"
@@ -631,25 +639,14 @@ class pdfdoc(object):
         fit_window=False,
         center_window=False,
         fullscreen=False,
-        with_pdfrw=True,
     ):
         if with_pdfrw:
-            try:
             from pdfrw import PdfWriter, PdfDict, PdfName, PdfString
-
-                self.with_pdfrw = True
-            except ImportError:
-                PdfWriter = MyPdfWriter
-                PdfDict = MyPdfDict
-                PdfName = MyPdfName
-                PdfString = MyPdfString
-                self.with_pdfrw = False
         else:
             PdfWriter = MyPdfWriter
             PdfDict = MyPdfDict
             PdfName = MyPdfName
             PdfString = MyPdfString
-            self.with_pdfrw = False
 
         now = datetime.now()
         self.info = PdfDict(indirect=True)
@@ -690,7 +687,7 @@ class pdfdoc(object):
         self.writer.version = version
         # this is done because pdfrw adds info, catalog and pages as the first
         # three objects in this order
-        if not self.with_pdfrw:
+        if not with_pdfrw:
             self.writer.addobj(self.info)
             self.writer.addobj(self.writer.catalog)
             self.writer.addobj(self.writer.pages)
@@ -726,7 +723,7 @@ class pdfdoc(object):
         trimborder=None,
         artborder=None,
     ):
-        if self.with_pdfrw:
+        if with_pdfrw:
             from pdfrw import PdfDict, PdfName, PdfObject, PdfString
             from pdfrw.py23_diffs import convert_load
         else:
@@ -743,7 +740,7 @@ class pdfdoc(object):
         elif color == Colorspace.CMYK or color == Colorspace["CMYK;I"]:
             colorspace = PdfName.DeviceCMYK
         elif color == Colorspace.P:
-            if self.with_pdfrw:
+            if with_pdfrw:
                 raise Exception(
                     "pdfrw does not support hex strings for "
                     "palette image input, re-run with "
@@ -871,7 +868,7 @@ class pdfdoc(object):
 
         self.writer.addpage(page)
 
-        if not self.with_pdfrw:
+        if not with_pdfrw:
             self.writer.addobj(content)
             self.writer.addobj(image)
 
@@ -881,7 +878,7 @@ class pdfdoc(object):
         return stream.getvalue()
 
     def tostream(self, outputstream):
-        if self.with_pdfrw:
+        if with_pdfrw:
             from pdfrw import PdfDict, PdfName, PdfArray, PdfObject
         else:
             PdfDict = MyPdfDict
@@ -899,7 +896,7 @@ class pdfdoc(object):
         # is added, so we can only start using it after all pages have been
         # written.
 
-        if self.with_pdfrw:
+        if with_pdfrw:
             catalog = self.writer.trailer.Root
         else:
             catalog = self.writer.catalog
@@ -1005,7 +1002,7 @@ class pdfdoc(object):
             raise ValueError("unknown page layout: %s" % self.page_layout)
 
         # now write out the PDF
-        if self.with_pdfrw:
+        if with_pdfrw:
             self.writer.trailer.Info = self.info
             self.writer.write(outputstream)
         else:
@@ -1809,7 +1806,6 @@ def convert(*images, **kwargs):
         viewer_fit_window=False,
         viewer_center_window=False,
         viewer_fullscreen=False,
-        with_pdfrw=True,
         outputstream=None,
         first_frame_only=False,
         allow_oversized=True,
@@ -1821,6 +1817,9 @@ def convert(*images, **kwargs):
     for kwname, default in _default_kwargs.items():
         if kwname not in kwargs:
             kwargs[kwname] = default
+    if 'with_pdfrw' in kwargs:
+        global with_pdfrw
+        with_pdfrw = kwargs['with_pdfrw']
 
     pdf = pdfdoc(
         "1.3",
@@ -1840,7 +1839,6 @@ def convert(*images, **kwargs):
         kwargs["viewer_fit_window"],
         kwargs["viewer_center_window"],
         kwargs["viewer_fullscreen"],
-        kwargs["with_pdfrw"],
     )
 
     # backwards compatibility with older img2pdf versions where the first
@@ -2492,6 +2490,8 @@ def gui():
                 v for v in Colorspace if v.name == args["colorspace"].get()
             )
 
+        global with_pdfrw
+        with_pdfrw = (not args["without_pdfrw"].get(),)
         convert(
             *infiles,
             title=args["title"].get() if args["title"].get() else None,
@@ -2516,7 +2516,6 @@ def gui():
             viewer_fit_window=(args["viewer_fit_window"].get() or None),
             viewer_center_window=(args["viewer_center_window"].get() or None),
             viewer_fullscreen=(args["viewer_fullscreen"].get() or None),
-            with_pdfrw=not args["without_pdfrw"].get(),
             outputstream=stream,
             first_frame_only=args["first_frame_only"].get(),
             cropborder=None,
@@ -3490,6 +3489,8 @@ and left/right, respectively. It is not possible to specify asymmetric borders.
             )
             exit(2)
 
+    global with_pdfrw
+    with_pdfrw = (not args.without_pdfrw,)
     try:
         convert(
             *args.images,
@@ -3511,7 +3512,6 @@ and left/right, respectively. It is not possible to specify asymmetric borders.
             viewer_fit_window=args.viewer_fit_window,
             viewer_center_window=args.viewer_center_window,
             viewer_fullscreen=args.viewer_fullscreen,
-            with_pdfrw=not args.without_pdfrw,
             outputstream=args.output,
             first_frame_only=args.first_frame_only,
             cropborder=args.crop_border,

src/tests/__init__.py

@@ -554,12 +554,17 @@ def test_suite():
         assert os.path.isfile(outputf)
 
         def handle(self, f=inputf, out=outputf, with_pdfrw=with_pdfrw):
+            try:
+                from pdfrw import PdfReader, PdfName, PdfWriter
+                from pdfrw.py23_diffs import convert_load, convert_store
+            except ImportError:
+                # the test requires pdfrw
+                self.skipTest("this test requires pdfrw")
+                return
             with open(f, "rb") as inf:
                 orig_imgdata = inf.read()
             output = img2pdf.convert(orig_imgdata, nodate=True,
                                      with_pdfrw=with_pdfrw)
-            from pdfrw import PdfReader, PdfName, PdfWriter
-            from pdfrw.py23_diffs import convert_load, convert_store
             x = PdfReader(PdfReaderIO(convert_load(output)))
             self.assertEqual(sorted(x.keys()), [PdfName.Info, PdfName.Root,
                              PdfName.Size])