1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
# Copyright (c) 2008-2019 the MRtrix3 contributors.
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
# Covered Software is provided under this License on an "as is"
# basis, without warranty of any kind, either expressed, implied, or
# statutory, including, without limitation, warranties that the
# Covered Software is free of defects, merchantable, fit for a
# particular purpose or non-infringing.
# See the Mozilla Public License v. 2.0 for more details.
#
# For more details, see http://www.mrtrix.org/.
import os, sys
try:
# since importlib code below only works on Python 3.5+
# https://stackoverflow.com/a/50395128
if sys.version_info < (3,5):
raise ImportError
import importlib.util
def imported(lib_path):
try:
spec = importlib.util.spec_from_file_location('mrtrix3', os.path.join (lib_path, 'mrtrix3', '__init__.py'))
module = importlib.util.module_from_spec (spec)
sys.modules[spec.name] = module
spec.loader.exec_module (module)
return True
except ImportError:
return False
except ImportError:
try:
import imp
except ImportError:
print ('failed to import either imp or importlib module!')
sys.exit(1)
def imported(lib_path):
success = False
fp = None
try:
fp, pathname, description = imp.find_module('mrtrix3', [ lib_path ])
imp.load_module('mrtrix3', fp, pathname, description)
success = True
except ImportError:
pass
finally:
if fp:
fp.close()
return success
# Can the MRtrix3 Python modules be found based on their relative location to this file?
# Note that this includes the case where this file is a softlink within an external module,
# which provides a direct link to the core installation
if not imported (os.path.normpath (os.path.join ( \
os.path.dirname (os.path.realpath (__file__)), os.pardir, 'lib') )):
# If this file is a duplicate, which has been stored in an external module,
# we may be able to figure out the location of the core library using the
# build script.
# case 1: build is a symbolic link:
if not imported (os.path.join (os.path.dirname (os.path.realpath ( \
os.path.join (os.path.dirname(__file__), os.pardir, 'build'))), 'lib')):
# case 2: build is a file containing the path to the core build script:
try:
with open (os.path.join (os.path.dirname(__file__), os.pardir, 'build')) as fp:
for line in fp:
build_path = line.split ('#',1)[0].strip()
if build_path:
break
except IOError:
pass
if not imported (os.path.join (os.path.dirname (build_path), 'lib')):
sys.stderr.write('''
ERROR: Unable to locate MRtrix3 Python modules
For detailed instructions, please refer to:
https://mrtrix.readthedocs.io/en/latest/tips_and_tricks/external_modules.html
''')
sys.stderr.flush()
sys.exit(1)
|
