CMS 3D CMS Logo

MatrixInjector.py
Go to the documentation of this file.
1 from __future__ import print_function
2 import sys
3 import json
4 import os
5 import copy
6 import multiprocessing
7 import time
8 
10  if opt.show:
11  print('Not injecting to wmagent in --show mode. Need to run the worklfows.')
12  sys.exit(-1)
13  if opt.wmcontrol=='init':
14  #init means it'll be in test mode
15  opt.nProcs=0
16  if opt.wmcontrol=='test':
17  #means the wf were created already, and we just dryRun it.
18  opt.dryRun=True
19  if opt.wmcontrol=='submit' and opt.nProcs==0:
20  print('Not injecting to wmagent in -j 0 mode. Need to run the worklfows.')
21  sys.exit(-1)
22  if opt.wmcontrol=='force':
23  print("This is an expert setting, you'd better know what you're doing")
24  opt.dryRun=True
25 
26 def upload_to_couch_oneArg(arguments):
27  from modules.wma import upload_to_couch
28  (filePath,labelInCouch,user,group,where) = arguments
29  cacheId=upload_to_couch(filePath,
30  labelInCouch,
31  user,
32  group,
33  test_mode=False,
34  url=where)
35  return cacheId
36 
37 
39 
40  def __init__(self,opt,mode='init',options=''):
41  self.count=1040
42 
43  self.dqmgui=None
44  self.wmagent=None
45  for k in options.split(','):
46  if k.startswith('dqm:'):
47  self.dqmgui=k.split(':',1)[-1]
48  elif k.startswith('wma:'):
49  self.wmagent=k.split(':',1)[-1]
50 
51  self.testMode=((mode!='submit') and (mode!='force'))
52  self.version =1
53  self.keep = opt.keep
54  self.memoryOffset = opt.memoryOffset
55  self.memPerCore = opt.memPerCore
56  self.batchName = ''
57  self.batchTime = str(int(time.time()))
58  if(opt.batchName):
59  self.batchName = '__'+opt.batchName+'-'+self.batchTime
60 
61  #wagemt stuff
62  if not self.wmagent:
63  self.wmagent=os.getenv('WMAGENT_REQMGR')
64  if not self.wmagent:
65  if not opt.testbed :
66  self.wmagent = 'cmsweb.cern.ch'
67  self.DbsUrl = "https://"+self.wmagent+"/dbs/prod/global/DBSReader"
68  else :
69  self.wmagent = 'cmsweb-testbed.cern.ch'
70  self.DbsUrl = "https://"+self.wmagent+"/dbs/int/global/DBSReader"
71 
72  if not self.dqmgui:
73  self.dqmgui="https://cmsweb.cern.ch/dqm/relval"
74  #couch stuff
75  self.couch = 'https://'+self.wmagent+'/couchdb'
76 # self.couchDB = 'reqmgr_config_cache'
77  self.couchCache={} # so that we do not upload like crazy, and recyle cfgs
78  self.user = os.getenv('USER')
79  self.group = 'ppd'
80  self.label = 'RelValSet_'+os.getenv('CMSSW_VERSION').replace('-','')+'_v'+str(self.version)
81  self.speciallabel=''
82  if opt.label:
83  self.speciallabel= '_'+opt.label
84 
85 
86  if not os.getenv('WMCORE_ROOT'):
87  print('\n\twmclient is not setup properly. Will not be able to upload or submit requests.\n')
88  if not self.testMode:
89  print('\n\t QUIT\n')
90  sys.exit(-18)
91  else:
92  print('\n\tFound wmclient\n')
93 
94  self.defaultChain={
95  "RequestType" : "TaskChain", #this is how we handle relvals
96  "SubRequestType" : "RelVal", #this is how we handle relvals, now that TaskChain is also used for central MC production
97  "RequestPriority": 500000,
98  "Requestor": self.user, #Person responsible
99  "Group": self.group, #group for the request
100  "CMSSWVersion": os.getenv('CMSSW_VERSION'), #CMSSW Version (used for all tasks in chain)
101  "Campaign": os.getenv('CMSSW_VERSION'), # = AcquisitionEra, will be reset later to the one of first task, will both be the CMSSW_VERSION
102  "ScramArch": os.getenv('SCRAM_ARCH'), #Scram Arch (used for all tasks in chain)
103  "ProcessingVersion": self.version, #Processing Version (used for all tasks in chain)
104  "GlobalTag": None, #Global Tag (overridden per task)
105  "ConfigCacheUrl": self.couch, #URL of CouchDB containing Config Cache
106  "DbsUrl": self.DbsUrl,
107  #- Will contain all configs for all Tasks
108  #"SiteWhitelist" : ["T2_CH_CERN", "T1_US_FNAL"], #Site whitelist
109  "TaskChain" : None, #Define number of tasks in chain.
110  "nowmTasklist" : [], #a list of tasks as we put them in
111  "Multicore" : 1, # do not set multicore for the whole chain
112  "Memory" : 3000,
113  "SizePerEvent" : 1234,
114  "TimePerEvent" : 10,
115  "PrepID": os.getenv('CMSSW_VERSION')
116  }
117 
119  "EnableHarvesting" : "True",
120  "DQMUploadUrl" : self.dqmgui,
121  "DQMConfigCacheID" : None,
122  "Multicore" : 1 # hardcode Multicore to be 1 for Harvest
123  }
124 
126  "TaskName" : None, #Task Name
127  "ConfigCacheID" : None, #Generator Config id
128  "GlobalTag": None,
129  "SplittingAlgo" : "EventBased", #Splitting Algorithm
130  "EventsPerJob" : None, #Size of jobs in terms of splitting algorithm
131  "RequestNumEvents" : None, #Total number of events to generate
132  "Seeding" : "AutomaticSeeding", #Random seeding method
133  "PrimaryDataset" : None, #Primary Dataset to be created
134  "nowmIO": {},
135  "Multicore" : opt.nThreads, # this is the per-taskchain Multicore; it's the default assigned to a task if it has no value specified
136  "KeepOutput" : False
137  }
139  "TaskName" : "DigiHLT", #Task Name
140  "ConfigCacheID" : None, #Processing Config id
141  "GlobalTag": None,
142  "InputDataset" : None, #Input Dataset to be processed
143  "SplittingAlgo" : "LumiBased", #Splitting Algorithm
144  "LumisPerJob" : 10, #Size of jobs in terms of splitting algorithm
145  "nowmIO": {},
146  "Multicore" : opt.nThreads, # this is the per-taskchain Multicore; it's the default assigned to a task if it has no value specified
147  "KeepOutput" : False
148  }
149  self.defaultTask={
150  "TaskName" : None, #Task Name
151  "InputTask" : None, #Input Task Name (Task Name field of a previous Task entry)
152  "InputFromOutputModule" : None, #OutputModule name in the input task that will provide files to process
153  "ConfigCacheID" : None, #Processing Config id
154  "GlobalTag": None,
155  "SplittingAlgo" : "LumiBased", #Splitting Algorithm
156  "LumisPerJob" : 10, #Size of jobs in terms of splitting algorithm
157  "nowmIO": {},
158  "Multicore" : opt.nThreads, # this is the per-taskchain Multicore; it's the default assigned to a task if it has no value specified
159  "KeepOutput" : False
160  }
161 
162  self.chainDicts={}
163 
164 
165  def prepare(self,mReader, directories, mode='init'):
166  try:
167  #from Configuration.PyReleaseValidation.relval_steps import wmsplit
168  wmsplit = {}
169  wmsplit['DIGIHI']=5
170  wmsplit['RECOHI']=5
171  wmsplit['HLTD']=5
172  wmsplit['RECODreHLT']=2
173  wmsplit['DIGIPU']=4
174  wmsplit['DIGIPU1']=4
175  wmsplit['RECOPU1']=1
176  wmsplit['DIGIUP15_PU50']=1
177  wmsplit['RECOUP15_PU50']=1
178  wmsplit['DIGIUP15_PU25']=1
179  wmsplit['RECOUP15_PU25']=1
180  wmsplit['DIGIUP15_PU25HS']=1
181  wmsplit['RECOUP15_PU25HS']=1
182  wmsplit['DIGIHIMIX']=5
183  wmsplit['RECOHIMIX']=5
184  wmsplit['RECODSplit']=1
185  wmsplit['SingleMuPt10_UP15_ID']=1
186  wmsplit['DIGIUP15_ID']=1
187  wmsplit['RECOUP15_ID']=1
188  wmsplit['TTbar_13_ID']=1
189  wmsplit['SingleMuPt10FS_ID']=1
190  wmsplit['TTbarFS_ID']=1
191  wmsplit['RECODR2_50nsreHLT']=5
192  wmsplit['RECODR2_25nsreHLT']=5
193  wmsplit['RECODR2_2016reHLT']=5
194  wmsplit['RECODR2_50nsreHLT_HIPM']=5
195  wmsplit['RECODR2_25nsreHLT_HIPM']=5
196  wmsplit['RECODR2_2016reHLT_HIPM']=1
197  wmsplit['RECODR2_2016reHLT_skimSingleMu']=1
198  wmsplit['RECODR2_2016reHLT_skimDoubleEG']=1
199  wmsplit['RECODR2_2016reHLT_skimMuonEG']=1
200  wmsplit['RECODR2_2016reHLT_skimJetHT']=1
201  wmsplit['RECODR2_2016reHLT_skimMET']=1
202  wmsplit['RECODR2_2016reHLT_skimSinglePh']=1
203  wmsplit['RECODR2_2016reHLT_skimMuOnia']=1
204  wmsplit['RECODR2_2016reHLT_skimSingleMu_HIPM']=1
205  wmsplit['RECODR2_2016reHLT_skimDoubleEG_HIPM']=1
206  wmsplit['RECODR2_2016reHLT_skimMuonEG_HIPM']=1
207  wmsplit['RECODR2_2016reHLT_skimJetHT_HIPM']=1
208  wmsplit['RECODR2_2016reHLT_skimMET_HIPM']=1
209  wmsplit['RECODR2_2016reHLT_skimSinglePh_HIPM']=1
210  wmsplit['RECODR2_2016reHLT_skimMuOnia_HIPM']=1
211  wmsplit['RECODR2_2017reHLT_Prompt']=1
212  wmsplit['RECODR2_2017reHLT_skimSingleMu_Prompt_Lumi']=1
213  wmsplit['RECODR2_2017reHLT_skimDoubleEG_Prompt']=1
214  wmsplit['RECODR2_2017reHLT_skimMET_Prompt']=1
215  wmsplit['RECODR2_2017reHLT_skimMuOnia_Prompt']=1
216  wmsplit['RECODR2_2017reHLT_Prompt_L1TEgDQM']=1
217  wmsplit['RECODR2_2018reHLT_Prompt']=1
218  wmsplit['RECODR2_2018reHLT_skimSingleMu_Prompt_Lumi']=1
219  wmsplit['RECODR2_2018reHLT_skimDoubleEG_Prompt']=1
220  wmsplit['RECODR2_2018reHLT_skimJetHT_Prompt']=1
221  wmsplit['RECODR2_2018reHLT_skimMET_Prompt']=1
222  wmsplit['RECODR2_2018reHLT_skimMuOnia_Prompt']=1
223  wmsplit['RECODR2_2018reHLT_skimEGamma_Prompt_L1TEgDQM']=1
224  wmsplit['RECODR2_2018reHLT_skimMuonEG_Prompt']=1
225  wmsplit['RECODR2_2018reHLT_skimCharmonium_Prompt']=1
226  wmsplit['RECODR2_2018reHLT_skimJetHT_Prompt_HEfail']=1
227  wmsplit['RECODR2_2018reHLT_skimJetHT_Prompt_BadHcalMitig']=1
228  wmsplit['HLTDR2_50ns']=1
229  wmsplit['HLTDR2_25ns']=1
230  wmsplit['HLTDR2_2016']=1
231  wmsplit['HLTDR2_2017']=1
232  wmsplit['HLTDR2_2018']=1
233  wmsplit['HLTDR2_2018_BadHcalMitig']=1
234  wmsplit['Hadronizer']=1
235  wmsplit['DIGIUP15']=1
236  wmsplit['RECOUP15']=1
237  wmsplit['RECOAODUP15']=5
238  wmsplit['DBLMINIAODMCUP15NODQM']=5
239  wmsplit['DigiFull']=5
240  wmsplit['RecoFull']=5
241  wmsplit['DigiFullPU']=1
242  wmsplit['RecoFullPU']=1
243  wmsplit['RECOHID11']=1
244  wmsplit['DigiFullTriggerPU_2023D17PU'] = 1
245  wmsplit['RecoFullGlobalPU_2023D17PU']=1
246  wmsplit['DIGIUP17']=1
247  wmsplit['RECOUP17']=1
248  wmsplit['DIGIUP17_PU25']=1
249  wmsplit['RECOUP17_PU25']=1
250  wmsplit['DIGICOS_UP17']=1
251  wmsplit['RECOCOS_UP17']=1
252  wmsplit['HYBRIDRepackHI2015VR']=1
253  wmsplit['HYBRIDZSHI2015']=1
254  wmsplit['RECOHID15']=1
255 
256  #import pprint
257  #pprint.pprint(wmsplit)
258  except:
259  print("Not set up for step splitting")
260  wmsplit={}
261 
262  acqEra=False
263  for (n,dir) in directories.items():
264  chainDict=copy.deepcopy(self.defaultChain)
265  print("inspecting",dir)
266  nextHasDSInput=None
267  for (x,s) in mReader.workFlowSteps.items():
268  #x has the format (num, prefix)
269  #s has the format (num, name, commands, stepList)
270  if x[0]==n:
271  #print "found",n,s[3]
272  #chainDict['RequestString']='RV'+chainDict['CMSSWVersion']+s[1].split('+')[0]
273  index=0
274  splitForThisWf=None
275  thisLabel=self.speciallabel
276  #if 'HARVESTGEN' in s[3]:
277  if len( [step for step in s[3] if "HARVESTGEN" in step] )>0:
278  chainDict['TimePerEvent']=0.01
279  thisLabel=thisLabel+"_gen"
280  # for double miniAOD test
281  if len( [step for step in s[3] if "DBLMINIAODMCUP15NODQM" in step] )>0:
282  thisLabel=thisLabel+"_dblMiniAOD"
283  processStrPrefix=''
284  setPrimaryDs=None
285  nanoedmGT=''
286  for step in s[3]:
287 
288  if 'INPUT' in step or (not isinstance(s[2][index],str)):
289  nextHasDSInput=s[2][index]
290 
291  else:
292 
293  if (index==0):
294  #first step and not input -> gen part
295  chainDict['nowmTasklist'].append(copy.deepcopy(self.defaultScratch))
296  try:
297  chainDict['nowmTasklist'][-1]['nowmIO']=json.loads(open('%s/%s.io'%(dir,step)).read())
298  except:
299  print("Failed to find",'%s/%s.io'%(dir,step),".The workflows were probably not run on cfg not created")
300  return -15
301 
302  chainDict['nowmTasklist'][-1]['PrimaryDataset']='RelVal'+s[1].split('+')[0]
303  if not '--relval' in s[2][index]:
304  print('Impossible to create task from scratch without splitting information with --relval')
305  return -12
306  else:
307  arg=s[2][index].split()
308  ns=map(int,arg[arg.index('--relval')+1].split(','))
309  chainDict['nowmTasklist'][-1]['RequestNumEvents'] = ns[0]
310  chainDict['nowmTasklist'][-1]['EventsPerJob'] = ns[1]
311  if 'FASTSIM' in s[2][index] or '--fast' in s[2][index]:
312  thisLabel+='_FastSim'
313  if 'lhe' in s[2][index] in s[2][index]:
314  chainDict['nowmTasklist'][-1]['LheInputFiles'] =True
315 
316  elif nextHasDSInput:
317  chainDict['nowmTasklist'].append(copy.deepcopy(self.defaultInput))
318  try:
319  chainDict['nowmTasklist'][-1]['nowmIO']=json.loads(open('%s/%s.io'%(dir,step)).read())
320  except:
321  print("Failed to find",'%s/%s.io'%(dir,step),".The workflows were probably not run on cfg not created")
322  return -15
323  chainDict['nowmTasklist'][-1]['InputDataset']=nextHasDSInput.dataSet
324  if ('DQMHLTonRAWAOD' in step) :
325  chainDict['nowmTasklist'][-1]['IncludeParents']=True
326  splitForThisWf=nextHasDSInput.split
327  chainDict['nowmTasklist'][-1]['LumisPerJob']=splitForThisWf
328  if step in wmsplit:
329  chainDict['nowmTasklist'][-1]['LumisPerJob']=wmsplit[step]
330  # get the run numbers or #events
331  if len(nextHasDSInput.run):
332  chainDict['nowmTasklist'][-1]['RunWhitelist']=nextHasDSInput.run
333  if len(nextHasDSInput.ls):
334  chainDict['nowmTasklist'][-1]['LumiList']=nextHasDSInput.ls
335  #print "what is s",s[2][index]
336  if '--data' in s[2][index] and nextHasDSInput.label:
337  thisLabel+='_RelVal_%s'%nextHasDSInput.label
338  if 'filter' in chainDict['nowmTasklist'][-1]['nowmIO']:
339  print("This has an input DS and a filter sequence: very likely to be the PyQuen sample")
340  processStrPrefix='PU_'
341  setPrimaryDs = 'RelVal'+s[1].split('+')[0]
342  if setPrimaryDs:
343  chainDict['nowmTasklist'][-1]['PrimaryDataset']=setPrimaryDs
344  nextHasDSInput=None
345  else:
346  #not first step and no inputDS
347  chainDict['nowmTasklist'].append(copy.deepcopy(self.defaultTask))
348  try:
349  chainDict['nowmTasklist'][-1]['nowmIO']=json.loads(open('%s/%s.io'%(dir,step)).read())
350  except:
351  print("Failed to find",'%s/%s.io'%(dir,step),".The workflows were probably not run on cfg not created")
352  return -15
353  if splitForThisWf:
354  chainDict['nowmTasklist'][-1]['LumisPerJob']=splitForThisWf
355  if step in wmsplit:
356  chainDict['nowmTasklist'][-1]['LumisPerJob']=wmsplit[step]
357 
358  # change LumisPerJob for Hadronizer steps.
359  if 'Hadronizer' in step:
360  chainDict['nowmTasklist'][-1]['LumisPerJob']=wmsplit['Hadronizer']
361 
362  #print step
363  chainDict['nowmTasklist'][-1]['TaskName']=step
364  if setPrimaryDs:
365  chainDict['nowmTasklist'][-1]['PrimaryDataset']=setPrimaryDs
366  chainDict['nowmTasklist'][-1]['ConfigCacheID']='%s/%s.py'%(dir,step)
367  chainDict['nowmTasklist'][-1]['GlobalTag']=chainDict['nowmTasklist'][-1]['nowmIO']['GT'] # copy to the proper parameter name
368  chainDict['GlobalTag']=chainDict['nowmTasklist'][-1]['nowmIO']['GT'] #set in general to the last one of the chain
369  if 'NANOEDM' in step :
370  nanoedmGT = chainDict['nowmTasklist'][-1]['nowmIO']['GT']
371  if 'NANOMERGE' in step :
372  chainDict['GlobalTag'] = nanoedmGT
373  if 'pileup' in chainDict['nowmTasklist'][-1]['nowmIO']:
374  chainDict['nowmTasklist'][-1]['MCPileup']=chainDict['nowmTasklist'][-1]['nowmIO']['pileup']
375  if '--pileup ' in s[2][index]: # catch --pileup (scenarion) and not --pileup_ (dataset to be mixed) => works also making PRE-MIXed dataset
376  processStrPrefix='PU_' # take care of pu overlay done with GEN-SIM mixing
377  if ( s[2][index].split()[ s[2][index].split().index('--pileup')+1 ] ).find('25ns') > 0 :
378  processStrPrefix='PU25ns_'
379  elif ( s[2][index].split()[ s[2][index].split().index('--pileup')+1 ] ).find('50ns') > 0 :
380  processStrPrefix='PU50ns_'
381  if 'premix_stage2' in s[2][index] and '--pileup_input' in s[2][index]: # take care of pu overlay done with DIGI mixing of premixed events
382  if s[2][index].split()[ s[2][index].split().index('--pileup_input')+1 ].find('25ns') > 0 :
383  processStrPrefix='PUpmx25ns_'
384  elif s[2][index].split()[ s[2][index].split().index('--pileup_input')+1 ].find('50ns') > 0 :
385  processStrPrefix='PUpmx50ns_'
386 
387  if acqEra:
388  #chainDict['AcquisitionEra'][step]=(chainDict['CMSSWVersion']+'-PU_'+chainDict['nowmTasklist'][-1]['GlobalTag']).replace('::All','')+thisLabel
389  chainDict['AcquisitionEra'][step]=chainDict['CMSSWVersion']
390  chainDict['ProcessingString'][step]=processStrPrefix+chainDict['nowmTasklist'][-1]['GlobalTag'].replace('::All','').replace('-','_')+thisLabel
391  if 'NANOMERGE' in step :
392  chainDict['ProcessingString'][step]=processStrPrefix+nanoedmGT.replace('::All','').replace('-','_')+thisLabel
393  else:
394  #chainDict['nowmTasklist'][-1]['AcquisitionEra']=(chainDict['CMSSWVersion']+'-PU_'+chainDict['nowmTasklist'][-1]['GlobalTag']).replace('::All','')+thisLabel
395  chainDict['nowmTasklist'][-1]['AcquisitionEra']=chainDict['CMSSWVersion']
396  chainDict['nowmTasklist'][-1]['ProcessingString']=processStrPrefix+chainDict['nowmTasklist'][-1]['GlobalTag'].replace('::All','').replace('-','_')+thisLabel
397  if 'NANOMERGE' in step :
398  chainDict['nowmTasklist'][-1]['ProcessingString']=processStrPrefix+nanoedmGT.replace('::All','').replace('-','_')+thisLabel
399 
400  if (self.batchName):
401  chainDict['nowmTasklist'][-1]['Campaign'] = chainDict['nowmTasklist'][-1]['AcquisitionEra']+self.batchName
402 
403  # specify different ProcessingString for double miniAOD dataset
404  if ('DBLMINIAODMCUP15NODQM' in step):
405  chainDict['nowmTasklist'][-1]['ProcessingString']=chainDict['nowmTasklist'][-1]['ProcessingString']+'_miniAOD'
406 
407  if( chainDict['nowmTasklist'][-1]['Multicore'] ):
408  # the scaling factor of 1.2GB / thread is empirical and measured on a SECOND round of tests with PU samples
409  # the number of threads is NO LONGER assumed to be the same for all tasks
410  # https://hypernews.cern.ch/HyperNews/CMS/get/edmFramework/3509/1/1/1.html
411  # now change to 1.5GB / additional thread according to discussion:
412  # https://hypernews.cern.ch/HyperNews/CMS/get/relval/4817/1/1.html
413 # chainDict['nowmTasklist'][-1]['Memory'] = 3000 + int( chainDict['nowmTasklist'][-1]['Multicore'] -1 )*1500
414  chainDict['nowmTasklist'][-1]['Memory'] = self.memoryOffset + int( chainDict['nowmTasklist'][-1]['Multicore'] -1 ) * self.memPerCore
415 
416  index+=1
417  #end of loop through steps
418  chainDict['RequestString']='RV'+chainDict['CMSSWVersion']+s[1].split('+')[0]
419  if processStrPrefix or thisLabel:
420  chainDict['RequestString']+='_'+processStrPrefix+thisLabel
421 
422 ### PrepID
423  chainDict['PrepID'] = chainDict['CMSSWVersion']+'__'+self.batchTime+'-'+s[1].split('+')[0]
424  if(self.batchName):
425  chainDict['PrepID'] = chainDict['CMSSWVersion']+self.batchName+'-'+s[1].split('+')[0]
426  if( 'HIN' in self.batchName ):
427  chainDict['SubRequestType'] = "HIRelVal"
428 
429  #wrap up for this one
430  import pprint
431  #print 'wrapping up'
432  #pprint.pprint(chainDict)
433  #loop on the task list
434  for i_second in reversed(range(len(chainDict['nowmTasklist']))):
435  t_second=chainDict['nowmTasklist'][i_second]
436  #print "t_second taskname", t_second['TaskName']
437  if 'primary' in t_second['nowmIO']:
438  #print t_second['nowmIO']['primary']
439  primary=t_second['nowmIO']['primary'][0].replace('file:','')
440  for i_input in reversed(range(0,i_second)):
441  t_input=chainDict['nowmTasklist'][i_input]
442  for (om,o) in t_input['nowmIO'].items():
443  if primary in o:
444  #print "found",primary,"procuced by",om,"of",t_input['TaskName']
445  t_second['InputTask'] = t_input['TaskName']
446  t_second['InputFromOutputModule'] = om
447  #print 't_second',pprint.pformat(t_second)
448  if t_second['TaskName'].startswith('HARVEST'):
449  chainDict.update(copy.deepcopy(self.defaultHarvest))
450  chainDict['DQMConfigCacheID']=t_second['ConfigCacheID']
451  ## the info are not in the task specific dict but in the general dict
452  #t_input.update(copy.deepcopy(self.defaultHarvest))
453  #t_input['DQMConfigCacheID']=t_second['ConfigCacheID']
454  break
455 
456  # agreed changes for wm injection:
457  # - Campaign: *optional* string during creation. It will default to AcqEra value if possible.
458  # Otherwise it will be empty.
459  # - AcquisitionEra: *mandatory* string at request level during creation. *optional* string
460  # at task level during creation. "optional" during assignment.
461  # - ProcessingString: *mandatory* string at request level during creation. *optional* string
462  # at task level during creation. "optional" during assignment.
463  # - ProcessingVersion: *optional* during creation (default 1). *optional* during assignment.
464  #
465  # Which requires following changes here:
466  # - reset Global AcuisitionEra, ProcessingString to be the one in the first task
467  # - and also Campaign to be always the same as the AcquisitionEra
468 
469  if acqEra:
470  chainDict['AcquisitionEra'] = chainDict['AcquisitionEra'].values()[0]
471  chainDict['ProcessingString'] = chainDict['ProcessingString'].values()[0]
472  else:
473  chainDict['AcquisitionEra'] = chainDict['nowmTasklist'][0]['AcquisitionEra']
474  chainDict['ProcessingString'] = chainDict['nowmTasklist'][0]['ProcessingString']
475 
476 ##### batch name appended to Campaign name
477 # chainDict['Campaign'] = chainDict['AcquisitionEra']
478  chainDict['Campaign'] = chainDict['AcquisitionEra']+self.batchName
479 
480  ## clean things up now
481  itask=0
482  if self.keep:
483  for i in self.keep:
484  if isinstance(i, int) and i < len(chainDict['nowmTasklist']):
485  chainDict['nowmTasklist'][i]['KeepOutput']=True
486  for (i,t) in enumerate(chainDict['nowmTasklist']):
487  if t['TaskName'].startswith('HARVEST'):
488  continue
489  if not self.keep:
490  t['KeepOutput']=True
491  elif t['TaskName'] in self.keep:
492  t['KeepOutput']=True
493  if t['TaskName'].startswith('HYBRIDRepackHI2015VR'):
494  t['KeepOutput']=False
495  t.pop('nowmIO')
496  itask+=1
497  chainDict['Task%d'%(itask)]=t
498 
499 
500  ##
501 
502 
503  ## provide the number of tasks
504  chainDict['TaskChain']=itask#len(chainDict['nowmTasklist'])
505 
506  chainDict.pop('nowmTasklist')
507  self.chainDicts[n]=chainDict
508 
509 
510  return 0
511 
512  def uploadConf(self,filePath,label,where):
513  labelInCouch=self.label+'_'+label
514  cacheName=filePath.split('/')[-1]
515  if self.testMode:
516  self.count+=1
517  print('\tFake upload of',filePath,'to couch with label',labelInCouch)
518  return self.count
519  else:
520  try:
521  from modules.wma import upload_to_couch,DATABASE_NAME
522  except:
523  print('\n\tUnable to find wmcontrol modules. Please include it in your python path\n')
524  print('\n\t QUIT\n')
525  sys.exit(-16)
526 
527  if cacheName in self.couchCache:
528  print("Not re-uploading",filePath,"to",where,"for",label)
529  cacheId=self.couchCache[cacheName]
530  else:
531  print("Loading",filePath,"to",where,"for",label)
532  ## totally fork the upload to couch to prevent cross loading of process configurations
533  pool = multiprocessing.Pool(1)
534  cacheIds = pool.map( upload_to_couch_oneArg, [(filePath,labelInCouch,self.user,self.group,where)] )
535  cacheId = cacheIds[0]
536  self.couchCache[cacheName]=cacheId
537  return cacheId
538 
539  def upload(self):
540  for (n,d) in self.chainDicts.items():
541  for it in d:
542  if it.startswith("Task") and it!='TaskChain':
543  #upload
544  couchID=self.uploadConf(d[it]['ConfigCacheID'],
545  str(n)+d[it]['TaskName'],
546  d['ConfigCacheUrl']
547  )
548  print(d[it]['ConfigCacheID']," uploaded to couchDB for",str(n),"with ID",couchID)
549  d[it]['ConfigCacheID']=couchID
550  if it =='DQMConfigCacheID':
551  couchID=self.uploadConf(d['DQMConfigCacheID'],
552  str(n)+'harvesting',
553  d['ConfigCacheUrl']
554  )
555  print(d['DQMConfigCacheID'],"uploaded to couchDB for",str(n),"with ID",couchID)
556  d['DQMConfigCacheID']=couchID
557 
558 
559  def submit(self):
560  try:
561  from modules.wma import makeRequest,approveRequest
562  from wmcontrol import random_sleep
563  print('\n\tFound wmcontrol\n')
564  except:
565  print('\n\tUnable to find wmcontrol modules. Please include it in your python path\n')
566  if not self.testMode:
567  print('\n\t QUIT\n')
568  sys.exit(-17)
569 
570  import pprint
571  for (n,d) in self.chainDicts.items():
572  if self.testMode:
573  print("Only viewing request",n)
574  print(pprint.pprint(d))
575  else:
576  #submit to wmagent each dict
577  print("For eyes before submitting",n)
578  print(pprint.pprint(d))
579  print("Submitting",n,"...........")
580  workFlow=makeRequest(self.wmagent,d,encodeDict=True)
581  print("...........",n,"submitted")
582  random_sleep()
583 
584 
585 
def prepare(self, mReader, directories, mode='init')
def performInjectionOptionTest(opt)
def uploadConf(self, filePath, label, where)
the info are not in the task specific dict but in the general dict t_input.update(copy.deepcopy(self.defaultHarvest)) t_input[&#39;DQMConfigCacheID&#39;]=t_second[&#39;ConfigCacheID&#39;]
def replace(string, replacements)
S & print(S &os, JobReport::InputFile const &f)
Definition: JobReport.cc:65
void find(edm::Handle< EcalRecHitCollection > &hits, DetId thisDet, std::vector< EcalRecHitCollection::const_iterator > &hit, bool debug=false)
Definition: FindCaloHit.cc:20
def __init__(self, opt, mode='init', options='')
#define str(s)
double split
Definition: MVATrainer.cc:139
def upload_to_couch_oneArg(arguments)