Controller.py

Go to the documentation of this file.

from config_parser import config_parser
import os
import cPickle
import Component
import datetime

class Controller():

    def __init__(self, config_file, verbose=False):
        self.verbose = verbose
        self.config  = config_parser(config_file)
        self.name    = self.config["name"]
        self.version = self.config["version"]
        self.release = self.config["release"]
        self.user    = os.environ["NOVAANAVALID_USERNAME"]
        
        optional_attributes = ["description"]
        for attr in optional_attributes:
            if attr in self.config.keys():
                setattr(self, attr, self.config[attr])
            else:
                setattr(self, attr, False)

        self.components = []
        for component_config in self.config["components"]:
            self.components.append(Component.Component(component_config, release=self.release, verbose=self.verbose, user=self.user))
        self.output_dir = False
        self.creation_time = datetime.datetime.now()

    def make_output_directory(self, output_directory, overwrite=False):
        self.output_dir = "%s/%s/v%s"%(output_directory, self.config["name"], self.config["version"])
        if self.verbose: print "cntrl:     output directory will be: %s"%self.output_dir
        if (not overwrite) and os.path.exists(self.output_dir):
            print "cntrl:     output directory:"
            print self.output_dir
            print "cntrl:     already exists, aborting\n"
            exit(1)
        os.system("mkdir -p %s"%self.output_dir)
        for component in self.components:
            component.output_dir = "%s/%s/v%s"%(self.output_dir, component.name, component.version)
            if component.dataset: component.output_dir += "/%s"%component.dataset
            os.system("mkdir -p %s"%component.output_dir)
            if component.sam:
                os.system("mkdir -p %s/hadd_hist"%component.output_dir)
                os.system("mkdir -p %s/logs"%component.output_dir)

    def write(self):
        if (not self.output_dir):
            print "cntrl: Output directory has not been set, so cannot write"
            return False
        self.file_name = self.output_dir+"/controller.pkl"
        print "cntrl: Saving as: %s"%self.file_name
        cPickle.dump(self,open(self.file_name,"w"))