Bugfix: allowing images path starting with 'data' to be passed to FPD…

…F.image() - fix py-pdf#794 (py-pdf#796)
andersonhc · Jun 1, 2023 · 6779a1f · 6779a1f
1 parent ba7d27a
commit 6779a1f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -20,6 +20,8 @@ This can also be enabled programmatically with `warnings.simplefilter('default',
 ### Added
 - [`FPDF.mirror()`](https://pyfpdf.github.io/fpdf2/fpdf/fpdf.html#fpdf.fpdf.FPDF.mirror) - New method: [documentation page](https://pyfpdf.github.io/fpdf2/Transformations.html) - Contributed by @sebastiantia
 - [`FPDF.table()`](https://pyfpdf.github.io/fpdf2/fpdf/fpdf.html#fpdf.fpdf.FPDF.table): new optional parameters `gutter_height`, `gutter_width` and `wrapmode`
+### Fixed
+- [`FPDF.image()`](https://pyfpdf.github.io/fpdf2/fpdf/fpdf.html#fpdf.fpdf.FPDF.image): allowing images path starting with `data` to be passed as input
 
 ## [2.7.4] - 2023-04-28
 ### Added

diff --git a/fpdf/image_parsing.py b/fpdf/image_parsing.py
@@ -73,7 +73,7 @@ def load_image(filename):
         # nosemgrep: python.lang.security.audit.dynamic-urllib-use-detected.dynamic-urllib-use-detected
         with urlopen(filename) as url_file:  # nosec B310
             return BytesIO(url_file.read())
-    elif filename.startswith("data"):
+    elif filename.startswith("data:"):
         return _decode_base64_image(filename)
     with open(filename, "rb") as local_file:
         return BytesIO(local_file.read())