update das-selected-lumis script to match the new das server format

cms-sw · Aug 29, 2017 · a67922f · a67922f
1 parent 53f615e
commit a67922f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/Configuration/PyReleaseValidation/scripts/das-selected-lumis.py b/Configuration/PyReleaseValidation/scripts/das-selected-lumis.py
@@ -29,8 +29,14 @@ def check_lumi_ranges(given_lumi_list , sub_range):
 
 def process_lumi(data):
   for lumi_info in data:
-    lumi_rang = lumi_info['lumi'][0]['number']
-    lumi_file = lumi_info['file'][0]['name']
+    if type(lumi_info['lumi']) is list:
+      lumi_nums = lumi_info['lumi'][0]['number']
+      lumi_file = lumi_info['file'][0]['name']
+    else:
+      lumi_nums = lumi_info['lumi']['number']
+      lumi_file = lumi_info['file']['name']
+    if not type(lumi_nums[0]) is list: lumi_rang = [ [n,n] for n in lumi_nums ]
+    else: lumi_rang = lumi_nums
     for sub_list in lumi_rang:
       if check_lumi_ranges(input_range,tuple(sub_list)):
         output_files_list.append(lumi_file)