FWCore/PythonUtilities/scripts/mergeJSON.py - master - githubmirror/cmssw

FWCore/PythonUtilities/scripts/mergeJSON.py

49 строк2 KB

Kevin Pedro

remove __future__ imports

19 окт 2023, 11:36

19 окт 2023, 11:369429d62

#!/usr/bin/env python3

import sys
from argparse import ArgumentParser
import re
from FWCore.PythonUtilities.LumiList import LumiList

def filterRuns (lumiList, minRun, maxRun):
    allRuns = lumiList.getRuns()
    runsToRemove = []
    for run in allRuns:
        if minRun and int(run) < minRun:
            runsToRemove.append (run)
        if maxRun and int(run) > maxRun:
            runsToRemove.append (run)
    lumiList.removeRuns (runsToRemove)

if __name__ == '__main__':
    
    parser = ArgumentParser()
    parser.add_argument('--output', dest='output', type=str,
                        help='Save output to file OUTPUT')
    parser.add_argument("alpha_json", metavar="alpha.json[:142300-145900]", type=str, nargs='+')
    # required parameters
    options = parser.parse_args()

    minMaxRE = re.compile (r'(\S+):(\d+)-(\d*)')

    finalList = LumiList()
    for filename in options.alpha_json:
        minRun = maxRun = 0
        match = minMaxRE.search (filename)
        if match:
            filename   =      match.group(1)
            minRun     = int( match.group(2) )
            try:
                maxRun = int( match.group(3) )
            except:
                pass
            if maxRun and minRun > maxRun:
                raise RuntimeError("Minimum value (%d) is greater than maximum value (%d) for file '%s'" % (minRun, maxRun, filename))
        localList = LumiList (filename = filename)
        filterRuns (localList, minRun, maxRun)
        finalList = finalList | localList

    if options.output:
        finalList.writeJSON (options.output)
    else:
        print(finalList)