From d4b49c510afbb8a9938ebb9ab977d280a204c814 Mon Sep 17 00:00:00 2001 From: mara004 Date: Mon, 26 Aug 2024 14:44:34 +0200 Subject: [PATCH] Minor cleanup --- src/img2pdf.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/img2pdf.py b/src/img2pdf.py index 6fc044d..5e5a640 100755 --- a/src/img2pdf.py +++ b/src/img2pdf.py @@ -1075,7 +1075,7 @@ class pdfdoc(object): self.tostream(stream) return stream.getvalue() - def tostream(self, outputstream, return_engine_doc=False): + def tostream(self, outputstream=None, return_engine_doc=False): if self.engine == Engine.pikepdf: PdfArray = pikepdf.Array PdfDict = pikepdf.Dictionary @@ -1271,6 +1271,8 @@ class pdfdoc(object): return self.writer, self.output_version # now write out the PDF + if outputstream is None: + raise TypeError("pdfdoc.tostream() requires outputstream unless return_engine_doc is True.") if self.engine == Engine.pikepdf: kwargs = {} if pikepdf.__version__ >= "6.2.0": @@ -2801,11 +2803,11 @@ def convert(*images, **kwargs): ) if kwargs["outputstream"]: - pdf.tostream(kwargs["outputstream"]) + pdf.tostream(outputstream=kwargs["outputstream"]) return if kwargs["return_engine_doc"]: - return pdf.tostream(None, return_engine_doc=True) + return pdf.tostream(return_engine_doc=True) return pdf.tostring()