GoodRunsListTool.py

Go to the documentation of this file.

# Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
 
from __future__ import print_function
 
import os, sys
 
def importModule(moduleName,className,objectName=""):
  if len(objectName)==0: objectName=className
  if len(moduleName)>0:
    modname = '%s.%s' % (moduleName,className)
  else:
    modname = className
  import importlib
  mod = importlib.import_module (modname)
  return getattr (mod, objectName)
 
def getObjectOfClass(moduleName,className=""):
  moduleArr = moduleName.split(".")
  if len(moduleArr)==2: 
    libraryName = moduleArr[0]
  else: libraryName = "GoodRunsLists"
  configModule = moduleArr[len(moduleArr)-1]
  if len(className)==0: className=configModule
 
  try:
    cls = importModule (libraryName,configModule,className)
  except Exception:
    try:
      cls = importModule (configModule,className)
    except Exception:
      print ("Could not import configuration module \'" + configModule + "\'")
      sys.exit(1)
 
  return cls()
 
def createAndVerifyGRL(configModule,prefix=""):
  xmlfile = createGRL(configModule,prefix)
  allOk = verifyGRL(xmlfile)
  return (allOk,xmlfile)
 
def createGRL(configModule,prefix=""):
  from CoolRunQuery.AtlRunQueryLib import AtlRunQuery
  from CoolRunQuery.AtlRunQueryOptions import AtlRunQueryOptions
  from CoolRunQuery.AtlRunQueryParser import ArgumentParser
 
  
  config = getObjectOfClass(configModule)
  config.setPrefix(prefix)
  query  = config.getsearchquery()
  print (">> Calling cmd equivalent of: ")
  print ("%s \"%s\"" % (config.querytool,query))
 
  
  ap = ArgumentParser()
  atlqueryarg = config.querytool + " " + ap.ParseArgument( query )
  (options, args) = AtlRunQueryOptions().parse(atlqueryarg)
 
  #print (atlqueryarg)
  #print (options)
 
  
  AtlRunQuery(options, html="NO", origQuery=query)  # html can be "YES", "NO", "AUTO"
 
  xmlfile = 'data/'+config.listname
  print (">> Good-run list stored as: \'%s\'" % xmlfile)
  return xmlfile
 
def verifyGRL(xmlfile):
  
  CWD = os.getcwd()
  
  from ROOT import gSystem
  os.chdir(CWD)
  
  gSystem.Load('libGoodRunsListsLib')
  from ROOT import Root
  
  reader = Root.TGoodRunsListReader()
  
  reader.SetXMLFile(xmlfile)
  reader.Interpret()
 
  grlcollection = reader.GetMergedGRLCollection()
  print (">> Found %d independent good-run list(s). Now checking content." % len(grlcollection))
 
  allOk = True
 
  
  for idx in range(len(grlcollection)):
    grl        = grlcollection[idx]
    grlName    = grl.GetName()
    grlVersion = grl.GetVersion()
    grlMDSize  = grl.GetMetaDataSize()
    grlIsEmpty = grl.IsEmpty()
    if len(grlName)==0: 
      print (">> ERROR: good run list %d has no name." % (idx))
      allOk = False
    if len(grlVersion)==0:
      print (">> ERROR: good run list %d, with name \'%s\', has no version." % (idx,grlName))
      allOk = False
    if grlMDSize==0:
      print (">> ERROR: good run list %d, with name \'%s\', has no metadata." % (idx,grlName))
      allOk = False
    if grlIsEmpty: 
      print (">> ERROR: good run list %d, with name \'%s\', has no run(s) and lumi-block range(s)." % (idx,grlName))
      allOk = False
 
  if allOk:
    print (">> Content okay.")
 
  return allOk