Properly load blah.config (SOFTWARE-2628)

Previous method didn't work because the options in blah.config weren't
ever exported so they never made it ot the environment

Author: brianhlin

src/scripts/blah.py

@@ -1,18 +1,26 @@
 """Common functions for BLAH python scripts"""
 
 import os
-import subprocess
+from ConfigParser import RawConfigParser
+from io import StringIO
 
-def load_env(config_dir):
-    """Load blah.config into the environment"""
-    load_config_path = os.path.join(config_dir, 'blah_load_config.sh')
-    command = ['bash', '-c', 'source %s && env' % load_config_path]
-    try:
-        config_proc = subprocess.Popen(command, stdout=subprocess.PIPE)
-        config_out, _ = config_proc.communicate()
+class BlahConfigParser(RawConfigParser, object):
+
+    def __init__(self, path='/etc/blah.config'):
+        # RawConfigParser requires ini-style [section headers] but since
+        # blah.config is also used as a shell script we need to fake one
+        self.header = 'blahp'
+        with open(path) as f:
+            config = f.read()
+        vfile = StringIO(u'[%s]\n%s' % (self.header, config))
+
+        super(BlahConfigParser, self).__init__()
+        # TODO: readfp() is replaced by read_file() in Python 3.2+
+        self.readfp(vfile)
+
+    def items(self):
+        return super(BlahConfigParser, self).items(self.header)
+
+    def get(self, option):
+        return super(BlahConfigParser, self).get(self.header, option)
 
-        for line in config_out.splitlines():
-            (key, _, val) = line.partition('=')
-            os.environ[key] = val
-    except IOError:
-        pass

src/scripts/pbs_status.py

@@ -233,7 +233,7 @@ def qstat(jobid=""):
     starttime = time.time()
     log("Starting qstat.")
     command = (qstat_bin, '-f')
-    if os.environ.get('pbs_pro').lower() != 'yes':
+    if config.get('pbs_pro').lower() != 'yes':
         command += ('-1',) # -1 conflicts with -f in PBS Pro
     if jobid:
         command += (jobid,)
@@ -358,7 +358,7 @@ def get_qstat_location():
     if _qstat_location_cache != None:
         return _qstat_location_cache
     try:
-        cmd = os.path.join(os.environ['pbs_binpath'], 'qstat')
+        cmd = os.path.join(config.get('pbs_binpath'), 'qstat')
     except KeyError:
         cmd = 'which qstat'
     child_stdout = os.popen(cmd)
@@ -527,7 +527,8 @@ def main():
     jobid = jobid_arg.split("/")[-1].split(".")[0]
 
     config_dir = os.path.dirname(os.path.abspath(__file__))
-    blah.load_env(config_dir)
+    global config
+    config = blah.BlahConfigParser()
 
     log("Checking cache for jobid %s" % jobid)
     cache_contents = None

src/scripts/slurm_status.py

@@ -336,7 +336,7 @@ def get_slurm_location(program):
     if _slurm_location_cache != None:
         return os.path.join(_slurm_location_cache, program)
     try:
-        cmd = os.path.join(os.environ['slurm_binpath'], program)
+        cmd = os.path.join(config.get('slurm_binpath'), program)
     except KeyError:
         cmd = 'which %s' % program
     child_stdout = os.popen(cmd)