forked from SciLifeLab/bcbio-nextgen-deploy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
moduleSystem.py
63 lines (56 loc) · 2.11 KB
/
moduleSystem.py
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
"""ModuleSystem is a module which manages the modules system (see
modules.sourceforge.net for more information about the module system).
Currently this module is capable of:
* List all available modules in a system
* List already loaded modules
Future approaches:
* List different versions of a module
Author: Guillermo Carrasco Hernandez
Contact: [email protected]
"""
import os
import subprocess
import sets
class moduleManager(object):
"""Class to list and manage the different modules in a system
(modules.sourceforge.net)
"""
def __new__(typ, *args, **kwargs):
"""Check if the module system is installed before cerating an instance
of the class.
"""
try:
os.environ['MODULESHOME']
return object.__new__(typ, *args, **kwargs)
except KeyError:
raise RuntimeError('No module system found!')
return None
def __init__(self):
self.env = dict(os.environ)
try:
self.modules_version = self.env['MODULES_REL']
except KeyError:
self.modules_version = 'Undefined'
module_list = 'module bash avait -t'
print "Obtaining information of the modules in the system..."
output = subprocess.check_output(module_list, shell = True, \
stderr=subprocess.STDOUT)
#Prepare the list of modules
output = output.split()
moduleList = dict()
for module in output:
if module[-1] != ':':
try:
(software, version) = module.split('/')
except ValueError:
(software, version) = module, None
if moduleList.has_key(software):
moduleList[software].add(version)
else:
moduleList[software] = set([version])
self._moduleList = moduleList
print "Module manager initialized correctly!"
def has_module(self, module):
"""Returns true if module is in the system
"""
return self._moduleList.has_key(module)