crabConfigParser.py

Go to the documentation of this file.

from configparser import RawConfigParser


class CrabConfigParser(RawConfigParser):

    
    def __init__(self):
        RawConfigParser.__init__(self)
        self.optionxform = str
    
    def writeCrabConfig(self,filename):
        sections = self.sections()
        fixedsections = ['General','JobType','Data','Site','User','Debug']
        outlines = []
        # Add inital header for crab config file
        outlines.append('from WMCore.Configuration import Configuration \n')
        outlines.append('config = Configuration()')
        # we will first add the main crab3 config sections in the given order
        for fixedsection in fixedsections:
            if fixedsection in sections:
                outlines.extend(self.getSectionLines(fixedsection))
                sections.remove(fixedsection)
        # add additional sections (may be added in future crab3 versions ?)
        for section in sections:
            outlines.extend(self.getSectionLines(section))
        #print filename
        with open(filename,'w') as outfile:
            for line in outlines:
                outfile.write(f"{line}\n")
    
    def getSectionLines(self,section):
        sectionLines = []
        sectionLines.append('\nconfig.section_("%s")'%section)
        configItems =  self.items(section)
        for configItem in configItems:
            if not isinstance(configItem[1], str):
                sectionLines.append('config.%s.%s = %s'%(section,configItem[0],configItem[1]))
            elif "True" in configItem[1] or "False" in configItem[1]:
                sectionLines.append('config.%s.%s = %s'%(section,configItem[0],configItem[1]))
            else:
                parsed = False
                if configItem[0]=="runRange" :
                    sectionLines.append('config.%s.%s = \'%s\''%(section,configItem[0],configItem[1]))
                    parsed = True
                if not parsed:
                    try:
                        sectionLines.append('config.%s.%s = %d'%(section,configItem[0],int(configItem[1])))
                        parsed = True
                    except:
                        pass
                if not parsed:
                    try:
                        sectionLines.append('config.%s.%s = %.2f'%(section,configItem[0],float(configItem[1])))
                        parsed = True
                    except:
                        pass
                if not parsed:
                    if isinstance(configItem[1], list):
                        sectionLines.append('config.%s.%s = %s'%(section,configItem[0],str(configItem[1])))
                        parsed = True
                if not parsed:
                    sectionLines.append('config.%s.%s = \'%s\''%(section,configItem[0],configItem[1]))

        return sectionLines