helperFunctions Namespace Reference

Functions
def	addIndex (filename, njobs, index=None)
def	boolfromstring (string, name)
def	cache (function)
def	castorDirExists (path)
def	clean_name (s)
def	cppboolstring (string, name)
def	getCommandOutput2 (command)
def	getTagsMap (db)
def	parsecolor (color)
def	parsestyle (style)
def	pythonboolstring (string, name)
def	recursivesubclasses (cls)
def	replaceByMap (target, the_map)
	— Helpers —############################ More...
def	replacelast (string, old, new, count=1)

Variables
list	fileExtensions = ["_cfg.py", ".sh", ".root"]

Function Documentation

addIndex()

def helperFunctions.addIndex	(		filename,
			njobs,
			index = None
	)

Definition at line 94 of file helperFunctions.py.

def addIndex(filename, njobs, index = None):
    if index is None:
        return [addIndex(filename, njobs, i) for i in range(njobs)]
    if njobs == 1:
        return filename
 
    fileExtension = None
    for extension in fileExtensions:
        if filename.endswith(extension):
            fileExtension = extension
    if fileExtension is None:
        raise AllInOneError(fileName + " does not end with any of the extensions "
                                     + str(fileExtensions))
    return replacelast(filename, fileExtension, "_" + str(index) + fileExtension)
 

References FastTimerService_cff.range, replacelast(), and str.

Referenced by genericValidation.GenericValidation.createFiles(), genericValidation.GenericValidationData.createScript(), and genericValidation.GenericValidationData.getRepMap().

boolfromstring()

def helperFunctions.boolfromstring	(		string,
			name
	)

Takes a string from the configuration file
and makes it into a bool

Definition at line 170 of file helperFunctions.py.

def boolfromstring(string, name):
    """
    Takes a string from the configuration file
    and makes it into a bool
    """
    #try as a string, not case sensitive
    if string.lower() == "true": return True
    if string.lower() == "false": return False
    #try as a number
    try:
        return str(bool(int(string)))
    except ValueError:
        pass
    #out of options
    raise ValueError("{} has to be true or false!".format(name))
    
 

References electrons_cff.bool, createfilelist.int, and str.

Referenced by pythonboolstring().

cache()

def helperFunctions.cache

(

function

)

Definition at line 156 of file helperFunctions.py.

def cache(function):
    cache = {}
    def newfunction(*args, **kwargs):
        try:
            return cache[args, tuple(sorted(six.iteritems(kwargs)))]
        except TypeError:
            print(args, tuple(sorted(six.iteritems(kwargs))))
            raise
        except KeyError:
            cache[args, tuple(sorted(six.iteritems(kwargs)))] = function(*args, **kwargs)
            return newfunction(*args, **kwargs)
    newfunction.__name__ = function.__name__
    return newfunction
 

References edm.print().

castorDirExists()

def helperFunctions.castorDirExists

(

path

)

This function checks if the directory given by `path` exists.

Arguments:
- `path`: Path to castor directory

Definition at line 67 of file helperFunctions.py.

def castorDirExists(path):
    """This function checks if the directory given by `path` exists.
 
    Arguments:
    - `path`: Path to castor directory
    """
 
    if path[-1] == "/":
        path = path[:-1]
    containingPath = os.path.join( *path.split("/")[:-1] )
    dirInQuestion = path.split("/")[-1]
    try:
        rawLines = getCommandOutput2("rfdir /"+containingPath).splitlines()
    except RuntimeError:
        return False
    for line in rawLines:
        if line.split()[0][0] == "d":
            if line.split()[8] == dirInQuestion:
                return True
    return False
 

References getCommandOutput2().

clean_name()

def helperFunctions.clean_name

(

s

)

Transforms a string into a valid variable or method name.

Arguments:
- `s`: input string

Definition at line 213 of file helperFunctions.py.

def clean_name(s):
    """Transforms a string into a valid variable or method name.
 
    Arguments:
    - `s`: input string
    """
 
    # Remove invalid characters
    s = re.sub(r"[^0-9a-zA-Z_]", "", s)
 
    # Remove leading characters until we find a letter or underscore
    s = re.sub(r"^[^a-zA-Z_]+", "", s)
 
    return s

Referenced by geometryComparison.GeometryComparison.getRepMap().

cppboolstring()

def helperFunctions.cppboolstring	(		string,
			name
	)

Takes a string from the configuration file
and makes it into a bool string for a C++ template

Definition at line 194 of file helperFunctions.py.

def cppboolstring(string, name):
    """
    Takes a string from the configuration file
    and makes it into a bool string for a C++ template
    """
    return pythonboolstring(string, name).lower()
 

References pythonboolstring().

Referenced by plottingOptions.PlottingOptionsZMuMu.__init__(), plottingOptions.PlottingOptionsOffline.__init__(), and plottingOptions.PlottingOptionsPrimaryVertex.__init__().

getCommandOutput2()

def helperFunctions.getCommandOutput2

(

command

)

This function executes `command` and returns it output.

Arguments:
- `command`: Shell command to be invoked by this function.

Definition at line 52 of file helperFunctions.py.

def getCommandOutput2(command):
    """This function executes `command` and returns it output.
 
    Arguments:
    - `command`: Shell command to be invoked by this function.
    """
 
    child = os.popen(command)
    data = child.read()
    err = child.close()
    if err:
        raise RuntimeError('%s failed w/ exit code %d' % (command, err))
    return data
 
 

Referenced by castorDirExists(), geometryComparison.GeometryComparison.createScript(), validateAlignments.main(), and validateAlignments.ValidationJob.runJob().

getTagsMap()

def helperFunctions.getTagsMap

(

db

)

Definition at line 201 of file helperFunctions.py.

def getTagsMap(db):
    con = conddblib.connect(url = conddblib.make_url(db))
    session = con.session()
    TAG = session.get_dbtype(conddblib.Tag)
    dictionary = {}
    for i in range(0,len(session.query(TAG.object_type).order_by(TAG.name).all())):
        q1 = session.query(TAG.object_type).order_by(TAG.name).all()[i][0]
        q2 = session.query(TAG.name).order_by(TAG.name).all()[i][0]
        dictionary[q1]=q2
 
    return dictionary
 

References python.cmstools.all(), conddblib.connect(), conddblib.make_url(), and FastTimerService_cff.range.

Referenced by alignment.Alignment.__getConditions(), and alignment.Alignment.__testDbExist().

parsecolor()

def helperFunctions.parsecolor

(

color

)

Definition at line 109 of file helperFunctions.py.

def parsecolor(color):
    try: #simplest case: it's an int
        return int(color)
    except ValueError:
        pass
 
    try:   #kRed, kBlue, ...
        color = str(getattr(ROOT, color))
        return int(color)
    except (AttributeError, ValueError):
        pass
 
    if color.count("+") + color.count("-") == 1:  #kRed+5, kGreen-2
        if "+" in color:                          #don't want to deal with nonassociativity of -
            split = color.split("+")
            color1 = parsecolor(split[0])
            color2 = parsecolor(split[1])
            return color1 + color2
 
        if "-" in color:
            split = color.split("-")
            color1 = parsecolor(split[0])
            color2 = parsecolor(split[1])
            return color1 - color2
 
    raise AllInOneError("color has to be an integer, a ROOT constant (kRed, kBlue, ...), or a two-term sum or difference (kGreen-5)!")
 

References createfilelist.int, and str.

Referenced by preexistingValidation.PreexistingValidation.getRepMap().

parsestyle()

def helperFunctions.parsestyle

(

style

)

Definition at line 136 of file helperFunctions.py.

def parsestyle(style):
    try: #simplest case: it's an int
        return int(style)
    except ValueError:
        pass
 
    try: #kStar, kDot, ...
        style = str(getattr(ROOT,style))
        return int(style)
    except (AttributeError, ValueError):
        pass
 
    raise AllInOneError("style has to be an integer or a ROOT constant (kDashed, kStar, ...)!")
 

References createfilelist.int, and str.

Referenced by preexistingValidation.PreexistingValidation.getRepMap().

pythonboolstring()

def helperFunctions.pythonboolstring	(		string,
			name
	)

Takes a string from the configuration file
and makes it into a bool string for a python template

Definition at line 187 of file helperFunctions.py.

def pythonboolstring(string, name):
    """
    Takes a string from the configuration file
    and makes it into a bool string for a python template
    """
    return str(boolfromstring(string, name))
 

References boolfromstring(), and str.

Referenced by primaryVertexValidation.PrimaryVertexValidation.__init__(), offlineValidation.OfflineValidation.__init__(), cppboolstring(), and geometryComparison.GeometryComparison.createConfiguration().

recursivesubclasses()

def helperFunctions.recursivesubclasses

(

cls

)

Definition at line 150 of file helperFunctions.py.

def recursivesubclasses(cls):
    result = [cls]
    for subcls in cls.__subclasses__():
        result += recursivesubclasses(subcls)
    return result
 

Referenced by presentation.ValidationPlots.validationclass().

replaceByMap()

def helperFunctions.replaceByMap	(		target,
			the_map
	)

— Helpers —############################

This function replaces `.oO[key]Oo.` by `the_map[key]` in target.

Arguments:
- `target`: String which contains symbolic tags of the form `.oO[key]Oo.`
- `the_map`: Dictionary which has to contain the `key`s in `target` as keys

Definition at line 13 of file helperFunctions.py.

def replaceByMap(target, the_map):
    """This function replaces `.oO[key]Oo.` by `the_map[key]` in target.
 
    Arguments:
    - `target`: String which contains symbolic tags of the form `.oO[key]Oo.`
    - `the_map`: Dictionary which has to contain the `key`s in `target` as keys
    """
 
    result = target
    for key in the_map:
        lifeSaver = 10e3
        iteration = 0
        while ".oO[" in result and "]Oo." in result:
            for key in the_map:
                try:
                    result = result.replace(".oO["+key+"]Oo.",the_map[key])
                except TypeError:   #try a dict
                    try:
                        for keykey, value in six.iteritems(the_map[key]):
                           result = result.replace(".oO[" + key + "['" + keykey + "']]Oo.", value)
                           result = result.replace(".oO[" + key + '["' + keykey + '"]]Oo.', value)
                    except AttributeError:   #try a list
                        try:
                            for index, value in enumerate(the_map[key]):
                                result = result.replace(".oO[" + key + "[" + str(index) + "]]Oo.", value)
                        except TypeError:
                            raise TypeError("Something is wrong in replaceByMap!  Need a string, dict, or list, but the_map(%s)=%s!"%(repr(key), repr(the_map[key])))
                iteration += 1
            if iteration > lifeSaver:
                problematicLines = ""
                for line in result.splitlines():
                    if  ".oO[" in result and "]Oo." in line:
                        problematicLines += "%s\n"%line
                msg = ("Oh Dear, there seems to be an endless loop in "
                       "replaceByMap!!\n%s\n%s"%(problematicLines, the_map))
                raise AllInOneError(msg)
    return result
 
 

References str.

Referenced by zMuMuValidation.ZMuMuValidation.appendToPlots(), geometryComparison.GeometryComparison.createConfiguration(), genericValidation.GenericValidation.createFiles(), validateAlignments.createMergeScript(), genericValidation.ValidationWithPlots.createPlottingScript(), geometryComparison.GeometryComparison.createScript(), genericValidation.ValidationWithComparison.doComparison(), genericValidation.ParallelValidation.doInitMerge(), genericValidation.ParallelValidation.doMerge(), genericValidation.ValidationWithPlots.doRunPlots(), zMuMuValidation.ZMuMuValidation.filesToCompare(), alignment.Alignment.getConditions(), genericValidation.GenericValidationData.getRepMap(), offlineValidation.OfflineValidation.initMerge(), and plottingOptions.PlottingOptionsTrackSplitting.validsubdets().

replacelast()

def helperFunctions.replacelast	(		string,
			old,
			new,
			count = 1
	)

Replace the last occurances of a string

Definition at line 88 of file helperFunctions.py.

def replacelast(string, old, new, count = 1):
    """Replace the last occurances of a string"""
    return new.join(string.rsplit(old,count))
 

Referenced by addIndex().

Variable Documentation

fileExtensions

list helperFunctions.fileExtensions = ["_cfg.py", ".sh", ".root"]

Definition at line 92 of file helperFunctions.py.