contentValuesCheck.py

Go to the documentation of this file.

#!/usr/bin/env python3

from __future__ import print_function
from contentValuesLib import *

class OptionParser(optparse.OptionParser):
  """ Option parser class """
  def __init__(self):
    optparse.OptionParser.__init__(self, usage="%prog [options] root_file ...", version="%prog 0.0.1", conflict_handler="resolve")
    self.add_option("--silent", "-s", action="store_true", dest="silent", default=False, help="silent mode: specify return code and exit")
    self.add_option("--subsystem", "-c", action="store", type="string", dest="subsystem", default=None, help="Specify test subsystem")

if __name__ == "__main__":
  
  # Create option parser and get options/arguments
  optManager  = OptionParser()
  (opts, args) = optManager.parse_args()
  opts = opts.__dict__

  # Check if at least one root file defined (can be many!)
  if len(args) == 0:
    print("At least one ROOT file must be priovided, use --help for hit")
    sys.exit(1)

  # Check if all files exists and are accessible
  for rfile in args:
    try:
      os.stat(rfile)
    except:
      print("File [", rfile, "] not exists or is not accessible?")
      sys.exit(2)

  ss = opts['subsystem']

  # Lets extract values from files one-by-one, construct hashmap and check values
  for rfile in args:

    (run_number, values) = getSummaryValues(file_name = rfile, shift_type = None, translate = False, filters = None)

    if values == None or len(values) == 0:
      print("No content summary values found. Skipping file: %s" % rfile)
      continue

    messages = []
    for sub in SUBSYSTEMS.keys():

      if not ss == None and not sub == ss:
        continue

      if sub not in values:
        messages.append("%s: missing subsystem!" % sub)
        continue

      skeys = {}
      sfolders = []

      for folder in FOLDERS.keys():

        if folder not in values[sub]:
          messages.append("%s: missing folder EventInfo/%s" % (sub, folder))
          continue

        if len(values[sub][folder]) == 0:
          messages.append("%s: empty folder EventInfo/%s" % (sub, FOLDERS[folder][1]))
          continue

        sfolders.append(folder)

        if 'Summary' not in values[sub][folder]:
          messages.append("%s: missing summary value EventInfo/%s" % (sub, FOLDERS[folder][1]))

        for key in values[sub][folder].keys():
          if key == 'Summary':
            continue
          if key not in skeys:
            skeys[key] = []
          skeys[key].append(folder)
          
      for key in skeys:
        nfound = []
        for folder in sfolders:
          if skeys[key].count(folder) == 0: nfound.append(folder)
        if len(nfound) > 0:
          messages.append("%s: value (%s)/%s not found in (%s)" % (sub, ','.join(skeys[key]), key, ','.join(nfound)))

    if not opts['silent']:
      for message in sorted(messages):  print(message)          
      print("%d errors found" % len(messages))

    if len(messages) > 0: sys.exit(1)

  sys.exit(0)