imghdr->puremagic

oda-hub · Nov 15, 2024 · ecaad7c · ecaad7c
1 parent fb1b8f4
commit ecaad7c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/oda_api/data_products.py b/oda_api/data_products.py
@@ -41,7 +41,7 @@
 from sys import path_importer_cache, version_info
 
 from io import StringIO, BytesIO
-import imghdr
+import puremagic
 import os
 import logging
 from matplotlib import image as mpimg
@@ -850,7 +850,7 @@ def __init__(self, binary_data, name=None, metadata={}, file_path=None, write_on
         else:
             self.file_path = None                        
         byte_stream = BytesIO(binary_data)
-        tp = imghdr.what(byte_stream)
+        tp = puremagic.what(byte_stream)
         if tp is None:
             raise ValueError('Provided data is not an image')
         self.img_type = tp

diff --git a/pyproject.toml b/pyproject.toml
@@ -29,7 +29,8 @@ dependencies = [
     "astroquery",
     "scipy",
     "rdflib",
-    "black"
+    "black",
+    "puremagic"
 ]
 
 [project.optional-dependencies]