contentValuesToRR.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("--shift", "-s", action="store", type="choice", dest="shift", choices=("online","offline"), help="specify shift type: online or offline values allowed. Default is offline.")
    self.add_option("--url", action="store", type="string", dest="url", default=SERVER_URL, help="specify RR XML-RPC server URL. Default is " + SERVER_URL)
    self.add_option("--dataset", "-t", action="store", type="string", dest="dataset", default=None, help="explicitly specify dataset name. If not set then script \
      (1) for offline shift will try to get it from the filename or (2) for online shift will set it to " + ONLINE_DATASET)
    self.add_option("--debug", "-d", action="store_true", dest="debug", default=False, help="print values and exit. Do not write to RR")
    self.add_option("--filter", "-f", action="store", type="string", dest="filter", default=None, help="Specify filters in the form \"('subsystem','folder','value')\" \
      in regexp expression. Default is None and this takes all the subsystems, all folders and allvalues")
 
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 shift type is provided (mandatory!)
  if not opts['shift']:
    print("Shift type must be provided, use --help for hit")
    sys.exit(1)
 
  # Get default dataset name (optional)
  default_dataset = opts['dataset']
  if default_dataset == None and opts['shift'] == 'online':
    default_dataset = ONLINE_DATASET
 
  # 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)
 
  # Take the filter
  filter = checkFilter(opts['filter'])
 
  server = xmlrpclib.ServerProxy(opts['url'])
 
  # Lets extract values from files one-by-one, construct hashmap and submit to
  # defined XML-RPC url 
  for rfile in args:
 
    (run_number, values) = getSummaryValues(file_name = rfile, translate = True, filters = filter)
 
    if default_dataset == None:
      dataset = getDatasetName(rfile)
    else:
      dataset = default_dataset
 
    if run_number == None:
      print("Run number does not determined. Skipping file: %s" % rfile)
      continue
    
    if dataset == None:
      print("Dataset name do not determined. Skipping file: %s" % rfile)
      continue
 
    if values == None or len(values) == 0:
      print("No content summary values found. Skipping file: %s" % rfile)
      continue
 
    try:
      if opts['debug']:
        print("Run number: %d" % run_number)
        print("Dataset: %s" % dataset)
        print("Data: ", values)
      else:
        result = server.SummaryValuesWriter.write(run_number, dataset, json.dumps(values))
        print("RR: %d rows modified for run# %d dataset %s" % (result, run_number, dataset))
    except xmlrpclib.Error as errstring:
      print("ERROR", errstring)
      sys.exit(3)
  
  sys.exit(0)