Ensure magic.mgc packaged in wheel gets recognised

ddelange · Jan 25, 2024 · 2efa36d · 2efa36d
1 parent 85d4422
commit 2efa36d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/magic/__init__.py b/magic/__init__.py
@@ -354,12 +354,11 @@ def magic_descriptor(cookie, fd):
 
 
 def magic_load(cookie, filename):
-    try:
-        return _magic_load(cookie, coerce_filename(filename))
-    except MagicException:
+    mime_db = os.path.join(os.path.dirname(__file__), 'magic.mgc')
+    if os.path.exists(mime_db):
         # wheels package the mime database in this directory
-        filename = os.path.join(os.path.dirname(__file__), 'magic.mgc')
-        return _magic_load(cookie, coerce_filename(filename))
+        return _magic_load(cookie, coerce_filename(mime_db))
+    return _magic_load(cookie, coerce_filename(filename))
 
 
 magic_setflags = libmagic.magic_setflags

diff --git a/magic/loader.py b/magic/loader.py
@@ -1,7 +1,7 @@
 from ctypes.util import find_library
 import ctypes
 import glob
-import os.path
+import os
 import subprocess
 import sys