diff env/lib/python3.9/site-packages/galaxy/util/image_util.py @ 0:4f3585e2f14b draft default tip

"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"
author shellac
date Mon, 22 Mar 2021 18:12:50 +0000
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/env/lib/python3.9/site-packages/galaxy/util/image_util.py	Mon Mar 22 18:12:50 2021 +0000
@@ -0,0 +1,40 @@
+"""Provides utilities for working with image files."""
+
+import imghdr
+import logging
+
+try:
+    import Image as PIL
+except ImportError:
+    try:
+        from PIL import Image as PIL
+    except ImportError:
+        PIL = None
+
+log = logging.getLogger(__name__)
+
+
+def image_type(filename):
+    fmt = None
+    if PIL is not None:
+        try:
+            im = PIL.open(filename)
+            fmt = im.format
+            im.close()
+        except Exception:
+            # We continue to try with imghdr, so this is a rare case of an
+            # exception we expect to happen frequently, so we're not logging
+            pass
+    if not fmt:
+        fmt = imghdr.what(filename)
+    if fmt:
+        return fmt.upper()
+    else:
+        return False
+
+
+def check_image_type(filename, types):
+    fmt = image_type(filename)
+    if fmt in types:
+        return True
+    return False