/data/refman/pasoursint/CMSSW_4_1_8_patch9/src/CalibMuon/DTCalibration/python/Workflow/CrabTask.py

Go to the documentation of this file.

from crabWrap import crabCreate,crabSubmit,crabWatch,getOutput
from tools import replaceTemplate
import os
#from threading import Thread

#class CrabTask(Thread):
class CrabTask:
    def __init__(self, dir, crab_cfg, pset=None, pset_name='mypset.py'):
        #Thread.__init__(self)
        self.dir = dir
  
        self.crabCfg_name = 'crab.cfg'
        self.crab_cfg = crab_cfg
  
        self.pset_name = pset_name
        self.pset = pset
        self.initializeTask(dir=self.dir)

    def initializeTask(self, dir):
        if not os.path.exists(dir): os.makedirs(dir)

        # Write pset 
        if self.pset:
            self.crab_cfg.set('CMSSW','pset',self.pset_name)
            open(dir + '/' + self.pset_name,'w').write(self.pset) 

        # Write CRAB cfg
        self.crab_cfg.write(open(dir + '/' + self.crabCfg_name,'w'))
         
    def create(self,dir):
        self.project = crabCreate(dir,self.crabCfg_name)
        return self.project

    def submit(self):
        if not self.project: raise RuntimeError
        crabSubmit(self.project)

    def getoutput(self):
        if not self.project: raise RuntimeError
        getOutput(self.project)

    #def watch(self):
    #    if not self.project: raise RuntimeError
    #    crabWatch(getOutput,self.project) 
        
    def run(self):
        #self.initializeTask(dir=self.dir)
        proj = self.create(self.dir) 
        self.submit()
        return proj

if __name__ == '__main__':

    run = None
    trial = None
    import sys
    for opt in sys.argv:
        if opt[:4] == 'run=':
            run = opt[4:] 
        if opt[:6] == 'trial=':
            trial = opt[6:] 
 
    if not run: raise ValueError,'Need to set run number' 
    if not trial: raise ValueError,'Need to set trial number'

    #run = 100850
    #ntrial = 1 
    pset_name = 'DTTTrigCalibration_cfg.py'

    crab_template = 'templates/crab/crab_ttrig_prod_TEMPL.cfg'
    crab_opts = {'DATASETPATH':'/Cosmics/Commissioning09-v3/RAW',
                 'RUNNUMBER':run,
                 'PSET':pset_name, 
                 'USERDIRCAF':'TTRIGCalibration/Run' + str(run) + '/v' + str(trial),
                 'EMAIL':'vilela@to.infn.it'}

    pset_template = 'templates/config/DTTTrigCalibration_TEMPL_cfg.py' 
    pset_opts = {'GLOBALTAG':'CRAFT_31X::All',
                 'DIGITEMPLATE':'muonDTDigis'}

    dir = 'Run%s'%run
    dir += '/Ttrig/Production'

    crab_cfg = replaceTemplate(crab_template,**crab_opts)
    pset = replaceTemplate(pset_template,**pset_opts)

    prod = CrabTask(dir,crab_cfg,pset,pset_name)
    project = prod.run()

    print "Sent jobs with project",project