def extendWorkflow(workflow,
user,
group,
verbose=False,
events=None,
firstlumi=None):
if events is None:
events = getMissingEvents(workflow)
events = int(events)
if firstlumi is None:
#get the last lumi of the dataset
dataset = reqMgrClient.outputdatasetsWorkflow(url, workflow).pop()
lastLumi = dbs3Client.getMaxLumi(dataset)
firstlumi = lastLumi
firstlumi = int(firstlumi)
# Get info about the workflow to be cloned
cache = reqMgrClient.getWorkflowInfo(url, workflow)
schema = modifySchema(cache, workflow, user, group, events, firstlumi,
None)
if verbose:
pprint(schema)
print 'Submitting workflow'
# Submit cloned workflow to ReqMgr
response = reqMgrClient.submitWorkflow(url, schema)
if verbose:
print "RESPONSE", response
#find the workflow name in response
m = re.search("details\/(.*)\'", response)
if m:
newWorkflow = m.group(1)
print 'Cloned workflow: ' + newWorkflow
print 'Extended with', events, 'events'
print response
# Move the request to Assignment-approved
print 'Approve request response:'
data = reqMgrClient.setWorkflowApproved(url, newWorkflow)
print data
else:
print response
def extendWorkflow(workflow, user, group, verbose=False, events=None, firstlumi=None):
if events is None:
events = getMissingEvents(workflow)
events = int(events)
if firstlumi is None:
#get the last lumi of the dataset
dataset = reqMgrClient.outputdatasetsWorkflow(url, workflow).pop()
lastLumi = dbs3Client.getMaxLumi(dataset)
firstlumi = lastLumi
firstlumi = int(firstlumi)
# Get info about the workflow to be cloned
helper = reqMgrClient.retrieveSchema(workflow)
schema = modifySchema(helper, workflow, user, group, events, firstlumi)
schema['OriginalRequestName'] = workflow
if verbose:
pprint(schema)
print 'Submitting workflow'
# Sumbit cloned workflow to ReqMgr
response = reqMgrClient.submitWorkflow(url,schema)
if verbose:
print "RESPONSE", response
#find the workflow name in response
m = re.search("details\/(.*)\'",response)
if m:
newWorkflow = m.group(1)
print 'Cloned workflow: '+newWorkflow
print 'Extended with', events, 'events'
print response
# Move the request to Assignment-approved
print 'Approve request response:'
data = reqMgrClient.setWorkflowApproved(url, newWorkflow)
print data
else:
print response
pass
def modifySchema(helper, workflow, user, group, events):
"""
Adapts schema to right parameters
"""
result = {}
#pprint.pprint(helper.data.request.schema.dictionary_())
for key, value in helper.data.request.schema.dictionary_().items():
#previous versions of tags
if key == 'ProcConfigCacheID':
result['ConfigCacheID'] = value
elif key == 'RequestSizeEvents':
result['RequestSizeEvents'] = value
#requestor info
elif key == 'Requestor':
result['Requestor'] = user
elif key == 'Group':
result['Group'] = group
#if emtpy
elif key in ["RunWhitelist", "RunBlacklist", "BlockWhitelist", "BlockBlacklist"] and not value:
result[key]=[]
#replace old DBS2 URL
elif value == "http://cmsdbsprod.cern.ch/cms_dbs_prod_global/servlet/DBSServlet":
result[key] = 'https://cmsweb.cern.ch/dbs/prod/global/DBSReader'
#copy the right LFN base
elif key == 'MergedLFNBase':
result['MergedLFNBase'] = helper.getMergedLFNBase()
#TODO deleting timeout so they will move to running-close as soon as they can
#elif key == 'OpenRunningTimeout':
#delete entry
# continue
#skip empty entries
elif not value:
continue
elif value != None:
result[key] = value
#extend workflow so it will safely start outside of the boundary
RequestNumEvents = int(result['RequestNumEvents'])
FirstEvent = int(result['FirstEvent'])
FirstLumi = int(result['FirstLumi'])
EventsPerLumi = int(result['EventsPerLumi'])
FilterEfficiency = float(result['FilterEfficiency'])
#FirstEvent_NEW > FirstEvent + RequestNumEvents
#the fist event needs to be oustide the range
result['FirstEvent'] = FirstEvent + RequestNumEvents + DELTA_EVENTS
#FirstLumi_NEW > FirstLumi + RequestNumEvents/events_per_job/filterEff
# same for the first lumi, needs to be after the last lumi
#get the last lumi of the dataset
dataset = reqMgrClient.outputdatasetsWorkflow(url, workflow).pop()
LastLumi = dbs3Client.getMaxLumi(dataset)
result['FirstLumi'] = LastLumi + DELTA_LUMIS
#only the desired events
result['RequestNumEvents'] = events
#prepend EXT_ to recognize as extension
result["RequestString"] = 'EXT_'+result["RequestString"]
#check MonteCarlo
if result['RequestType']=='MonteCarlo':
#check assigning parameters
#seek for events per job on helper
splitting = helper.listJobSplittingParametersByTask()
eventsPerJob = 120000
eventsPerLumi = 100000
for k, v in splitting.items():
print k,":",v
if k.endswith('/Production'):
if 'events_per_job' in v:
eventsPerJob = v['events_per_job']
elif 'events_per_lumi' in v:
eventsPerLumi = v['events_per_lumi']
result['EventsPerJob'] = eventsPerJob
#result['EventsPerLumi'] = eventsPerLumi
#Merged LFN
if 'MergedLFNBase' not in result:
result['MergedLFNBase'] = helper.getMergedLFNBase()
#update information from reqMgr
# Add AcquisitionEra, ProcessingString and ProcessingVersion
result["ProcessingString"] = helper.getProcessingString()
result["AcquisitionEra"] = helper.getAcquisitionEra()
#try to parse processing version as an integer, if don't, assign 1
try:
result["ProcessingVersion"] = int(helper.getProcessingVersion())
except ValueError:
result["ProcessingVersion"] = 1
return result
def modifySchema(helper, workflow, user, group, events):
"""
Adapts schema to right parameters
"""
result = {}
# Add AcquisitionEra, ProcessingString and ProcessingVersion
result["ProcessingString"] = helper.getProcessingString()
result["ProcessingVersion"] = helper.getProcessingVersion()
result["AcquisitionEra"] = helper.getAcquisitionEra()
for key, value in helper.data.request.schema.dictionary_().items():
#previous versions of tags
if key == 'ProcConfigCacheID':
result['ConfigCacheID'] = value
elif key == 'RequestSizeEvents':
result['RequestSizeEvents'] = value
#requestor info
elif key == 'Requestor':
result['Requestor'] = user
elif key == 'Group':
result['Group'] = group
#preppend EXT to recognize as an extension
elif key == 'RequestString':
result['RequestString'] = 'EXT_'+str(value)
#if emtpy
elif key in ["RunWhitelist", "RunBlacklist", "BlockWhitelist", "BlockBlacklist"] and not value:
result[key]=[]
#skip empty entries
elif not value:
continue
elif value != None:
result[key] = value
#extend workflow so it will safely start outside of the boundary
RequestNumEvents = int(result['RequestNumEvents'])
FirstEvent = int(result['FirstEvent'])
FirstLumi = int(result['FirstLumi'])
EventsPerLumi = int(result['EventsPerLumi'])
FilterEfficiency = float(result['FilterEfficiency'])
#FirstEvent_NEW > FirstEvent + RequestNumEvents
#the fist event needs to be oustide the range
result['FirstEvent'] = FirstEvent + RequestNumEvents + DELTA_EVENTS
#FirstLumi_NEW > FirstLumi + RequestNumEvents/events_per_job/filterEff
# same for the first lumi, needs to be after the last lumi
"""
result['FirstLumi'] = int(FirstLumi
+ math.ceil( RequestNumEvents / float(EventsPerLumi) / FilterEfficiency )
+ DELTA_LUMIS / FilterEfficiency )
"""
#get the last lumi of the dataset
dataset = reqMgrClient.outputdatasetsWorkflow(url, workflow).pop()
LastLumi = dbs3Client.getMaxLumi(dataset)
result['FirstLumi'] = LastLumi + DELTA_LUMIS
#only the desired events
result['RequestNumEvents'] = events
if 'LumisPerJob' not in result and result['RequestType']=='MonteCarlo':
#seek for lumis per job on helper
splitting = helper.listJobSplittingParametersByTask()
lumisPerJob = 300
for k, v in splitting.items():
if k.endswith('/Production'):
if 'lumis_per_job' in v:
lumisPerJob = v['lumis_per_job']
result['LumisPerJob'] = lumisPerJob
#TODO do this always?
if 'EventsPerJob' not in result and result['RequestType']=='MonteCarlo':
#seek for events per job on helper
splitting = helper.listJobSplittingParametersByTask()
eventsPerJob = 120000
for k, v in splitting.items():
if k.endswith('/Production'):
if 'events_per_job' in v:
eventsPerJob = v['events_per_job']
result['EventsPerJob'] = eventsPerJob
if 'MergedLFNBase' not in result:
result['MergedLFNBase'] = helper.getMergedLFNBase()
return result