parse task1 for relevant information to computed expected number of e…

…vents in output
franzoni · Aug 22, 2014 · f07b9f2 · f07b9f2
1 parent 83f0fbf
commit f07b9f2
Showing 1 changed file with 32 additions and 12 deletions.
diff --git a/tools/statsMonitoring.py b/tools/statsMonitoring.py
@@ -338,38 +338,58 @@ def get_expected_events_withdict(dict_from_workload):
     rne=dict_from_workload['request']['schema']['RequestNumEvents']
   elif 'RequestSizeEvents' in dict_from_workload['request']['schema']:
     rne=dict_from_workload['request']['schema']['RequestSizeEvents']
+  elif 'Task1' in dict_from_workload['request']['schema'] and 'RequestNumEvents' in dict_from_workload['request']['schema']['Task1']:
+    rne=dict_from_workload['request']['schema']['Task1']['RequestNumEvents']
   else:
     rne=None
 
-  if not 'FilterEfficiency' in dict_from_workload['request']['schema']:
-    f=1.
-  else:
+  if 'FilterEfficiency' in dict_from_workload['request']['schema']:
     f=float(dict_from_workload['request']['schema']['FilterEfficiency'])
-
-  if not 'InputDatasets' in dict_from_workload['request']['schema']:
-    ids=[]
+  elif 'Task1' in dict_from_workload['request']['schema'] and 'FilterEfficiency' in dict_from_workload['request']['schema']['Task1']:
+    f=float(dict_from_workload['request']['schema']['Task1']['FilterEfficiency'])
   else:
+    f=1.
+
+  if 'InputDatasets' in dict_from_workload['request']['schema']:
     try:
       ids=dict_from_workload['request']['schema']['InputDatasets'].split(',')
     except:
       ids=dict_from_workload['request']['schema']['InputDatasets']
+  elif 'Task1' in dict_from_workload['request']['schema'] and  'InputDataset' in dict_from_workload['request']['schema']['Task1']:
+    try:
+      ids=dict_from_workload['request']['schema']['Task1']['InputDataset'].split(',')
+    except:
+      ids=dict_from_workload['request']['schema']['Task1']['InputDataset']
+  else:
+    ids=[]
+
 
 
-  if not 'BlockWhitelist' in dict_from_workload['request']['schema']:
-    bwl=[]
-  else:
+  if 'BlockWhitelist' in dict_from_workload['request']['schema']:
     try:
       bwl=dict_from_workload['request']['schema']['BlockWhitelist'].split(',')
     except:
       bwl=dict_from_workload['request']['schema']['BlockWhitelist']
-
-  if not 'RunWhitelist' in dict_from_workload['request']['schema']:
-    rwl=[]
+  elif 'Task1' in dict_from_workload['request']['schema'] and  'BlockWhitelist' in dict_from_workload['request']['schema']['Task1']:
+    try:
+      bwl=dict_from_workload['request']['schema']['Task1']['BlockWhitelist'].split(',')
+    except:
+      bwl=dict_from_workload['request']['schema']['Task1']['BlockWhitelist']
   else:
+    bwl=[]
+
+  if 'RunWhitelist' in dict_from_workload['request']['schema']:
     try:
       rwl=dict_from_workload['request']['schema']['RunWhitelist'].split(',')
     except:
       rwl=dict_from_workload['request']['schema']['RunWhitelist']
+  elif 'Task1' in dict_from_workload['request']['schema'] and 'RunWhitelist' in dict_from_workload['request']['schema']['Task1']:
+    try:
+      rwl=dict_from_workload['request']['schema']['Task1']['RunWhitelist'].split(',')
+    except:
+      rwl=dict_from_workload['request']['schema']['Task1']['RunWhitelist']
+  else:
+    rwl=[]
 
   return get_expected_events_withinput(rne,
                                        ids,