add .libs to path. Fixes #6 (#7)

fujiisoup · Apr 12, 2020 · df2d439 · df2d439
1 parent 59ad68e
commit df2d439
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/py3nj/__init__.py b/py3nj/__init__.py
@@ -1,3 +1,12 @@
+# https://github.com/fujiisoup/py3nj/issues/6
+import os
+import sys
+extra_dll_dir = os.path.join(os.path.dirname(__file__), '.libs')
+
+if sys.platform == 'win32' and os.path.isdir(extra_dll_dir):
+    os.environ.setdefault('PATH', '')
+    os.environ['PATH'] += os.pathsep + extra_dll_dir
+
 from ._version import __version__
 
 from .wigner import wigner3j, wigner6j, wigner9j, clebsch_gordan