|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
#!/usr/bin/env python3
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)
|