Update __init__.py
This commit is contained in:
@ -6,29 +6,24 @@ import sysconfig
|
||||
import ctypes
|
||||
import platform
|
||||
|
||||
library_dir = sysconfig.get_config_vars('LIBDIR')[0]
|
||||
library_name = sysconfig.get_config_vars('LIBRARY')[0]
|
||||
library = library_dir + "/" + library_name
|
||||
|
||||
library_dir = sysconfig.get_config_vars('prefix')[0]
|
||||
py_ver = sysconfig.get_config_vars('VERSION')[0]
|
||||
OS_name = platform.system()
|
||||
|
||||
if OS_name == "Linux":
|
||||
SHLIB_SUFFIX = '.so'
|
||||
library = library_dir + '/lib/' + 'libpython' + py_ver + SHLIB_SUFFIX
|
||||
elif OS_name == "Darwin":
|
||||
SHLIB_SUFFIX = '.dylib'
|
||||
library = library_dir + '/lib/' + 'libpython' + py_ver + SHLIB_SUFFIX
|
||||
elif OS_name == "Windows":
|
||||
SHLIB_SUFFIX = '.dll'
|
||||
else:
|
||||
SHLIB_SUFFIX = sysconfig.get_config_vars('SHLIB_SUFFIX')
|
||||
library = library_dir + '\\' + 'python' + py_ver + SHLIB_SUFFIX
|
||||
|
||||
try:
|
||||
pylib = ctypes.CDLL(library)
|
||||
except OSError as e:
|
||||
if library.endswith(".a"):
|
||||
library = library.strip(".a") + SHLIB_SUFFIX
|
||||
pylib = ctypes.CDLL(library)
|
||||
else:
|
||||
raise e
|
||||
except:
|
||||
OSError
|
||||
|
||||
if not pylib.Py_IsInitialized():
|
||||
raise RuntimeError("This interpreter is not compatible with python-based mliap for LAMMPS.")
|
||||
|
||||
Reference in New Issue
Block a user