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
|
# Copyright (C) by Josh Blum. See LICENSE.txt for licensing information.
import os
import imp
import traceback
#try to import module
def __try_module_import(filename):
directory, module_name = os.path.split(filename)
module_name = os.path.splitext(module_name)[0]
try:
fp, pathname, description = imp.find_module(module_name, [directory])
module = imp.load_module(module_name, fp, pathname, description)
except Exception as ex:
print 'Could not import', filename, ex
print traceback.format_exc()
#recursive search for modules in path
def __module_import(p):
if not os.path.exists(p): return
if os.path.isfile(p):
return __try_module_import(p)
if not os.path.isdir(p): return
if os.path.exists(os.path.join(p, '__init__.py')):
return __try_module_import(p)
for sub in os.listdir(p):
name, ext = os.path.splitext(sub)
#prefer .pyo over .pyc, prefer .pyc over .py
has_pyc = os.path.exists(os.path.join(p, name+'.pyc'))
has_pyo = os.path.exists(os.path.join(p, name+'.pyo'))
if ext == ".py" and (has_pyc or has_pyo): continue
if ext == ".pyc" and has_pyo: continue
__module_import(os.path.join(p, sub))
#separate the paths and load each one
def __load_modules_from_paths(paths, suffix):
if not paths: return
for path in paths.split(os.pathsep):
if not path: continue
if suffix: path = os.path.join(path, suffix)
__module_import(path)
__load_modules_from_paths(os.getenv("GRAS_ROOT", "@GRAS_ROOT@"), os.path.join("lib@LIB_SUFFIX@", "gras", "python"))
__load_modules_from_paths(os.getenv("GRAS_PATH", ""), os.path.join("lib@LIB_SUFFIX@", "gras", "python"))
__load_modules_from_paths(os.getenv("GRAS_PYTHON_PATH", ""), "")
|