Merge pull request #24 from simracingstudio/patch-3

Update library.py
tyrylu · Apr 1, 2021 · 806e402 · 806e402
2 parents 24798bd + 2c5700f
commit 806e402
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/pyfmodex/studio/library.py b/pyfmodex/studio/library.py
@@ -2,6 +2,7 @@
 
 import os
 import platform
+import sys
 
 arch = platform.architecture()[0]
 if os.name == "nt":
@@ -15,10 +16,18 @@
 
 LIBRARY = None
 
-
 def get_library():
     """Load libary file according to architecture's ctype method."""
     global LIBRARY
     if not LIBRARY:
-        LIBRARY = library_type.LoadLibrary(LIBRARY_NAME)
+        try:
+            LIBRARY = library_type.LoadLibrary(LIBRARY_NAME)
+        except:
+            current_directory = os.path.dirname(os.path.realpath(sys.argv[0]))
+            try:
+                LIBRARY = library_type.LoadLibrary(os.path.join(current_directory, LIBRARY_NAME))
+            except:
+                raise RuntimeError(
+                    "Pyfmodex could not find the " + LIBRARY_NAME + " library")
+
     return LIBRARY