From 2b563c0982d59d9d75c252d91930ed1ea73a611d Mon Sep 17 00:00:00 2001 From: mara004 Date: Tue, 27 Aug 2024 22:26:22 +0200 Subject: [PATCH] Update to `getattr()` This now a one-liner, compared to 6 lines originally. And it should be (algorithmically) faster. --- src/img2pdf.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/img2pdf.py b/src/img2pdf.py index 76439c3..3b9ba99 100755 --- a/src/img2pdf.py +++ b/src/img2pdf.py @@ -1832,10 +1832,7 @@ def read_images( ) else: logger.debug("PIL format = %s", imgdata.format) - try: - imgformat = ImageFormat[imgdata.format] - except KeyError: - imgformat = ImageFormat.other + imgformat = getattr(ImageFormat, imgdata.format, ImageFormat.other) def cleanup(): if imgdata is not None: