ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/COMP/CRAB/python/cms_cmssw.py
(Generate patch)

Comparing COMP/CRAB/python/cms_cmssw.py (file contents):
Revision 1.116.2.8 by fanzago, Wed Oct 17 13:25:00 2007 UTC vs.
Revision 1.231 by spiga, Mon Aug 25 16:13:05 2008 UTC

# Line 2 | Line 2 | from JobType import JobType
2   from crab_logger import Logger
3   from crab_exceptions import *
4   from crab_util import *
5 < from BlackWhiteListParser import BlackWhiteListParser
5 > from BlackWhiteListParser import SEBlackWhiteListParser
6   import common
7   import Scram
8 + from LFNBaseName import *
9  
10   import os, string, glob
11  
12   class Cmssw(JobType):
13 <    def __init__(self, cfg_params, ncjobs):
13 >    def __init__(self, cfg_params, ncjobs,skip_blocks, isNew):
14          JobType.__init__(self, 'CMSSW')
15          common.logger.debug(3,'CMSSW::__init__')
16 +        self.skip_blocks = skip_blocks
17 +
18 +        self.argsList = []
19  
20          self._params = {}
21          self.cfg_params = cfg_params
18
22          # init BlackWhiteListParser
23 <        self.blackWhiteListParser = BlackWhiteListParser(cfg_params)
23 >        self.blackWhiteListParser = SEBlackWhiteListParser(cfg_params)
24  
25 <        try:
23 <            self.MaxTarBallSize = float(self.cfg_params['EDG.maxtarballsize'])
24 <        except KeyError:
25 <            self.MaxTarBallSize = 9.5
25 >        self.MaxTarBallSize = float(self.cfg_params.get('EDG.maxtarballsize',9.5))
26  
27          # number of jobs requested to be created, limit obj splitting
28          self.ncjobs = ncjobs
29  
30          log = common.logger
31 <        
31 >
32          self.scram = Scram.Scram(cfg_params)
33          self.additional_inbox_files = []
34          self.scriptExe = ''
35          self.executable = ''
36          self.executable_arch = self.scram.getArch()
37          self.tgz_name = 'default.tgz'
38        self.additional_tgz_name = 'additional.tgz'
38          self.scriptName = 'CMSSW.sh'
39 <        self.pset = ''      #scrip use case Da  
40 <        self.datasetPath = '' #scrip use case Da
39 >        self.pset = ''
40 >        self.datasetPath = ''
41  
42          # set FJR file name
43          self.fjrFileName = 'crab_fjr.xml'
44  
45          self.version = self.scram.getSWVersion()
46 <        
47 <        #
48 <        # Try to block creation in case of arch/version mismatch
49 <        #
50 <
51 <        a = string.split(self.version, "_")
52 <
53 <        if int(a[1]) == 1 and (int(a[2]) < 5 and self.executable_arch.find('slc4') == 0):
54 <            msg = "Error: CMS does not support %s with %s architecture"%(self.version, self.executable_arch)
55 <            raise CrabException(msg)
57 <        if int(a[1]) == 1 and (int(a[2]) >= 5 and self.executable_arch.find('slc3') == 0):
58 <            msg = "Error: CMS does not support %s with %s architecture"%(self.version, self.executable_arch)
46 >        version_array = self.version.split('_')
47 >        self.CMSSW_major = 0
48 >        self.CMSSW_minor = 0
49 >        self.CMSSW_patch = 0
50 >        try:
51 >            self.CMSSW_major = int(version_array[1])
52 >            self.CMSSW_minor = int(version_array[2])
53 >            self.CMSSW_patch = int(version_array[3])
54 >        except:
55 >            msg = "Cannot parse CMSSW version string: " + self.version + " for major and minor release number!"
56              raise CrabException(msg)
60        
61        common.taskDB.setDict('codeVersion',self.version)
62        self.setParam_('application', self.version)
57  
58          ### collect Data cards
59  
60 <        ## get DBS mode
61 <        try:
68 <            self.use_dbs_1 = int(self.cfg_params['CMSSW.use_dbs_1'])
69 <        except KeyError:
70 <            self.use_dbs_1 = 0
71 <            
72 <        try:
73 <            tmp =  cfg_params['CMSSW.datasetpath']
74 <            log.debug(6, "CMSSW::CMSSW(): datasetPath = "+tmp)
75 <            if string.lower(tmp)=='none':
76 <                self.datasetPath = None
77 <                self.selectNoInput = 1
78 <            else:
79 <                self.datasetPath = tmp
80 <                self.selectNoInput = 0
81 <        except KeyError:
82 <            msg = "Error: datasetpath not defined "  
60 >        if not cfg_params.has_key('CMSSW.datasetpath'):
61 >            msg = "Error: datasetpath not defined "
62              raise CrabException(msg)
63  
64 <        # ML monitoring
65 <        # split dataset path style: /PreProdR3Minbias/SIM/GEN-SIM
66 <        if not self.datasetPath:
67 <            self.setParam_('dataset', 'None')
68 <            self.setParam_('owner', 'None')
64 >        ### Temporary: added to remove input file control in the case of PU
65 >        self.dataset_pu = cfg_params.get('CMSSW.dataset_pu', None)
66 >
67 >        tmp =  cfg_params['CMSSW.datasetpath']
68 >        log.debug(6, "CMSSW::CMSSW(): datasetPath = "+tmp)
69 >        if string.lower(tmp)=='none':
70 >            self.datasetPath = None
71 >            self.selectNoInput = 1
72          else:
73 <            try:
74 <                datasetpath_split = self.datasetPath.split("/")
93 <                # standard style
94 <                self.setParam_('datasetFull', self.datasetPath)
95 <                if self.use_dbs_1 == 1 :
96 <                    self.setParam_('dataset', datasetpath_split[1])
97 <                    self.setParam_('owner', datasetpath_split[-1])
98 <                else:
99 <                    self.setParam_('dataset', datasetpath_split[1])
100 <                    self.setParam_('owner', datasetpath_split[2])
101 <            except:
102 <                self.setParam_('dataset', self.datasetPath)
103 <                self.setParam_('owner', self.datasetPath)
104 <                
105 <        self.setTaskid_()
106 <        self.setParam_('taskId', self.cfg_params['taskId'])
73 >            self.datasetPath = tmp
74 >            self.selectNoInput = 0
75  
76          self.dataTiers = []
77 <
77 >        self.debugWrap = ''
78 >        self.debug_wrapper = cfg_params.get('USER.debug_wrapper',False)
79 >        if self.debug_wrapper: self.debugWrap='--debug'
80          ## now the application
81 <        try:
82 <            self.executable = cfg_params['CMSSW.executable']
113 <            self.setParam_('exe', self.executable)
114 <            log.debug(6, "CMSSW::CMSSW(): executable = "+self.executable)
115 <            msg = "Default executable cmsRun overridden. Switch to " + self.executable
116 <            log.debug(3,msg)
117 <        except KeyError:
118 <            self.executable = 'cmsRun'
119 <            self.setParam_('exe', self.executable)
120 <            msg = "User executable not defined. Use cmsRun"
121 <            log.debug(3,msg)
122 <            pass
81 >        self.executable = cfg_params.get('CMSSW.executable','cmsRun')
82 >        log.debug(6, "CMSSW::CMSSW(): executable = "+self.executable)
83  
84 <        try:
125 <            self.pset = cfg_params['CMSSW.pset']
126 <            log.debug(6, "Cmssw::Cmssw(): PSet file = "+self.pset)
127 <            if self.pset.lower() != 'none' :
128 <                if (not os.path.exists(self.pset)):
129 <                    raise CrabException("User defined PSet file "+self.pset+" does not exist")
130 <            else:
131 <                self.pset = None
132 <        except KeyError:
84 >        if not cfg_params.has_key('CMSSW.pset'):
85              raise CrabException("PSet file missing. Cannot run cmsRun ")
86 +        self.pset = cfg_params['CMSSW.pset']
87 +        log.debug(6, "Cmssw::Cmssw(): PSet file = "+self.pset)
88 +        if self.pset.lower() != 'none' :
89 +            if (not os.path.exists(self.pset)):
90 +                raise CrabException("User defined PSet file "+self.pset+" does not exist")
91 +        else:
92 +            self.pset = None
93  
94          # output files
95          ## stuff which must be returned always via sandbox
# Line 140 | Line 99 | class Cmssw(JobType):
99          self.output_file_sandbox.append(self.fjrFileName)
100  
101          # other output files to be returned via sandbox or copied to SE
102 <        try:
103 <            self.output_file = []
104 <            tmp = cfg_params['CMSSW.output_file']
105 <            if tmp != '':
106 <                tmpOutFiles = string.split(cfg_params['CMSSW.output_file'],',')
107 <                log.debug(7, 'cmssw::cmssw(): output files '+str(tmpOutFiles))
108 <                for tmp in tmpOutFiles:
109 <                    tmp=string.strip(tmp)
151 <                    self.output_file.append(tmp)
152 <                    pass
153 <            else:
154 <                log.message("No output file defined: only stdout/err and the CRAB Framework Job Report will be available\n")
155 <                pass
156 <            pass
157 <        except KeyError:
158 <            log.message("No output file defined: only stdout/err and the CRAB Framework Job Report will be available\n")
159 <            pass
102 >        outfileflag = False
103 >        self.output_file = []
104 >        tmp = cfg_params.get('CMSSW.output_file',None)
105 >        if tmp :
106 >            self.output_file = [x.strip() for x in tmp.split(',')]
107 >            outfileflag = True #output found
108 >        #else:
109 >        #    log.message("No output file defined: only stdout/err and the CRAB Framework Job Report will be available\n")
110  
111          # script_exe file as additional file in inputSandbox
112 <        try:
113 <            self.scriptExe = cfg_params['USER.script_exe']
114 <            if self.scriptExe != '':
115 <               if not os.path.isfile(self.scriptExe):
116 <                  msg ="ERROR. file "+self.scriptExe+" not found"
117 <                  raise CrabException(msg)
168 <               self.additional_inbox_files.append(string.strip(self.scriptExe))
169 <        except KeyError:
170 <            self.scriptExe = ''
112 >        self.scriptExe = cfg_params.get('USER.script_exe',None)
113 >        if self.scriptExe :
114 >            if not os.path.isfile(self.scriptExe):
115 >                msg ="ERROR. file "+self.scriptExe+" not found"
116 >                raise CrabException(msg)
117 >            self.additional_inbox_files.append(string.strip(self.scriptExe))
118  
172        #CarlosDaniele
119          if self.datasetPath == None and self.pset == None and self.scriptExe == '' :
120 <           msg ="Error. script_exe  not defined"
121 <           raise CrabException(msg)
120 >            msg ="Error. script_exe  not defined"
121 >            raise CrabException(msg)
122 >
123 >        # use parent files...
124 >        self.useParent = self.cfg_params.get('CMSSW.use_parent',False)
125  
126          ## additional input files
127 <        try:
127 >        if cfg_params.has_key('USER.additional_input_files'):
128              tmpAddFiles = string.split(cfg_params['USER.additional_input_files'],',')
129              for tmp in tmpAddFiles:
130                  tmp = string.strip(tmp)
# Line 192 | Line 141 | class Cmssw(JobType):
141                      if not os.path.exists(file):
142                          raise CrabException("Additional input file not found: "+file)
143                      pass
195                    # fname = string.split(file, '/')[-1]
196                    # storedFile = common.work_space.pathForTgz()+'share/'+fname
197                    # shutil.copyfile(file, storedFile)
144                      self.additional_inbox_files.append(string.strip(file))
145                  pass
146              pass
147              common.logger.debug(5,"Additional input files: "+str(self.additional_inbox_files))
148 <        except KeyError:
203 <            pass
204 <
205 <        # files per job
206 <        try:
207 <            if (cfg_params['CMSSW.files_per_jobs']):
208 <                raise CrabException("files_per_jobs no longer supported.  Quitting.")
209 <        except KeyError:
210 <            pass
148 >        pass
149  
150          ## Events per job
151 <        try:
151 >        if cfg_params.has_key('CMSSW.events_per_job'):
152              self.eventsPerJob =int( cfg_params['CMSSW.events_per_job'])
153              self.selectEventsPerJob = 1
154 <        except KeyError:
154 >        else:
155              self.eventsPerJob = -1
156              self.selectEventsPerJob = 0
157 <    
157 >
158          ## number of jobs
159 <        try:
159 >        if cfg_params.has_key('CMSSW.number_of_jobs'):
160              self.theNumberOfJobs =int( cfg_params['CMSSW.number_of_jobs'])
161              self.selectNumberOfJobs = 1
162 <        except KeyError:
162 >        else:
163              self.theNumberOfJobs = 0
164              self.selectNumberOfJobs = 0
165  
166 <        try:
166 >        if cfg_params.has_key('CMSSW.total_number_of_events'):
167              self.total_number_of_events = int(cfg_params['CMSSW.total_number_of_events'])
168              self.selectTotalNumberEvents = 1
169 <        except KeyError:
169 >            if self.selectNumberOfJobs  == 1:
170 >                if (self.total_number_of_events != -1) and int(self.total_number_of_events) < int(self.theNumberOfJobs):
171 >                    msg = 'Must specify at least one event per job. total_number_of_events > number_of_jobs '
172 >                    raise CrabException(msg)
173 >        else:
174              self.total_number_of_events = 0
175              self.selectTotalNumberEvents = 0
176  
177 <        if self.pset != None: #CarlosDaniele
177 >        if self.pset != None:
178               if ( (self.selectTotalNumberEvents + self.selectEventsPerJob + self.selectNumberOfJobs) != 2 ):
179                   msg = 'Must define exactly two of total_number_of_events, events_per_job, or number_of_jobs.'
180                   raise CrabException(msg)
# Line 241 | Line 183 | class Cmssw(JobType):
183                   msg = 'Must specify  number_of_jobs.'
184                   raise CrabException(msg)
185  
186 <        ## source seed for pythia
187 <        try:
188 <            self.sourceSeed = int(cfg_params['CMSSW.pythia_seed'])
189 <        except KeyError:
190 <            self.sourceSeed = None
191 <            common.logger.debug(5,"No seed given")
192 <
193 <        try:
194 <            self.sourceSeedVtx = int(cfg_params['CMSSW.vtx_seed'])
195 <        except KeyError:
196 <            self.sourceSeedVtx = None
197 <            common.logger.debug(5,"No vertex seed given")
198 <
199 <        try:
200 <            self.sourceSeedG4 = int(cfg_params['CMSSW.g4_seed'])
201 <        except KeyError:
202 <            self.sourceSeedG4 = None
203 <            common.logger.debug(5,"No g4 sim hits seed given")
186 >        ## New method of dealing with seeds
187 >        self.incrementSeeds = []
188 >        self.preserveSeeds = []
189 >        if cfg_params.has_key('CMSSW.preserve_seeds'):
190 >            tmpList = cfg_params['CMSSW.preserve_seeds'].split(',')
191 >            for tmp in tmpList:
192 >                tmp.strip()
193 >                self.preserveSeeds.append(tmp)
194 >        if cfg_params.has_key('CMSSW.increment_seeds'):
195 >            tmpList = cfg_params['CMSSW.increment_seeds'].split(',')
196 >            for tmp in tmpList:
197 >                tmp.strip()
198 >                self.incrementSeeds.append(tmp)
199 >
200 >        ## FUTURE: Can remove in CRAB 2.4.0
201 >        self.sourceSeed    = cfg_params.get('CMSSW.pythia_seed',None)
202 >        self.sourceSeedVtx = cfg_params.get('CMSSW.vtx_seed',None)
203 >        self.sourceSeedG4  = cfg_params.get('CMSSW.g4_seed',None)
204 >        self.sourceSeedMix = cfg_params.get('CMSSW.mix_seed',None)
205 >        if self.sourceSeed or self.sourceSeedVtx or self.sourceSeedG4 or self.sourceSeedMix:
206 >            msg = 'pythia_seed, vtx_seed, g4_seed, and mix_seed are no longer valid settings. You must use increment_seeds or preserve_seeds'
207 >            raise CrabException(msg)
208  
209 <        try:
264 <            self.sourceSeedMix = int(cfg_params['CMSSW.mix_seed'])
265 <        except KeyError:
266 <            self.sourceSeedMix = None
267 <            common.logger.debug(5,"No mix seed given")
209 >        self.firstRun = cfg_params.get('CMSSW.first_run',None)
210  
211 <        try:
212 <            self.firstRun = int(cfg_params['CMSSW.first_run'])
213 <        except KeyError:
272 <            self.firstRun = None
273 <            common.logger.debug(5,"No first run given")
274 <        if self.pset != None: #CarlosDaniele
275 <            ver = string.split(self.version,"_")
276 <            if (int(ver[1])>=1 and int(ver[2])>=5):
277 <                import PsetManipulator150 as pp
278 <            else:
279 <                import PsetManipulator as pp
280 <            PsetEdit = pp.PsetManipulator(self.pset) #Daniele Pset
211 >        # Copy/return
212 >        self.copy_data = int(cfg_params.get('USER.copy_data',0))
213 >        self.return_data = int(cfg_params.get('USER.return_data',0))
214  
215          #DBSDLS-start
216 <        ## Initialize the variables that are extracted from DBS/DLS and needed in other places of the code
216 >        ## Initialize the variables that are extracted from DBS/DLS and needed in other places of the code
217          self.maxEvents=0  # max events available   ( --> check the requested nb. of evts in Creator.py)
218          self.DBSPaths={}  # all dbs paths requested ( --> input to the site local discovery script)
219          self.jobDestination=[]  # Site destination(s) for each job (list of lists)
# Line 289 | Line 222 | class Cmssw(JobType):
222          blockSites = {}
223          if self.datasetPath:
224              blockSites = self.DataDiscoveryAndLocation(cfg_params)
225 <        #DBSDLS-end          
225 >        #DBSDLS-end
226  
294        self.tgzNameWithPath = self.getTarBall(self.executable)
295    
227          ## Select Splitting
228 <        if self.selectNoInput:
229 <            if self.pset == None: #CarlosDaniele
228 >        if self.selectNoInput:
229 >            if self.pset == None:
230                  self.jobSplittingForScript()
231              else:
232                  self.jobSplittingNoInput()
233          else:
234              self.jobSplittingByBlocks(blockSites)
235  
236 <        # modify Pset
237 <        if self.pset != None: #CarlosDaniele
238 <            try:
239 <                if (self.datasetPath): # standard job
240 <                    # allow to processa a fraction of events in a file
241 <                    PsetEdit.inputModule("INPUT")
242 <                    PsetEdit.maxEvent("INPUTMAXEVENTS")
243 <                    PsetEdit.skipEvent("INPUTSKIPEVENTS")
244 <                else:  # pythia like job
236 >        # modify Pset only the first time
237 >        if isNew:
238 >            if self.pset != None:
239 >                import PsetManipulator as pp
240 >                PsetEdit = pp.PsetManipulator(self.pset)
241 >                try:
242 >                    # Add FrameworkJobReport to parameter-set, set max events.
243 >                    # Reset later for data jobs by writeCFG which does all modifications
244 >                    PsetEdit.addCrabFJR(self.fjrFileName) # FUTURE: Job report addition not needed by CMSSW>1.5
245                      PsetEdit.maxEvent(self.eventsPerJob)
246 <                    if (self.firstRun):
247 <                        PsetEdit.pythiaFirstRun("INPUTFIRSTRUN")  #First Run
248 <                    if (self.sourceSeed) :
249 <                        PsetEdit.pythiaSeed("INPUT")
250 <                        if (self.sourceSeedVtx) :
251 <                            PsetEdit.vtxSeed("INPUTVTX")
252 <                        if (self.sourceSeedG4) :
253 <                            PsetEdit.g4Seed("INPUTG4")
254 <                        if (self.sourceSeedMix) :
255 <                            PsetEdit.mixSeed("INPUTMIX")
256 <                # add FrameworkJobReport to parameter-set
257 <                PsetEdit.addCrabFJR(self.fjrFileName)
258 <                PsetEdit.psetWriter(self.configFilename())
259 <            except:
260 <                msg='Error while manipuliating ParameterSet: exiting...'
261 <                raise CrabException(msg)
246 >                    PsetEdit.psetWriter(self.configFilename())
247 >                    ## If present, add TFileService to output files
248 >                    if not int(cfg_params.get('CMSSW.skip_TFileService_output',0)):
249 >                        tfsOutput = PsetEdit.getTFileService()
250 >                        if tfsOutput:
251 >                            if tfsOutput in self.output_file:
252 >                                common.logger.debug(5,"Output from TFileService "+tfsOutput+" already in output files")
253 >                            else:
254 >                                outfileflag = True #output found
255 >                                self.output_file.append(tfsOutput)
256 >                                common.logger.message("Adding "+tfsOutput+" to output files (from TFileService)")
257 >                            pass
258 >                        pass
259 >                    ## If present and requested, add PoolOutputModule to output files
260 >                    if int(cfg_params.get('CMSSW.get_edm_output',0)):
261 >                        edmOutput = PsetEdit.getPoolOutputModule()
262 >                        if edmOutput:
263 >                            if edmOutput in self.output_file:
264 >                                common.logger.debug(5,"Output from PoolOutputModule "+edmOutput+" already in output files")
265 >                            else:
266 >                                self.output_file.append(edmOutput)
267 >                                common.logger.message("Adding "+edmOutput+" to output files (from PoolOutputModule)")
268 >                            pass
269 >                        pass
270 >                except CrabException:
271 >                    msg='Error while manipulating ParameterSet: exiting...'
272 >                    raise CrabException(msg)
273 >            ## Prepare inputSandbox TarBall (only the first time)
274 >            self.tgzNameWithPath = self.getTarBall(self.executable)
275  
276      def DataDiscoveryAndLocation(self, cfg_params):
277  
278          import DataDiscovery
335        import DataDiscovery_DBS2
279          import DataLocation
280          common.logger.debug(10,"CMSSW::DataDiscoveryAndLocation()")
281  
# Line 341 | Line 284 | class Cmssw(JobType):
284          ## Contact the DBS
285          common.logger.message("Contacting Data Discovery Services ...")
286          try:
287 <
345 <            if self.use_dbs_1 == 1 :
346 <                self.pubdata=DataDiscovery.DataDiscovery(datasetPath, cfg_params)
347 <            else :
348 <                self.pubdata=DataDiscovery_DBS2.DataDiscovery_DBS2(datasetPath, cfg_params)
287 >            self.pubdata=DataDiscovery.DataDiscovery(datasetPath, cfg_params,self.skip_blocks)
288              self.pubdata.fetchDBSInfo()
289  
290          except DataDiscovery.NotExistingDatasetError, ex :
# Line 357 | Line 296 | class Cmssw(JobType):
296          except DataDiscovery.DataDiscoveryError, ex:
297              msg = 'ERROR ***: failed Data Discovery in DBS :  %s'%ex.getErrorMessage()
298              raise CrabException(msg)
360        except DataDiscovery_DBS2.NotExistingDatasetError_DBS2, ex :
361            msg = 'ERROR ***: failed Data Discovery in DBS : %s'%ex.getErrorMessage()
362            raise CrabException(msg)
363        except DataDiscovery_DBS2.NoDataTierinProvenanceError_DBS2, ex :
364            msg = 'ERROR ***: failed Data Discovery in DBS : %s'%ex.getErrorMessage()
365            raise CrabException(msg)
366        except DataDiscovery_DBS2.DataDiscoveryError_DBS2, ex:
367            msg = 'ERROR ***: failed Data Discovery in DBS :  %s'%ex.getErrorMessage()
368            raise CrabException(msg)
299  
300          self.filesbyblock=self.pubdata.getFiles()
301          self.eventsbyblock=self.pubdata.getEventsPerBlock()
302          self.eventsbyfile=self.pubdata.getEventsPerFile()
303 +        self.parentFiles=self.pubdata.getParent()
304  
305          ## get max number of events
306 <        self.maxEvents=self.pubdata.getMaxEvents() ##  self.maxEvents used in Creator.py
306 >        self.maxEvents=self.pubdata.getMaxEvents()
307  
308          ## Contact the DLS and build a list of sites hosting the fileblocks
309          try:
# Line 381 | Line 312 | class Cmssw(JobType):
312          except DataLocation.DataLocationError , ex:
313              msg = 'ERROR ***: failed Data Location in DLS \n %s '%ex.getErrorMessage()
314              raise CrabException(msg)
315 <        
315 >
316  
317          sites = dataloc.getSites()
318          allSites = []
# Line 395 | Line 326 | class Cmssw(JobType):
326          common.logger.message("Requested dataset: " + datasetPath + " has " + str(self.maxEvents) + " events in " + str(len(self.filesbyblock.keys())) + " blocks.\n")
327  
328          return sites
329 <    
329 >
330      def jobSplittingByBlocks(self, blockSites):
331          """
332          Perform job splitting. Jobs run over an integer number of files
# Line 445 | Line 376 | class Cmssw(JobType):
376              totalNumberOfJobs = 999999999
377          else :
378              totalNumberOfJobs = self.ncjobs
448            
379  
380          blocks = blockSites.keys()
381          blockCount = 0
# Line 465 | Line 395 | class Cmssw(JobType):
395              blockCount += 1
396              if block not in jobsOfBlock.keys() :
397                  jobsOfBlock[block] = []
398 <            
398 >
399              if self.eventsbyblock.has_key(block) :
400                  numEventsInBlock = self.eventsbyblock[block]
401                  common.logger.debug(5,'Events in Block File '+str(numEventsInBlock))
402 <            
402 >
403                  files = self.filesbyblock[block]
404                  numFilesInBlock = len(files)
405                  if (numFilesInBlock <= 0):
# Line 477 | Line 407 | class Cmssw(JobType):
407                  fileCount = 0
408  
409                  # ---- New block => New job ---- #
410 <                parString = "\\{"
410 >                parString = ""
411                  # counter for number of events in files currently worked on
412                  filesEventCount = 0
413                  # flag if next while loop should touch new file
414                  newFile = 1
415                  # job event counter
416                  jobSkipEventCount = 0
417 <            
417 >
418                  # ---- Iterate over the files in the block until we've met the requested ---- #
419                  # ---- total # of events or we've gone over all the files in this block  ---- #
420 +                pString=''
421                  while ( (eventsRemaining > 0) and (fileCount < numFilesInBlock) and (jobCount < totalNumberOfJobs) ):
422                      file = files[fileCount]
423 +                    if self.useParent:
424 +                        parent = self.parentFiles[file]
425 +                        for f in parent :
426 +                            pString += '\\\"' + f + '\\\"\,'
427 +                        common.logger.debug(6, "File "+str(file)+" has the following parents: "+str(parent))
428 +                        common.logger.write("File "+str(file)+" has the following parents: "+str(parent))
429                      if newFile :
430                          try:
431                              numEventsInFile = self.eventsbyfile[file]
# Line 500 | Line 437 | class Cmssw(JobType):
437                              newFile = 0
438                          except KeyError:
439                              common.logger.message("File "+str(file)+" has unknown number of events: skipping")
503                        
440  
441 +                    eventsPerJobRequested = min(eventsPerJobRequested, eventsRemaining)
442                      # if less events in file remain than eventsPerJobRequested
443 <                    if ( filesEventCount - jobSkipEventCount < eventsPerJobRequested ) :
443 >                    if ( filesEventCount - jobSkipEventCount < eventsPerJobRequested):
444                          # if last file in block
445                          if ( fileCount == numFilesInBlock-1 ) :
446                              # end job using last file, use remaining events in block
447                              # close job and touch new file
448                              fullString = parString[:-2]
449 <                            fullString += '\\}'
450 <                            list_of_lists.append([fullString,str(-1),str(jobSkipEventCount)])
449 >                            if self.useParent:
450 >                                fullParentString = pString[:-2]
451 >                                list_of_lists.append([fullString,fullParentString,str(-1),str(jobSkipEventCount)])
452 >                            else:
453 >                                list_of_lists.append([fullString,str(-1),str(jobSkipEventCount)])
454                              common.logger.debug(3,"Job "+str(jobCount+1)+" can run over "+str(filesEventCount - jobSkipEventCount)+" events (last file in block).")
455                              self.jobDestination.append(blockSites[block])
456                              common.logger.debug(5,"Job "+str(jobCount+1)+" Destination: "+str(self.jobDestination[jobCount]))
# Line 522 | Line 462 | class Cmssw(JobType):
462                              eventsRemaining = eventsRemaining - filesEventCount + jobSkipEventCount
463                              jobSkipEventCount = 0
464                              # reset file
465 <                            parString = "\\{"
465 >                            pString = ""
466 >                            parString = ""
467                              filesEventCount = 0
468                              newFile = 1
469                              fileCount += 1
# Line 534 | Line 475 | class Cmssw(JobType):
475                      elif ( filesEventCount - jobSkipEventCount == eventsPerJobRequested ) :
476                          # close job and touch new file
477                          fullString = parString[:-2]
478 <                        fullString += '\\}'
479 <                        list_of_lists.append([fullString,str(eventsPerJobRequested),str(jobSkipEventCount)])
478 >                        if self.useParent:
479 >                            fullParentString = pString[:-2]
480 >                            list_of_lists.append([fullString,fullParentString,str(eventsPerJobRequested),str(jobSkipEventCount)])
481 >                        else:
482 >                            list_of_lists.append([fullString,str(eventsPerJobRequested),str(jobSkipEventCount)])
483                          common.logger.debug(3,"Job "+str(jobCount+1)+" can run over "+str(eventsPerJobRequested)+" events.")
484                          self.jobDestination.append(blockSites[block])
485                          common.logger.debug(5,"Job "+str(jobCount+1)+" Destination: "+str(self.jobDestination[jobCount]))
# Line 546 | Line 490 | class Cmssw(JobType):
490                          eventsRemaining = eventsRemaining - eventsPerJobRequested
491                          jobSkipEventCount = 0
492                          # reset file
493 <                        parString = "\\{"
493 >                        pString = ""
494 >                        parString = ""
495                          filesEventCount = 0
496                          newFile = 1
497                          fileCount += 1
498 <                        
498 >
499                      # if more events in file remain than eventsPerJobRequested
500                      else :
501                          # close job but don't touch new file
502                          fullString = parString[:-2]
503 <                        fullString += '\\}'
504 <                        list_of_lists.append([fullString,str(eventsPerJobRequested),str(jobSkipEventCount)])
503 >                        if self.useParent:
504 >                            fullParentString = pString[:-2]
505 >                            list_of_lists.append([fullString,fullParentString,str(eventsPerJobRequested),str(jobSkipEventCount)])
506 >                        else:
507 >                            list_of_lists.append([fullString,str(eventsPerJobRequested),str(jobSkipEventCount)])
508                          common.logger.debug(3,"Job "+str(jobCount+1)+" can run over "+str(eventsPerJobRequested)+" events.")
509                          self.jobDestination.append(blockSites[block])
510                          common.logger.debug(5,"Job "+str(jobCount+1)+" Destination: "+str(self.jobDestination[jobCount]))
# Line 570 | Line 518 | class Cmssw(JobType):
518                          jobSkipEventCount = eventsPerJobRequested - (filesEventCount - jobSkipEventCount - self.eventsbyfile[file])
519                          # remove all but the last file
520                          filesEventCount = self.eventsbyfile[file]
521 <                        parString = "\\{"
522 <                        parString += '\\\"' + file + '\\\"\,'
521 >                        if self.useParent:
522 >                            for f in parent : pString += '\\\"' + f + '\\\"\,'
523 >                        parString = '\\\"' + file + '\\\"\,'
524                      pass # END if
525                  pass # END while (iterate over files in the block)
526          pass # END while (iterate over blocks in the dataset)
# Line 579 | Line 528 | class Cmssw(JobType):
528          if (eventsRemaining > 0 and jobCount < totalNumberOfJobs ):
529              common.logger.message("Could not run on all requested events because some blocks not hosted at allowed sites.")
530          common.logger.message(str(jobCount)+" job(s) can run on "+str(totalEventCount)+" events.\n")
531 <        
531 >
532          # screen output
533          screenOutput = "List of jobs and available destination sites:\n\n"
534  
# Line 591 | Line 540 | class Cmssw(JobType):
540          for block in blocks:
541              if block in jobsOfBlock.keys() :
542                  blockCounter += 1
543 <                screenOutput += "Block %5i: jobs %20s: sites: %s\n" % (blockCounter,spanRanges(jobsOfBlock[block]),','.join(self.blackWhiteListParser.checkWhiteList(self.blackWhiteListParser.checkBlackList(blockSites[block],block),block)))
543 >                screenOutput += "Block %5i: jobs %20s: sites: %s\n" % (blockCounter,spanRanges(jobsOfBlock[block]),
544 >                    ','.join(self.blackWhiteListParser.checkWhiteList(self.blackWhiteListParser.checkBlackList(blockSites[block],block),block)))
545                  if len(self.blackWhiteListParser.checkWhiteList(self.blackWhiteListParser.checkBlackList(blockSites[block],block),block)) == 0:
546 <                    noSiteBlock.append( spanRanges(jobsOfBlock[block]) )
546 >                    noSiteBlock.append( spanRanges(jobsOfBlock[block]) )
547                      bloskNoSite.append( blockCounter )
548 <        
548 >
549          common.logger.message(screenOutput)
550          if len(noSiteBlock) > 0 and len(bloskNoSite) > 0:
551              msg = 'WARNING: No sites are hosting any part of data for block:\n                '
# Line 611 | Line 561 | class Cmssw(JobType):
561              for range_jobs in noSiteBlock:
562                  msg += str(range_jobs) + virgola
563              msg += '\n               will not be submitted and this block of data can not be analyzed!\n'
564 +            if self.cfg_params.has_key('EDG.se_white_list'):
565 +                msg += 'WARNING: SE White List: '+self.cfg_params['EDG.se_white_list']+'\n'
566 +                msg += '(Hint: By whitelisting you force the job to run at this particular site(s).\n'
567 +                msg += 'Please check if the dataset is available at this site!)\n'
568 +            if self.cfg_params.has_key('EDG.ce_white_list'):
569 +                msg += 'WARNING: CE White List: '+self.cfg_params['EDG.ce_white_list']+'\n'
570 +                msg += '(Hint: By whitelisting you force the job to run at this particular site(s).\n'
571 +                msg += 'Please check if the dataset is available at this site!)\n'
572 +
573              common.logger.message(msg)
574  
575          self.list_of_args = list_of_lists
# Line 621 | Line 580 | class Cmssw(JobType):
580          Perform job splitting based on number of event per job
581          """
582          common.logger.debug(5,'Splitting per events')
583 <        
584 <        if (self.selectEventsPerJob):
583 >
584 >        if (self.selectEventsPerJob):
585              common.logger.message('Required '+str(self.eventsPerJob)+' events per job ')
586          if (self.selectNumberOfJobs):
587              common.logger.message('Required '+str(self.theNumberOfJobs)+' jobs in total ')
# Line 636 | Line 595 | class Cmssw(JobType):
595          if (self.selectEventsPerJob):
596              if (self.selectTotalNumberEvents):
597                  self.total_number_of_jobs = int(self.total_number_of_events/self.eventsPerJob)
598 <            elif(self.selectNumberOfJobs) :  
598 >            elif(self.selectNumberOfJobs) :
599                  self.total_number_of_jobs =self.theNumberOfJobs
600 <                self.total_number_of_events =int(self.theNumberOfJobs*self.eventsPerJob)
600 >                self.total_number_of_events =int(self.theNumberOfJobs*self.eventsPerJob)
601  
602          elif (self.selectNumberOfJobs) :
603              self.total_number_of_jobs = self.theNumberOfJobs
604              self.eventsPerJob = int(self.total_number_of_events/self.total_number_of_jobs)
605 <
605 >
606          common.logger.debug(5,'N jobs  '+str(self.total_number_of_jobs))
607  
608          # is there any remainder?
# Line 659 | Line 618 | class Cmssw(JobType):
618          self.list_of_args = []
619          for i in range(self.total_number_of_jobs):
620              ## Since there is no input, any site is good
621 <           # self.jobDestination.append(["Any"])
663 <            self.jobDestination.append([""]) #must be empty to write correctly the xml
621 >            self.jobDestination.append([""]) #must be empty to write correctly the xml
622              args=[]
623              if (self.firstRun):
624 <                    ## pythia first run
667 <                #self.list_of_args.append([(str(self.firstRun)+str(i))])
624 >                ## pythia first run
625                  args.append(str(self.firstRun)+str(i))
669            else:
670                ## no first run
671                #self.list_of_args.append([str(i)])
672                args.append(str(i))
673            if (self.sourceSeed):
674                args.append(str(self.sourceSeed)+str(i))
675                if (self.sourceSeedVtx):
676                    ## + vtx random seed
677                    args.append(str(self.sourceSeedVtx)+str(i))
678                if (self.sourceSeedG4):
679                    ## + G4 random seed
680                    args.append(str(self.sourceSeedG4)+str(i))
681                if (self.sourceSeedMix):    
682                    ## + Mix random seed
683                    args.append(str(self.sourceSeedMix)+str(i))
684                pass
685            pass
626              self.list_of_args.append(args)
687        pass
688            
689        # print self.list_of_args
627  
628          return
629  
630  
631 <    def jobSplittingForScript(self):#CarlosDaniele
631 >    def jobSplittingForScript(self):
632          """
633          Perform job splitting based on number of job
634          """
# Line 707 | Line 644 | class Cmssw(JobType):
644          # argument is seed number.$i
645          self.list_of_args = []
646          for i in range(self.total_number_of_jobs):
710            ## Since there is no input, any site is good
711           # self.jobDestination.append(["Any"])
647              self.jobDestination.append([""])
713            ## no random seed
648              self.list_of_args.append([str(i)])
649          return
650  
651 <    def split(self, jobParams):
652 <
719 <        common.jobDB.load()
720 <        #### Fabio
651 >    def split(self, jobParams,firstJobID):
652 >
653          njobs = self.total_number_of_jobs
654          arglist = self.list_of_args
655          # create the empty structure
656          for i in range(njobs):
657              jobParams.append("")
726        
727        for job in range(njobs):
728            jobParams[job] = arglist[job]
729            # print str(arglist[job])
730            # print jobParams[job]
731            common.jobDB.setArguments(job, jobParams[job])
732            common.logger.debug(5,"Job "+str(job)+" Destination: "+str(self.jobDestination[job]))
733            common.jobDB.setDestination(job, self.jobDestination[job])
658  
659 <        common.jobDB.save()
659 >        listID=[]
660 >        listField=[]
661 >        for id in range(njobs):
662 >            job = id + int(firstJobID)
663 >            jobParams[id] = arglist[id]
664 >            listID.append(job+1)
665 >            job_ToSave ={}
666 >            concString = ' '
667 >            argu=''
668 >            if len(jobParams[id]):
669 >                argu +=   concString.join(jobParams[id] )
670 >            job_ToSave['arguments']= str(job+1)+' '+argu
671 >            job_ToSave['dlsDestination']= self.jobDestination[id]
672 >            listField.append(job_ToSave)
673 >            msg="Job "+str(job)+" Arguments:   "+str(job+1)+" "+argu+"\n"  \
674 >            +"                     Destination: "+str(self.jobDestination[id])
675 >            common.logger.debug(5,msg)
676 >        common._db.updateJob_(listID,listField)
677 >        self.argsList = (len(jobParams[0])+1)
678 >
679          return
680 <    
738 <    def getJobTypeArguments(self, nj, sched):
739 <        result = ''
740 <        for i in common.jobDB.arguments(nj):
741 <            result=result+str(i)+" "
742 <        return result
743 <  
680 >
681      def numberOfJobs(self):
745        # Fabio
682          return self.total_number_of_jobs
683  
684      def getTarBall(self, exe):
685          """
686          Return the TarBall with lib and exe
687          """
752        
753        # if it exist, just return it
754        #
755        # Marco. Let's start to use relative path for Boss XML files
756        #
688          self.tgzNameWithPath = common.work_space.pathForTgz()+'share/'+self.tgz_name
689          if os.path.exists(self.tgzNameWithPath):
690              return self.tgzNameWithPath
# Line 767 | Line 698 | class Cmssw(JobType):
698  
699          # First of all declare the user Scram area
700          swArea = self.scram.getSWArea_()
770        #print "swArea = ", swArea
771        # swVersion = self.scram.getSWVersion()
772        # print "swVersion = ", swVersion
701          swReleaseTop = self.scram.getReleaseTop_()
702 <        #print "swReleaseTop = ", swReleaseTop
775 <        
702 >
703          ## check if working area is release top
704          if swReleaseTop == '' or swArea == swReleaseTop:
705 +            common.logger.debug(3,"swArea = "+swArea+" swReleaseTop ="+swReleaseTop)
706              return
707  
708          import tarfile
# Line 785 | Line 713 | class Cmssw(JobType):
713                  exeWithPath = self.scram.findFile_(executable)
714                  if ( not exeWithPath ):
715                      raise CrabException('User executable '+executable+' not found')
716 <    
716 >
717                  ## then check if it's private or not
718                  if exeWithPath.find(swReleaseTop) == -1:
719                      # the exe is private, so we must ship
# Line 801 | Line 729 | class Cmssw(JobType):
729                  else:
730                      # the exe is from release, we'll find it on WN
731                      pass
732 <    
732 >
733              ## Now get the libraries: only those in local working area
734              libDir = 'lib'
735              lib = swArea+'/' +libDir
736              common.logger.debug(5,"lib "+lib+" to be tarred")
737              if os.path.exists(lib):
738                  tar.add(lib,libDir)
739 <    
739 >
740              ## Now check if module dir is present
741              moduleDir = 'module'
742              module = swArea + '/' + moduleDir
# Line 816 | Line 744 | class Cmssw(JobType):
744                  tar.add(module,moduleDir)
745  
746              ## Now check if any data dir(s) is present
747 <            swAreaLen=len(swArea)
748 <            for root, dirs, files in os.walk(swArea):
749 <                if "data" in dirs:
750 <                    common.logger.debug(5,"data "+root+"/data"+" to be tarred")
751 <                    tar.add(root+"/data",root[swAreaLen:]+"/data")
752 <
753 <            ## Add ProdAgent dir to tar
754 <            paDir = 'ProdAgentApi'
755 <            pa = os.environ['CRABDIR'] + '/' + 'ProdAgentApi'
756 <            if os.path.isdir(pa):
757 <                tar.add(pa,paDir)
758 <
759 <            ### FEDE FOR DBS PUBLICATION
760 <            ## Add PRODCOMMON dir to tar
761 <            prodcommonDir = 'ProdCommon'
762 <            prodcommonPath = os.environ['CRABDIR'] + '/' + 'ProdCommon'
763 <            if os.path.isdir(prodcommonPath):
764 <                tar.add(prodcommonPath,prodcommonDir)
765 <            #############################    
766 <        
747 >            self.dataExist = False
748 >            todo_list = [(i, i) for i in  os.listdir(swArea+"/src")]
749 >            while len(todo_list):
750 >                entry, name = todo_list.pop()
751 >                if name.startswith('crab_0_') or  name.startswith('.') or name == 'CVS':
752 >                    continue
753 >                if os.path.isdir(swArea+"/src/"+entry):
754 >                    entryPath = entry + '/'
755 >                    todo_list += [(entryPath + i, i) for i in  os.listdir(swArea+"/src/"+entry)]
756 >                    if name == 'data':
757 >                        self.dataExist=True
758 >                        common.logger.debug(5,"data "+entry+" to be tarred")
759 >                        tar.add(swArea+"/src/"+entry,"src/"+entry)
760 >                    pass
761 >                pass
762 >
763 >            ### CMSSW ParameterSet
764 >            if not self.pset is None:
765 >                cfg_file = common.work_space.jobDir()+self.configFilename()
766 >                tar.add(cfg_file,self.configFilename())
767 >                common.logger.debug(5,"File added to "+self.tgzNameWithPath+" : "+str(tar.getnames()))
768 >
769 >
770 >            ## Add ProdCommon dir to tar
771 >            prodcommonDir = './'
772 >            prodcommonPath = os.environ['CRABDIR'] + '/' + 'external/'
773 >            neededStuff = ['ProdCommon/__init__.py','ProdCommon/FwkJobRep', 'ProdCommon/CMSConfigTools','ProdCommon/Core','ProdCommon/MCPayloads', 'IMProv']
774 >            for file in neededStuff:
775 >                tar.add(prodcommonPath+file,prodcommonDir+file)
776              common.logger.debug(5,"Files added to "+self.tgzNameWithPath+" : "+str(tar.getnames()))
777 +
778 +            ##### ML stuff
779 +            ML_file_list=['report.py', 'DashboardAPI.py', 'Logger.py', 'ProcInfo.py', 'apmon.py']
780 +            path=os.environ['CRABDIR'] + '/python/'
781 +            for file in ML_file_list:
782 +                tar.add(path+file,file)
783 +            common.logger.debug(5,"Files added to "+self.tgzNameWithPath+" : "+str(tar.getnames()))
784 +
785 +            ##### Utils
786 +            Utils_file_list=['parseCrabFjr.py','writeCfg.py', 'fillCrabFjr.py']
787 +            for file in Utils_file_list:
788 +                tar.add(path+file,file)
789 +            common.logger.debug(5,"Files added to "+self.tgzNameWithPath+" : "+str(tar.getnames()))
790 +
791 +            ##### AdditionalFiles
792 +            for file in self.additional_inbox_files:
793 +                tar.add(file,string.split(file,'/')[-1])
794 +            common.logger.debug(5,"Files added to "+self.tgzNameWithPath+" : "+str(tar.getnames()))
795 +
796              tar.close()
797 <        except :
798 <            raise CrabException('Could not create tar-ball')
797 >        except IOError:
798 >            raise CrabException('Could not create tar-ball '+self.tgzNameWithPath)
799 >        except tarfile.TarError:
800 >            raise CrabException('Could not create tar-ball '+self.tgzNameWithPath)
801  
802          ## check for tarball size
803          tarballinfo = os.stat(self.tgzNameWithPath)
# Line 847 | Line 805 | class Cmssw(JobType):
805              raise CrabException('Input sandbox size of ' + str(float(tarballinfo.st_size)/1024.0/1024.0) + ' MB is larger than the allowed ' + str(self.MaxTarBallSize) + ' MB input sandbox limit and not supported by the used GRID submission system. Please make sure that no unnecessary files are in all data directories in your local CMSSW project area as they are automatically packed into the input sandbox.')
806  
807          ## create tar-ball with ML stuff
850        self.MLtgzfile =  common.work_space.pathForTgz()+'share/MLfiles.tgz'
851        try:
852            tar = tarfile.open(self.MLtgzfile, "w:gz")
853            path=os.environ['CRABDIR'] + '/python/'
854            for file in ['report.py', 'DashboardAPI.py', 'Logger.py', 'ProcInfo.py', 'apmon.py', 'parseCrabFjr.py']:
855                tar.add(path+file,file)
856            common.logger.debug(5,"Files added to "+self.MLtgzfile+" : "+str(tar.getnames()))
857            tar.close()
858        except :
859            raise CrabException('Could not create ML files tar-ball')
860        
861        return
862        
863    def additionalInputFileTgz(self):
864        """
865        Put all additional files into a tar ball and return its name
866        """
867        import tarfile
868        tarName=  common.work_space.pathForTgz()+'share/'+self.additional_tgz_name
869        tar = tarfile.open(tarName, "w:gz")
870        for file in self.additional_inbox_files:
871            tar.add(file,string.split(file,'/')[-1])
872        common.logger.debug(5,"Files added to "+self.additional_tgz_name+" : "+str(tar.getnames()))
873        tar.close()
874        return tarName
808  
809 <    def wsSetupEnvironment(self, nj):
809 >    def wsSetupEnvironment(self, nj=0):
810          """
811          Returns part of a job script which prepares
812          the execution environment for the job 'nj'.
813          """
814 +        if (self.CMSSW_major >= 2 and self.CMSSW_minor >= 1) or (self.CMSSW_major >= 3):
815 +            psetName = 'pset.py'
816 +        else:
817 +            psetName = 'pset.cfg'
818          # Prepare JobType-independent part
819 <        txt = ''
820 <  
821 <        ## OLI_Daniele at this level  middleware already known
885 <
886 <        txt += 'if [ $middleware == LCG ]; then \n'
887 <        txt += '    echo "### First set SCRAM ARCH and BUILD_ARCH to ' + self.executable_arch + ' ###"\n'
888 <        txt += '    export SCRAM_ARCH='+self.executable_arch+'\n'
889 <        txt += '    export BUILD_ARCH='+self.executable_arch+'\n'
819 >        txt = '\n#Written by cms_cmssw::wsSetupEnvironment\n'
820 >        txt += 'echo ">>> setup environment"\n'
821 >        txt += 'if [ $middleware == LCG ]; then \n'
822          txt += self.wsSetupCMSLCGEnvironment_()
823          txt += 'elif [ $middleware == OSG ]; then\n'
824          txt += '    WORKING_DIR=`/bin/mktemp  -d $OSG_WN_TMP/cms_XXXXXXXXXXXX`\n'
825 <        txt += '    echo "Created working directory: $WORKING_DIR"\n'
826 <        txt += '    if [ ! -d $WORKING_DIR ] ;then\n'
827 <        txt += '        echo "SET_CMS_ENV 10016 ==> OSG $WORKING_DIR could not be created on WN `hostname`"\n'
828 <        txt += '    echo "JOB_EXIT_STATUS = 10016"\n'
897 <        txt += '    echo "JobExitCode=10016" | tee -a $RUNTIME_AREA/$repo\n'
898 <        txt += '    dumpStatus $RUNTIME_AREA/$repo\n'
899 <        txt += '        rm -f $RUNTIME_AREA/$repo \n'
900 <        txt += '        echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
901 <        txt += '        echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
902 <        txt += '        exit 1\n'
825 >        txt += '    if [ ! $? == 0 ] ;then\n'
826 >        txt += '        echo "ERROR ==> OSG $WORKING_DIR could not be created on WN `hostname`"\n'
827 >        txt += '        job_exit_code=10016\n'
828 >        txt += '        func_exit\n'
829          txt += '    fi\n'
830 +        txt += '    echo ">>> Created working directory: $WORKING_DIR"\n'
831          txt += '\n'
832          txt += '    echo "Change to working directory: $WORKING_DIR"\n'
833          txt += '    cd $WORKING_DIR\n'
834 <        txt += self.wsSetupCMSOSGEnvironment_()
835 <        txt += '    echo "### Set SCRAM ARCH to ' + self.executable_arch + ' ###"\n'
909 <        txt += '    export SCRAM_ARCH='+self.executable_arch+'\n'
834 >        txt += '    echo ">>> current directory (WORKING_DIR): $WORKING_DIR"\n'
835 >        txt += self.wsSetupCMSOSGEnvironment_()
836          txt += 'fi\n'
837  
838          # Prepare JobType-specific part
839          scram = self.scram.commandName()
840          txt += '\n\n'
841 <        txt += 'echo "### SPECIFIC JOB SETUP ENVIRONMENT ###"\n'
841 >        txt += 'echo ">>> specific cmssw setup environment:"\n'
842 >        txt += 'echo "CMSSW_VERSION =  '+self.version+'"\n'
843          txt += scram+' project CMSSW '+self.version+'\n'
844          txt += 'status=$?\n'
845          txt += 'if [ $status != 0 ] ; then\n'
846 <        txt += '   echo "SET_EXE_ENV 10034 ==>ERROR CMSSW '+self.version+' not found on `hostname`" \n'
847 <        txt += '   echo "JOB_EXIT_STATUS = 10034"\n'
848 <        txt += '   echo "JobExitCode=10034" | tee -a $RUNTIME_AREA/$repo\n'
922 <        txt += '   dumpStatus $RUNTIME_AREA/$repo\n'
923 <        txt += '   rm -f $RUNTIME_AREA/$repo \n'
924 <        txt += '   echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
925 <        txt += '   echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
926 <        ## OLI_Daniele
927 <        txt += '    if [ $middleware == OSG ]; then \n'
928 <        txt += '        echo "Remove working directory: $WORKING_DIR"\n'
929 <        txt += '        cd $RUNTIME_AREA\n'
930 <        txt += '        /bin/rm -rf $WORKING_DIR\n'
931 <        txt += '        if [ -d $WORKING_DIR ] ;then\n'
932 <        txt += '            echo "SET_CMS_ENV 10018 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after CMSSW CMSSW_0_6_1 not found on `hostname`"\n'
933 <        txt += '            echo "JOB_EXIT_STATUS = 10018"\n'
934 <        txt += '            echo "JobExitCode=10018" | tee -a $RUNTIME_AREA/$repo\n'
935 <        txt += '            dumpStatus $RUNTIME_AREA/$repo\n'
936 <        txt += '            rm -f $RUNTIME_AREA/$repo \n'
937 <        txt += '            echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
938 <        txt += '            echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
939 <        txt += '        fi\n'
940 <        txt += '    fi \n'
941 <        txt += '   exit 1 \n'
846 >        txt += '    echo "ERROR ==> CMSSW '+self.version+' not found on `hostname`" \n'
847 >        txt += '    job_exit_code=10034\n'
848 >        txt += '    func_exit\n'
849          txt += 'fi \n'
943        txt += 'echo "CMSSW_VERSION =  '+self.version+'"\n'
850          txt += 'cd '+self.version+'\n'
945        ########## FEDE FOR DBS2 ######################
851          txt += 'SOFTWARE_DIR=`pwd`\n'
852 <        txt += 'echo SOFTWARE_DIR=$SOFTWARE_DIR \n'
948 <        ###############################################
949 <        ### needed grep for bug in scramv1 ###
950 <        txt += scram+' runtime -sh\n'
852 >        txt += 'echo ">>> current directory (SOFTWARE_DIR): $SOFTWARE_DIR" \n'
853          txt += 'eval `'+scram+' runtime -sh | grep -v SCRAMRT_LSB_JOBNAME`\n'
854 <        txt += 'echo $PATH\n'
855 <
854 >        txt += 'if [ $? != 0 ] ; then\n'
855 >        txt += '    echo "ERROR ==> Problem with the command: "\n'
856 >        txt += '    echo "eval \`'+scram+' runtime -sh | grep -v SCRAMRT_LSB_JOBNAME \` at `hostname`"\n'
857 >        txt += '    job_exit_code=10034\n'
858 >        txt += '    func_exit\n'
859 >        txt += 'fi \n'
860          # Handle the arguments:
861          txt += "\n"
862          txt += "## number of arguments (first argument always jobnumber)\n"
863          txt += "\n"
864 < #        txt += "narg=$#\n"
959 <        txt += "if [ $nargs -lt 2 ]\n"
864 >        txt += "if [ $nargs -lt "+str(self.argsList)+" ]\n"
865          txt += "then\n"
866 <        txt += "    echo 'SET_EXE_ENV 1 ==> ERROR Too few arguments' +$nargs+ \n"
867 <        txt += '    echo "JOB_EXIT_STATUS = 50113"\n'
868 <        txt += '    echo "JobExitCode=50113" | tee -a $RUNTIME_AREA/$repo\n'
964 <        txt += '    dumpStatus $RUNTIME_AREA/$repo\n'
965 <        txt += '    rm -f $RUNTIME_AREA/$repo \n'
966 <        txt += '    echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
967 <        txt += '    echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
968 <        ## OLI_Daniele
969 <        txt += '    if [ $middleware == OSG ]; then \n'
970 <        txt += '        echo "Remove working directory: $WORKING_DIR"\n'
971 <        txt += '        cd $RUNTIME_AREA\n'
972 <        txt += '        /bin/rm -rf $WORKING_DIR\n'
973 <        txt += '        if [ -d $WORKING_DIR ] ;then\n'
974 <        txt += '            echo "SET_EXE_ENV 50114 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after Too few arguments for CRAB job wrapper"\n'
975 <        txt += '            echo "JOB_EXIT_STATUS = 50114"\n'
976 <        txt += '            echo "JobExitCode=50114" | tee -a $RUNTIME_AREA/$repo\n'
977 <        txt += '            dumpStatus $RUNTIME_AREA/$repo\n'
978 <        txt += '            rm -f $RUNTIME_AREA/$repo \n'
979 <        txt += '            echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
980 <        txt += '            echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
981 <        txt += '        fi\n'
982 <        txt += '    fi \n'
983 <        txt += "    exit 1\n"
866 >        txt += "    echo 'ERROR ==> Too few arguments' +$nargs+ \n"
867 >        txt += '    job_exit_code=50113\n'
868 >        txt += "    func_exit\n"
869          txt += "fi\n"
870          txt += "\n"
871  
872          # Prepare job-specific part
873          job = common.job_list[nj]
874 <        ### FEDE FOR DBS OUTPUT PUBLICATION
990 <        if (self.datasetPath):
874 >        if (self.datasetPath):
875              txt += '\n'
876              txt += 'DatasetPath='+self.datasetPath+'\n'
877  
878              datasetpath_split = self.datasetPath.split("/")
879 <            
879 >            ### FEDE FOR NEW LFN ###
880 >            self.primaryDataset = datasetpath_split[1]
881 >            ########################
882              txt += 'PrimaryDataset='+datasetpath_split[1]+'\n'
883              txt += 'DataTier='+datasetpath_split[2]+'\n'
998            #txt += 'ProcessedDataset='+datasetpath_split[3]+'\n'
884              txt += 'ApplicationFamily=cmsRun\n'
885  
886          else:
887              txt += 'DatasetPath=MCDataTier\n'
888 +            ### FEDE FOR NEW LFN ###
889 +            self.primaryDataset = 'null'
890 +            ########################
891              txt += 'PrimaryDataset=null\n'
892              txt += 'DataTier=null\n'
1005            #txt += 'ProcessedDataset=null\n'
893              txt += 'ApplicationFamily=MCDataTier\n'
894 <        if self.pset != None: #CarlosDaniele
894 >        if self.pset != None:
895              pset = os.path.basename(job.configFilename())
896              txt += '\n'
897              txt += 'cp  $RUNTIME_AREA/'+pset+' .\n'
898              if (self.datasetPath): # standard job
899 <                #txt += 'InputFiles=$2\n'
900 <                txt += 'InputFiles=${args[1]}\n'
901 <                txt += 'MaxEvents=${args[2]}\n'
902 <                txt += 'SkipEvents=${args[3]}\n'
899 >                txt += 'InputFiles=${args[1]}; export InputFiles\n'
900 >                if (self.useParent):
901 >                    txt += 'ParentFiles=${args[2]}; export ParentFiles\n'
902 >                    txt += 'MaxEvents=${args[3]}; export MaxEvents\n'
903 >                    txt += 'SkipEvents=${args[4]}; export SkipEvents\n'
904 >                else:
905 >                    txt += 'MaxEvents=${args[2]}; export MaxEvents\n'
906 >                    txt += 'SkipEvents=${args[3]}; export SkipEvents\n'
907                  txt += 'echo "Inputfiles:<$InputFiles>"\n'
908 <                txt += 'sed "s#{\'INPUT\'}#$InputFiles#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
908 >                if (self.useParent): txt += 'echo "ParentFiles:<$ParentFiles>"\n'
909                  txt += 'echo "MaxEvents:<$MaxEvents>"\n'
1019                txt += 'sed "s#INPUTMAXEVENTS#$MaxEvents#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
910                  txt += 'echo "SkipEvents:<$SkipEvents>"\n'
1021                txt += 'sed "s#INPUTSKIPEVENTS#$SkipEvents#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
911              else:  # pythia like job
912 <                seedIndex=1
912 >                txt += 'PreserveSeeds='  + ','.join(self.preserveSeeds)  + '; export PreserveSeeds\n'
913 >                txt += 'IncrementSeeds=' + ','.join(self.incrementSeeds) + '; export IncrementSeeds\n'
914 >                txt += 'echo "PreserveSeeds: <$PreserveSeeds>"\n'
915 >                txt += 'echo "IncrementSeeds:<$IncrementSeeds>"\n'
916                  if (self.firstRun):
917 <                    txt += 'FirstRun=${args['+str(seedIndex)+']}\n'
917 >                    txt += 'FirstRun=${args[1]}; export FirstRun\n'
918                      txt += 'echo "FirstRun: <$FirstRun>"\n'
1027                    txt += 'sed "s#\<INPUTFIRSTRUN\>#$FirstRun#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1028                    seedIndex=seedIndex+1
919  
920 <                if (self.sourceSeed):
1031 <                    txt += 'Seed=${args['+str(seedIndex)+']}\n'
1032 <                    txt += 'sed "s#\<INPUT\>#$Seed#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1033 <                    seedIndex=seedIndex+1
1034 <                    ## the following seeds are not always present
1035 <                    if (self.sourceSeedVtx):
1036 <                        txt += 'VtxSeed=${args['+str(seedIndex)+']}\n'
1037 <                        txt += 'echo "VtxSeed: <$VtxSeed>"\n'
1038 <                        txt += 'sed "s#\<INPUTVTX\>#$VtxSeed#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1039 <                        seedIndex += 1
1040 <                    if (self.sourceSeedG4):
1041 <                        txt += 'G4Seed=${args['+str(seedIndex)+']}\n'
1042 <                        txt += 'echo "G4Seed: <$G4Seed>"\n'
1043 <                        txt += 'sed "s#\<INPUTG4\>#$G4Seed#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1044 <                        seedIndex += 1
1045 <                    if (self.sourceSeedMix):
1046 <                        txt += 'mixSeed=${args['+str(seedIndex)+']}\n'
1047 <                        txt += 'echo "MixSeed: <$mixSeed>"\n'
1048 <                        txt += 'sed "s#\<INPUTMIX\>#$mixSeed#" '+pset+' > tmp && mv -f tmp '+pset+'\n'
1049 <                        seedIndex += 1
1050 <                    pass
1051 <                pass
1052 <            txt += 'mv -f '+pset+' pset.cfg\n'
920 >            txt += 'mv -f ' + pset + ' ' + psetName + '\n'
921  
1054        if len(self.additional_inbox_files) > 0:
1055            txt += 'if [ -e $RUNTIME_AREA/'+self.additional_tgz_name+' ] ; then\n'
1056            txt += '  tar xzvf $RUNTIME_AREA/'+self.additional_tgz_name+'\n'
1057            txt += 'fi\n'
1058            pass
922  
923 <        if self.pset != None: #CarlosDaniele
924 <            txt += 'echo "### END JOB SETUP ENVIRONMENT ###"\n\n'
1062 <        
1063 <            txt += '\n'
1064 <            txt += 'echo "***** cat pset.cfg *********"\n'
1065 <            txt += 'cat pset.cfg\n'
1066 <            txt += 'echo "****** end pset.cfg ********"\n'
923 >        if self.pset != None:
924 >            # FUTURE: Can simply for 2_1_x and higher
925              txt += '\n'
926 <            ### FEDE FOR DBS OUTPUT PUBLICATION
927 <            txt += 'PSETHASH=`EdmConfigHash < pset.cfg` \n'
926 >            if self.debug_wrapper==True:
927 >                txt += 'echo "***** cat ' + psetName + ' *********"\n'
928 >                txt += 'cat ' + psetName + '\n'
929 >                txt += 'echo "****** end ' + psetName + ' ********"\n'
930 >                txt += '\n'
931 >            if (self.CMSSW_major >= 2 and self.CMSSW_minor >= 1) or (self.CMSSW_major >= 3):
932 >                txt += 'PSETHASH=`edmConfigHash ' + psetName + '` \n'
933 >            else:
934 >                txt += 'PSETHASH=`edmConfigHash < ' + psetName + '` \n'
935              txt += 'echo "PSETHASH = $PSETHASH" \n'
1071            ##############
936              txt += '\n'
1073            # txt += 'echo "***** cat pset1.cfg *********"\n'
1074            # txt += 'cat pset1.cfg\n'
1075            # txt += 'echo "****** end pset1.cfg ********"\n'
937          return txt
938  
939 <    def wsBuildExe(self, nj=0):
939 >    def wsUntarSoftware(self, nj=0):
940          """
941          Put in the script the commands to build an executable
942          or a library.
943          """
944  
945 <        txt = ""
945 >        txt = '\n#Written by cms_cmssw::wsUntarSoftware\n'
946  
947          if os.path.isfile(self.tgzNameWithPath):
948 <            txt += 'echo "tar xzvf $RUNTIME_AREA/'+os.path.basename(self.tgzNameWithPath)+'"\n'
948 >            txt += 'echo ">>> tar xzvf $RUNTIME_AREA/'+os.path.basename(self.tgzNameWithPath)+' :" \n'
949              txt += 'tar xzvf $RUNTIME_AREA/'+os.path.basename(self.tgzNameWithPath)+'\n'
950 +            if  self.debug_wrapper:
951 +                txt += 'ls -Al \n'
952              txt += 'untar_status=$? \n'
953              txt += 'if [ $untar_status -ne 0 ]; then \n'
954 <            txt += '   echo "SET_EXE 1 ==> ERROR Untarring .tgz file failed"\n'
955 <            txt += '   echo "JOB_EXIT_STATUS = $untar_status" \n'
956 <            txt += '   echo "JobExitCode=$untar_status" | tee -a $RUNTIME_AREA/$repo\n'
1094 <            txt += '   if [ $middleware == OSG ]; then \n'
1095 <            txt += '       echo "Remove working directory: $WORKING_DIR"\n'
1096 <            txt += '       cd $RUNTIME_AREA\n'
1097 <            txt += '       /bin/rm -rf $WORKING_DIR\n'
1098 <            txt += '       if [ -d $WORKING_DIR ] ;then\n'
1099 <            txt += '           echo "SET_EXE 50999 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after Untarring .tgz file failed"\n'
1100 <            txt += '           echo "JOB_EXIT_STATUS = 50999"\n'
1101 <            txt += '           echo "JobExitCode=50999" | tee -a $RUNTIME_AREA/$repo\n'
1102 <            txt += '           dumpStatus $RUNTIME_AREA/$repo\n'
1103 <            txt += '           rm -f $RUNTIME_AREA/$repo \n'
1104 <            txt += '           echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1105 <            txt += '           echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1106 <            txt += '       fi\n'
1107 <            txt += '   fi \n'
1108 <            txt += '   \n'
1109 <            txt += '   exit 1 \n'
954 >            txt += '   echo "ERROR ==> Untarring .tgz file failed"\n'
955 >            txt += '   job_exit_code=$untar_status\n'
956 >            txt += '   func_exit\n'
957              txt += 'else \n'
958              txt += '   echo "Successful untar" \n'
959              txt += 'fi \n'
960              txt += '\n'
961 <            txt += 'echo "Include ProdAgentApi and PRODCOMMON in PYTHONPATH"\n'
961 >            txt += 'echo ">>> Include $RUNTIME_AREA in PYTHONPATH:"\n'
962              txt += 'if [ -z "$PYTHONPATH" ]; then\n'
963 <            #### FEDE FOR DBS OUTPUT PUBLICATION
1117 <            txt += '   export PYTHONPATH=$SOFTWARE_DIR/ProdAgentApi:$SOFTWARE_DIR/ProdCommon\n'
1118 <            #txt += '   export PYTHONPATH=`pwd`/ProdAgentApi:`pwd`/ProdCommon\n'
1119 <            #txt += '   export PYTHONPATH=ProdAgentApi\n'
963 >            txt += '   export PYTHONPATH=$RUNTIME_AREA/\n'
964              txt += 'else\n'
965 <            txt += '   export PYTHONPATH=$SOFTWARE_DIR/ProdAgentApi:$SOFTWARE_DIR/ProdCommon:${PYTHONPATH}\n'
1122 <            #txt += '   export PYTHONPATH=`pwd`/ProdAgentApi:`pwd`/ProdCommon:${PYTHONPATH}\n'
1123 <            #txt += '   export PYTHONPATH=ProdAgentApi:${PYTHONPATH}\n'
965 >            txt += '   export PYTHONPATH=$RUNTIME_AREA/:${PYTHONPATH}\n'
966              txt += 'echo "PYTHONPATH=$PYTHONPATH"\n'
1125            ###################  
967              txt += 'fi\n'
968              txt += '\n'
969  
970              pass
971 <        
971 >
972          return txt
973  
974 <    def modifySteeringCards(self, nj):
974 >    def wsBuildExe(self, nj=0):
975          """
976 <        modify the card provided by the user,
977 <        writing a new card into share dir
976 >        Put in the script the commands to build an executable
977 >        or a library.
978          """
979 <        
979 >
980 >        txt = '\n#Written by cms_cmssw::wsBuildExe\n'
981 >        txt += 'echo ">>> moving CMSSW software directories in `pwd`" \n'
982 >
983 >        txt += 'rm -r lib/ module/ \n'
984 >        txt += 'mv $RUNTIME_AREA/lib/ . \n'
985 >        txt += 'mv $RUNTIME_AREA/module/ . \n'
986 >        if self.dataExist == True:
987 >            txt += 'rm -r src/ \n'
988 >            txt += 'mv $RUNTIME_AREA/src/ . \n'
989 >        if len(self.additional_inbox_files)>0:
990 >            for file in self.additional_inbox_files:
991 >                txt += 'mv $RUNTIME_AREA/'+os.path.basename(file)+' . \n'
992 >        # txt += 'mv $RUNTIME_AREA/ProdCommon/ . \n'
993 >        # txt += 'mv $RUNTIME_AREA/IMProv/ . \n'
994 >
995 >        txt += 'echo ">>> Include $RUNTIME_AREA in PYTHONPATH:"\n'
996 >        txt += 'if [ -z "$PYTHONPATH" ]; then\n'
997 >        txt += '   export PYTHONPATH=$RUNTIME_AREA/\n'
998 >        txt += 'else\n'
999 >        txt += '   export PYTHONPATH=$RUNTIME_AREA/:${PYTHONPATH}\n'
1000 >        txt += 'echo "PYTHONPATH=$PYTHONPATH"\n'
1001 >        txt += 'fi\n'
1002 >        txt += '\n'
1003 >
1004 >        return txt
1005 >
1006 >
1007      def executableName(self):
1008 <        if self.scriptExe: #CarlosDaniele
1008 >        if self.scriptExe:
1009              return "sh "
1010          else:
1011              return self.executable
1012  
1013      def executableArgs(self):
1014 +        # FUTURE: This function tests the CMSSW version. Can be simplified as we drop support for old versions
1015          if self.scriptExe:#CarlosDaniele
1016              return   self.scriptExe + " $NJob"
1017          else:
1018 <            # if >= CMSSW_1_5_X, add -e
1019 <            version_array = self.scram.getSWVersion().split('_')
1020 <            major = 0
1021 <            minor = 0
1022 <            try:
1023 <                major = int(version_array[1])
1024 <                minor = int(version_array[2])
1025 <            except:
1157 <                msg = "Cannot parse CMSSW version string: " + "_".join(version_array) + " for major and minor release number!"  
1158 <                raise CrabException(msg)
1159 <            if major >= 1 and minor >= 5 :
1160 <                return " -e -p pset.cfg"
1018 >            ex_args = ""
1019 >            # FUTURE: This tests the CMSSW version. Can remove code as versions deprecated
1020 >            # Framework job report
1021 >            if (self.CMSSW_major >= 1 and self.CMSSW_minor >= 5) or (self.CMSSW_major >= 2):
1022 >                ex_args += " -j $RUNTIME_AREA/crab_fjr_$NJob.xml"
1023 >            # Type of config file
1024 >            if self.CMSSW_major >= 2 :
1025 >                ex_args += " -p pset.py"
1026              else:
1027 <                return " -p pset.cfg"
1027 >                ex_args += " -p pset.cfg"
1028 >            return ex_args
1029  
1030      def inputSandbox(self, nj):
1031          """
1032          Returns a list of filenames to be put in JDL input sandbox.
1033          """
1034          inp_box = []
1169        # # dict added to delete duplicate from input sandbox file list
1170        # seen = {}
1171        ## code
1035          if os.path.isfile(self.tgzNameWithPath):
1036              inp_box.append(self.tgzNameWithPath)
1037 <        if os.path.isfile(self.MLtgzfile):
1038 <            inp_box.append(self.MLtgzfile)
1176 <        ## config
1177 <        if not self.pset is None:
1178 <            inp_box.append(common.work_space.pathForTgz() + 'job/' + self.configFilename())
1179 <        ## additional input files
1180 <        tgz = self.additionalInputFileTgz()
1181 <        inp_box.append(tgz)
1037 >        wrapper = os.path.basename(str(common._db.queryTask('scriptName')))
1038 >        inp_box.append(common.work_space.pathForTgz() +'job/'+ wrapper)
1039          return inp_box
1040  
1041      def outputSandbox(self, nj):
# Line 1189 | Line 1046 | class Cmssw(JobType):
1046  
1047          ## User Declared output files
1048          for out in (self.output_file+self.output_file_sandbox):
1049 <            n_out = nj + 1
1050 <            out_box.append(self.numberFile_(out,str(n_out)))
1049 >            n_out = nj + 1
1050 >            out_box.append(numberFile(out,str(n_out)))
1051          return out_box
1052  
1196    def prepareSteeringCards(self):
1197        """
1198        Make initial modifications of the user's steering card file.
1199        """
1200        return
1053  
1054      def wsRenameOutput(self, nj):
1055          """
1056          Returns part of a job script which renames the produced files.
1057          """
1058  
1059 <        txt = '\n'
1060 <        txt += '# directory content\n'
1061 <        txt += 'ls \n'
1062 <
1063 <        txt += 'output_exit_status=0\n'
1064 <        
1065 <        for fileWithSuffix in (self.output_file_sandbox):
1214 <            output_file_num = self.numberFile_(fileWithSuffix, '$NJob')
1215 <            txt += '\n'
1216 <            txt += '# check output file\n'
1217 <            txt += 'if [ -e ./'+fileWithSuffix+' ] ; then\n'
1218 <            txt += '    mv '+fileWithSuffix+' $RUNTIME_AREA\n'
1219 <            txt += '    cp $RUNTIME_AREA/'+fileWithSuffix+' $RUNTIME_AREA/'+output_file_num+'\n'
1220 <            txt += 'else\n'
1221 <            txt += '    exit_status=60302\n'
1222 <            txt += '    echo "ERROR: Problem with output file '+fileWithSuffix+'"\n'
1223 <            if common.scheduler.boss_scheduler_name == 'condor_g':
1224 <                txt += '    if [ $middleware == OSG ]; then \n'
1225 <                txt += '        echo "prepare dummy output file"\n'
1226 <                txt += '        echo "Processing of job output failed" > $RUNTIME_AREA/'+output_file_num+'\n'
1227 <                txt += '    fi \n'
1228 <            txt += 'fi\n'
1229 <        
1059 >        txt = '\n#Written by cms_cmssw::wsRenameOutput\n'
1060 >        txt += 'echo ">>> current directory (SOFTWARE_DIR): $SOFTWARE_DIR" \n'
1061 >        txt += 'echo ">>> current directory content:"\n'
1062 >        if self.debug_wrapper:
1063 >            txt += 'ls -Al\n'
1064 >        txt += '\n'
1065 >
1066          for fileWithSuffix in (self.output_file):
1067 <            output_file_num = self.numberFile_(fileWithSuffix, '$NJob')
1067 >            output_file_num = numberFile(fileWithSuffix, '$NJob')
1068              txt += '\n'
1069              txt += '# check output file\n'
1070              txt += 'if [ -e ./'+fileWithSuffix+' ] ; then\n'
1071 <            txt += '    mv '+fileWithSuffix+' $RUNTIME_AREA\n'
1072 <            txt += '    cp $RUNTIME_AREA/'+fileWithSuffix+' $RUNTIME_AREA/'+output_file_num+'\n'
1071 >            if (self.copy_data == 1):  # For OSG nodes, file is in $WORKING_DIR, should not be moved to $RUNTIME_AREA
1072 >                txt += '    mv '+fileWithSuffix+' '+output_file_num+'\n'
1073 >                txt += '    ln -s `pwd`/'+output_file_num+' $RUNTIME_AREA/'+fileWithSuffix+'\n'
1074 >            else:
1075 >                txt += '    mv '+fileWithSuffix+' $RUNTIME_AREA/'+output_file_num+'\n'
1076 >                txt += '    ln -s $RUNTIME_AREA/'+output_file_num+' $RUNTIME_AREA/'+fileWithSuffix+'\n'
1077              txt += 'else\n'
1078 <            txt += '    exit_status=60302\n'
1079 <            txt += '    echo "ERROR: Problem with output file '+fileWithSuffix+'"\n'
1080 <            txt += '    echo "JOB_EXIT_STATUS = $exit_status"\n'
1241 <            txt += '    output_exit_status=$exit_status\n'
1242 <            if common.scheduler.boss_scheduler_name == 'condor_g':
1078 >            txt += '    job_exit_code=60302\n'
1079 >            txt += '    echo "WARNING: Output file '+fileWithSuffix+' not found"\n'
1080 >            if common.scheduler.name().upper() == 'CONDOR_G':
1081                  txt += '    if [ $middleware == OSG ]; then \n'
1082                  txt += '        echo "prepare dummy output file"\n'
1083                  txt += '        echo "Processing of job output failed" > $RUNTIME_AREA/'+output_file_num+'\n'
# Line 1247 | Line 1085 | class Cmssw(JobType):
1085              txt += 'fi\n'
1086          file_list = []
1087          for fileWithSuffix in (self.output_file):
1088 <             file_list.append(self.numberFile_(fileWithSuffix, '$NJob'))
1089 <            
1088 >             file_list.append(numberFile(fileWithSuffix, '$NJob'))
1089 >
1090          txt += 'file_list="'+string.join(file_list,' ')+'"\n'
1091 +        txt += '\n'
1092 +        txt += 'echo ">>> current directory (SOFTWARE_DIR): $SOFTWARE_DIR" \n'
1093 +        txt += 'echo ">>> current directory content:"\n'
1094 +        if self.debug_wrapper:
1095 +            txt += 'ls -Al\n'
1096 +        txt += '\n'
1097          txt += 'cd $RUNTIME_AREA\n'
1098 +        txt += 'echo ">>> current directory (RUNTIME_AREA):  $RUNTIME_AREA"\n'
1099          return txt
1100  
1256    def numberFile_(self, file, txt):
1257        """
1258        append _'txt' before last extension of a file
1259        """
1260        p = string.split(file,".")
1261        # take away last extension
1262        name = p[0]
1263        for x in p[1:-1]:
1264            name=name+"."+x
1265        # add "_txt"
1266        if len(p)>1:
1267            ext = p[len(p)-1]
1268            result = name + '_' + txt + "." + ext
1269        else:
1270            result = name + '_' + txt
1271        
1272        return result
1273
1101      def getRequirements(self, nj=[]):
1102          """
1103 <        return job requirements to add to jdl files
1103 >        return job requirements to add to jdl files
1104          """
1105          req = ''
1106          if self.version:
1107              req='Member("VO-cms-' + \
1108                   self.version + \
1109                   '", other.GlueHostApplicationSoftwareRunTimeEnvironment)'
1110 <        ## SL add requirement for OS version only if SL4
1284 <        #reSL4 = re.compile( r'slc4' )
1285 <        if self.executable_arch: # and reSL4.search(self.executable_arch):
1110 >        if self.executable_arch:
1111              req+=' && Member("VO-cms-' + \
1112                   self.executable_arch + \
1113                   '", other.GlueHostApplicationSoftwareRunTimeEnvironment)'
1114  
1115          req = req + ' && (other.GlueHostNetworkAdapterOutboundIP)'
1116 +        if ( common.scheduler.name() == "glitecoll" ) or ( common.scheduler.name() == "glite"):
1117 +            req += ' && other.GlueCEStateStatus == "Production" '
1118  
1119          return req
1120  
1121      def configFilename(self):
1122          """ return the config filename """
1123 <        return self.name()+'.cfg'
1123 >        # FUTURE: Can remove cfg mode for CMSSW >= 2_1_x
1124 >        if (self.CMSSW_major >= 2 and self.CMSSW_minor >= 1) or (self.CMSSW_major >= 3):
1125 >          return self.name()+'.py'
1126 >        else:
1127 >          return self.name()+'.cfg'
1128  
1298    ### OLI_DANIELE
1129      def wsSetupCMSOSGEnvironment_(self):
1130          """
1131          Returns part of a job script which is prepares
1132          the execution environment and which is common for all CMS jobs.
1133          """
1134 <        txt = '\n'
1135 <        txt += '   echo "### SETUP CMS OSG  ENVIRONMENT ###"\n'
1136 <        txt += '   if [ -f $GRID3_APP_DIR/cmssoft/cmsset_default.sh ] ;then\n'
1137 <        txt += '      # Use $GRID3_APP_DIR/cmssoft/cmsset_default.sh to setup cms software\n'
1138 <        txt += '       export SCRAM_ARCH='+self.executable_arch+'\n'
1139 <        txt += '       source $GRID3_APP_DIR/cmssoft/cmsset_default.sh '+self.version+'\n'
1310 <        txt += '   elif [ -f $OSG_APP/cmssoft/cms/cmsset_default.sh ] ;then\n'
1134 >        txt = '\n#Written by cms_cmssw::wsSetupCMSOSGEnvironment_\n'
1135 >        txt += '    echo ">>> setup CMS OSG environment:"\n'
1136 >        txt += '    echo "set SCRAM ARCH to ' + self.executable_arch + '"\n'
1137 >        txt += '    export SCRAM_ARCH='+self.executable_arch+'\n'
1138 >        txt += '    echo "SCRAM_ARCH = $SCRAM_ARCH"\n'
1139 >        txt += '    if [ -f $OSG_APP/cmssoft/cms/cmsset_default.sh ] ;then\n'
1140          txt += '      # Use $OSG_APP/cmssoft/cms/cmsset_default.sh to setup cms software\n'
1141 <        txt += '       export SCRAM_ARCH='+self.executable_arch+'\n'
1142 <        txt += '       source $OSG_APP/cmssoft/cms/cmsset_default.sh '+self.version+'\n'
1143 <        txt += '   else\n'
1144 <        txt += '       echo "SET_CMS_ENV 10020 ==> ERROR $GRID3_APP_DIR/cmssoft/cmsset_default.sh and $OSG_APP/cmssoft/cms/cmsset_default.sh file not found"\n'
1145 <        txt += '       echo "JOB_EXIT_STATUS = 10020"\n'
1146 <        txt += '       echo "JobExitCode=10020" | tee -a $RUNTIME_AREA/$repo\n'
1318 <        txt += '       dumpStatus $RUNTIME_AREA/$repo\n'
1319 <        txt += '       rm -f $RUNTIME_AREA/$repo \n'
1320 <        txt += '       echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1321 <        txt += '       echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1322 <        txt += '       exit 1\n'
1323 <        txt += '\n'
1324 <        txt += '       echo "Remove working directory: $WORKING_DIR"\n'
1325 <        txt += '       cd $RUNTIME_AREA\n'
1326 <        txt += '       /bin/rm -rf $WORKING_DIR\n'
1327 <        txt += '       if [ -d $WORKING_DIR ] ;then\n'
1328 <        txt += '           echo "SET_CMS_ENV 10017 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after $GRID3_APP_DIR/cmssoft/cmsset_default.sh and $OSG_APP/cmssoft/cms/cmsset_default.sh file not found"\n'
1329 <        txt += '           echo "JOB_EXIT_STATUS = 10017"\n'
1330 <        txt += '           echo "JobExitCode=10017" | tee -a $RUNTIME_AREA/$repo\n'
1331 <        txt += '           dumpStatus $RUNTIME_AREA/$repo\n'
1332 <        txt += '           rm -f $RUNTIME_AREA/$repo \n'
1333 <        txt += '           echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1334 <        txt += '           echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1335 <        txt += '       fi\n'
1336 <        txt += '\n'
1337 <        txt += '       exit 1\n'
1338 <        txt += '   fi\n'
1141 >        txt += '        source $OSG_APP/cmssoft/cms/cmsset_default.sh '+self.version+'\n'
1142 >        txt += '    else\n'
1143 >        txt += '        echo "ERROR ==> $OSG_APP/cmssoft/cms/cmsset_default.sh file not found"\n'
1144 >        txt += '        job_exit_code=10020\n'
1145 >        txt += '        func_exit\n'
1146 >        txt += '    fi\n'
1147          txt += '\n'
1148 <        txt += '   echo "SET_CMS_ENV 0 ==> setup cms environment ok"\n'
1149 <        txt += '   echo " END SETUP CMS OSG  ENVIRONMENT "\n'
1148 >        txt += '    echo "==> setup cms environment ok"\n'
1149 >        txt += '    echo "SCRAM_ARCH = $SCRAM_ARCH"\n'
1150  
1151          return txt
1152 <
1345 <    ### OLI_DANIELE
1152 >
1153      def wsSetupCMSLCGEnvironment_(self):
1154          """
1155          Returns part of a job script which is prepares
1156          the execution environment and which is common for all CMS jobs.
1157          """
1158 <        txt  = '   \n'
1159 <        txt += '   echo " ### SETUP CMS LCG  ENVIRONMENT ### "\n'
1160 <        txt += '   if [ ! $VO_CMS_SW_DIR ] ;then\n'
1161 <        txt += '       echo "SET_CMS_ENV 10031 ==> ERROR CMS software dir not found on WN `hostname`"\n'
1162 <        txt += '       echo "JOB_EXIT_STATUS = 10031" \n'
1163 <        txt += '       echo "JobExitCode=10031" | tee -a $RUNTIME_AREA/$repo\n'
1164 <        txt += '       dumpStatus $RUNTIME_AREA/$repo\n'
1165 <        txt += '       rm -f $RUNTIME_AREA/$repo \n'
1166 <        txt += '       echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1167 <        txt += '       echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1168 <        txt += '       exit 1\n'
1169 <        txt += '   else\n'
1170 <        txt += '       echo "Sourcing environment... "\n'
1171 <        txt += '       if [ ! -s $VO_CMS_SW_DIR/cmsset_default.sh ] ;then\n'
1172 <        txt += '           echo "SET_CMS_ENV 10020 ==> ERROR cmsset_default.sh file not found into dir $VO_CMS_SW_DIR"\n'
1173 <        txt += '           echo "JOB_EXIT_STATUS = 10020"\n'
1174 <        txt += '           echo "JobExitCode=10020" | tee -a $RUNTIME_AREA/$repo\n'
1175 <        txt += '           dumpStatus $RUNTIME_AREA/$repo\n'
1176 <        txt += '           rm -f $RUNTIME_AREA/$repo \n'
1177 <        txt += '           echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1178 <        txt += '           echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1179 <        txt += '           exit 1\n'
1180 <        txt += '       fi\n'
1181 <        txt += '       echo "sourcing $VO_CMS_SW_DIR/cmsset_default.sh"\n'
1182 <        txt += '       source $VO_CMS_SW_DIR/cmsset_default.sh\n'
1183 <        txt += '       result=$?\n'
1184 <        txt += '       if [ $result -ne 0 ]; then\n'
1378 <        txt += '           echo "SET_CMS_ENV 10032 ==> ERROR problem sourcing $VO_CMS_SW_DIR/cmsset_default.sh"\n'
1379 <        txt += '           echo "JOB_EXIT_STATUS = 10032"\n'
1380 <        txt += '           echo "JobExitCode=10032" | tee -a $RUNTIME_AREA/$repo\n'
1381 <        txt += '           dumpStatus $RUNTIME_AREA/$repo\n'
1382 <        txt += '           rm -f $RUNTIME_AREA/$repo \n'
1383 <        txt += '           echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1384 <        txt += '           echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1385 <        txt += '           exit 1\n'
1386 <        txt += '       fi\n'
1387 <        txt += '   fi\n'
1388 <        txt += '   \n'
1389 <        txt += '   echo "SET_CMS_ENV 0 ==> setup cms environment ok"\n'
1390 <        txt += '   echo "### END SETUP CMS LCG ENVIRONMENT ###"\n'
1158 >        txt = '\n#Written by cms_cmssw::wsSetupCMSLCGEnvironment_\n'
1159 >        txt += '    echo ">>> setup CMS LCG environment:"\n'
1160 >        txt += '    echo "set SCRAM ARCH and BUILD_ARCH to ' + self.executable_arch + ' ###"\n'
1161 >        txt += '    export SCRAM_ARCH='+self.executable_arch+'\n'
1162 >        txt += '    export BUILD_ARCH='+self.executable_arch+'\n'
1163 >        txt += '    if [ ! $VO_CMS_SW_DIR ] ;then\n'
1164 >        txt += '        echo "ERROR ==> CMS software dir not found on WN `hostname`"\n'
1165 >        txt += '        job_exit_code=10031\n'
1166 >        txt += '        func_exit\n'
1167 >        txt += '    else\n'
1168 >        txt += '        echo "Sourcing environment... "\n'
1169 >        txt += '        if [ ! -s $VO_CMS_SW_DIR/cmsset_default.sh ] ;then\n'
1170 >        txt += '            echo "ERROR ==> cmsset_default.sh file not found into dir $VO_CMS_SW_DIR"\n'
1171 >        txt += '            job_exit_code=10020\n'
1172 >        txt += '            func_exit\n'
1173 >        txt += '        fi\n'
1174 >        txt += '        echo "sourcing $VO_CMS_SW_DIR/cmsset_default.sh"\n'
1175 >        txt += '        source $VO_CMS_SW_DIR/cmsset_default.sh\n'
1176 >        txt += '        result=$?\n'
1177 >        txt += '        if [ $result -ne 0 ]; then\n'
1178 >        txt += '            echo "ERROR ==> problem sourcing $VO_CMS_SW_DIR/cmsset_default.sh"\n'
1179 >        txt += '            job_exit_code=10032\n'
1180 >        txt += '            func_exit\n'
1181 >        txt += '        fi\n'
1182 >        txt += '    fi\n'
1183 >        txt += '    \n'
1184 >        txt += '    echo "==> setup cms environment ok"\n'
1185          return txt
1186  
1393    ### FEDE FOR DBS OUTPUT PUBLICATION
1187      def modifyReport(self, nj):
1188          """
1189 <        insert the part of the script that modifies the FrameworkJob Report
1189 >        insert the part of the script that modifies the FrameworkJob Report
1190          """
1191 <
1192 <        txt = ''
1193 <        try:
1401 <            publish_data = int(self.cfg_params['USER.publish_data'])          
1402 <        except KeyError:
1403 <            publish_data = 0
1404 <        if (publish_data == 1):  
1405 <            txt += 'echo "Modify Job Report" \n'
1406 <            #txt += 'chmod a+x $RUNTIME_AREA/'+self.version+'/ProdAgentApi/FwkJobRep/ModifyJobReport.py\n'
1407 <            ################ FEDE FOR DBS2 #############################################
1408 <            txt += 'chmod a+x $SOFTWARE_DIR/ProdAgentApi/FwkJobRep/ModifyJobReport.py\n'
1409 <            #############################################################################
1410 <            #try:
1411 <            #    publish_data = int(self.cfg_params['USER.publish_data'])          
1412 <            #except KeyError:
1413 <            #    publish_data = 0
1414 <
1415 <            txt += 'if [ -z "$SE" ]; then\n'
1416 <            txt += '    SE="" \n'
1417 <            txt += 'fi \n'
1418 <            txt += 'if [ -z "$SE_PATH" ]; then\n'
1419 <            txt += '    SE_PATH="" \n'
1420 <            txt += 'fi \n'
1421 <            txt += 'echo "SE = $SE"\n'
1422 <            txt += 'echo "SE_PATH = $SE_PATH"\n'
1423 <
1424 <        #if (publish_data == 1):  
1425 <            #processedDataset = self.cfg_params['USER.processed_datasetname']
1191 >        txt = '\n#Written by cms_cmssw::modifyReport\n'
1192 >        publish_data = int(self.cfg_params.get('USER.publish_data',0))
1193 >        if (publish_data == 1):
1194              processedDataset = self.cfg_params['USER.publish_data_name']
1195 <            txt += 'ProcessedDataset='+processedDataset+'\n'
1196 <            #### LFN=/store/user/<user>/processedDataset_PSETHASH
1197 <            txt += 'if [ "$SE_PATH" == "" ]; then\n'
1198 <            #### FEDE: added slash in LFN ##############
1195 >            if (self.primaryDataset == 'null'):
1196 >                 self.primaryDataset = processedDataset
1197 >            if (common.scheduler.name().upper() == "CAF" or common.scheduler.name().upper() == "LSF"):
1198 >                ### FEDE FOR NEW LFN ###
1199 >                LFNBaseName = LFNBase(self.primaryDataset, processedDataset, LocalUser=True)
1200 >                self.user = getUserName(LocalUser=True)
1201 >                ########################
1202 >            else :
1203 >                ### FEDE FOR NEW LFN ###
1204 >                LFNBaseName = LFNBase(self.primaryDataset, processedDataset)
1205 >                self.user = getUserName()
1206 >                ########################
1207 >
1208 >            txt += 'if [ $copy_exit_status -eq 0 ]; then\n'
1209 >            ### FEDE FOR NEW LFN ###
1210 >            #txt += '    FOR_LFN=%s_${PSETHASH}/\n'%(LFNBaseName)
1211 >            txt += '    FOR_LFN=%s/${PSETHASH}/\n'%(LFNBaseName)
1212 >            ########################
1213 >            txt += 'else\n'
1214              txt += '    FOR_LFN=/copy_problems/ \n'
1215 <            txt += 'else \n'
1216 <            txt += '    tmp=`echo $SE_PATH | awk -F \'store\' \'{print$2}\'` \n'
1217 <            #####  FEDE TO BE CHANGED, BECAUSE STORE IS HARDCODED!!!! ########
1218 <            txt += '    FOR_LFN=/store$tmp \n'
1219 <            txt += 'fi \n'
1215 >            txt += '    SE=""\n'
1216 >            txt += '    SE_PATH=""\n'
1217 >            txt += 'fi\n'
1218 >
1219 >            txt += 'echo ">>> Modify Job Report:" \n'
1220 >            txt += 'chmod a+x $RUNTIME_AREA/ProdCommon/FwkJobRep/ModifyJobReport.py\n'
1221 >            txt += 'ProcessedDataset='+processedDataset+'\n'
1222              txt += 'echo "ProcessedDataset = $ProcessedDataset"\n'
1223 +            txt += 'echo "SE = $SE"\n'
1224 +            txt += 'echo "SE_PATH = $SE_PATH"\n'
1225              txt += 'echo "FOR_LFN = $FOR_LFN" \n'
1226              txt += 'echo "CMSSW_VERSION = $CMSSW_VERSION"\n\n'
1227 <            #txt += 'echo "$RUNTIME_AREA/'+self.version+'/ProdAgentApi/FwkJobRep/ModifyJobReport.py crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH"\n'
1228 <            txt += 'echo "$SOFTWARE_DIR/ProdAgentApi/FwkJobRep/ModifyJobReport.py crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH"\n'
1229 <            txt += '$SOFTWARE_DIR/ProdAgentApi/FwkJobRep/ModifyJobReport.py crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH\n'
1230 <            #txt += '$RUNTIME_AREA/'+self.version+'/ProdAgentApi/FwkJobRep/ModifyJobReport.py crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier $ProcessedDataset $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH\n'
1444 <      
1227 >            ### FEDE FOR NEW LFN ###
1228 >            txt += 'echo "$RUNTIME_AREA/ProdCommon/FwkJobRep/ModifyJobReport.py $RUNTIME_AREA/crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier ' + self.user + '-$ProcessedDataset-$PSETHASH $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH"\n'
1229 >            txt += '$RUNTIME_AREA/ProdCommon/FwkJobRep/ModifyJobReport.py $RUNTIME_AREA/crab_fjr_$NJob.xml $NJob $FOR_LFN $PrimaryDataset $DataTier ' + self.user + '-$ProcessedDataset-$PSETHASH $ApplicationFamily $executable $CMSSW_VERSION $PSETHASH $SE $SE_PATH\n'
1230 >            ########################
1231              txt += 'modifyReport_result=$?\n'
1446            txt += 'echo modifyReport_result = $modifyReport_result\n'
1232              txt += 'if [ $modifyReport_result -ne 0 ]; then\n'
1233 <            txt += '    exit_status=1\n'
1234 <            txt += '    echo "ERROR: Problem with ModifyJobReport"\n'
1233 >            txt += '    modifyReport_result=70500\n'
1234 >            txt += '    job_exit_code=$modifyReport_result\n'
1235 >            txt += '    echo "ModifyReportResult=$modifyReport_result" | tee -a $RUNTIME_AREA/$repo\n'
1236 >            txt += '    echo "WARNING: Problem with ModifyJobReport"\n'
1237              txt += 'else\n'
1238 <            txt += '    mv NewFrameworkJobReport.xml crab_fjr_$NJob.xml\n'
1238 >            txt += '    mv NewFrameworkJobReport.xml $RUNTIME_AREA/crab_fjr_$NJob.xml\n'
1239              txt += 'fi\n'
1453        else:
1454            txt += 'echo "no data publication required"\n'
1455            #txt += 'ProcessedDataset=no_data_to_publish \n'
1456            #### FEDE: added slash in LFN ##############
1457            #txt += 'FOR_LFN=/local/ \n'
1458            #txt += 'echo "ProcessedDataset = $ProcessedDataset"\n'
1459            #txt += 'echo "FOR_LFN = $FOR_LFN" \n'
1240          return txt
1241  
1242 <    def cleanEnv(self):
1243 <        ### OLI_DANIELE
1244 <        txt = ''
1245 <        txt += 'if [ $middleware == OSG ]; then\n'  
1246 <        txt += '    cd $RUNTIME_AREA\n'
1247 <        txt += '    echo "Remove working directory: $WORKING_DIR"\n'
1248 <        txt += '    /bin/rm -rf $WORKING_DIR\n'
1249 <        txt += '    if [ -d $WORKING_DIR ] ;then\n'
1250 <        txt += '              echo "SET_EXE 60999 ==> OSG $WORKING_DIR could not be deleted on WN `hostname` after cleanup of WN"\n'
1251 <        txt += '              echo "JOB_EXIT_STATUS = 60999"\n'
1252 <        txt += '              echo "JobExitCode=60999" | tee -a $RUNTIME_AREA/$repo\n'
1253 <        txt += '              dumpStatus $RUNTIME_AREA/$repo\n'
1254 <        txt += '        rm -f $RUNTIME_AREA/$repo \n'
1255 <        txt += '        echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
1256 <        txt += '        echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
1242 >    def wsParseFJR(self):
1243 >        """
1244 >        Parse the FrameworkJobReport to obtain useful infos
1245 >        """
1246 >        txt = '\n#Written by cms_cmssw::wsParseFJR\n'
1247 >        txt += 'echo ">>> Parse FrameworkJobReport crab_fjr.xml"\n'
1248 >        txt += 'if [ -s $RUNTIME_AREA/crab_fjr_$NJob.xml ]; then\n'
1249 >        txt += '    if [ -s $RUNTIME_AREA/parseCrabFjr.py ]; then\n'
1250 >        txt += '        cmd_out=`python $RUNTIME_AREA/parseCrabFjr.py --input $RUNTIME_AREA/crab_fjr_$NJob.xml --dashboard $MonitorID,$MonitorJobID '+self.debugWrap+'`\n'
1251 >        if self.debug_wrapper :
1252 >            txt += '        echo "Result of parsing the FrameworkJobReport crab_fjr.xml: $cmd_out"\n'
1253 >        txt += '        executable_exit_status=`python $RUNTIME_AREA/parseCrabFjr.py --input $RUNTIME_AREA/crab_fjr_$NJob.xml --exitcode`\n'
1254 >        txt += '        if [ $executable_exit_status -eq 50115 ];then\n'
1255 >        txt += '            echo ">>> crab_fjr.xml contents: "\n'
1256 >        txt += '            cat $RUNTIME_AREA/crab_fjr_$NJob.xml\n'
1257 >        txt += '            echo "Wrong FrameworkJobReport --> does not contain useful info. ExitStatus: $executable_exit_status"\n'
1258 >        txt += '        elif [ $executable_exit_status -eq -999 ];then\n'
1259 >        txt += '            echo "ExitStatus from FrameworkJobReport not available. not available. Using exit code of executable from command line."\n'
1260 >        txt += '        else\n'
1261 >        txt += '            echo "Extracted ExitStatus from FrameworkJobReport parsing output: $executable_exit_status"\n'
1262 >        txt += '        fi\n'
1263 >        txt += '    else\n'
1264 >        txt += '        echo "CRAB python script to parse CRAB FrameworkJobReport crab_fjr.xml is not available, using exit code of executable from command line."\n'
1265          txt += '    fi\n'
1266 +          #### Patch to check input data reading for CMSSW16x Hopefully we-ll remove it asap
1267 +
1268 +        if (self.datasetPath and not (self.dataset_pu or self.useParent) :
1269 +          # VERIFY PROCESSED DATA
1270 +            txt += '    if [ $executable_exit_status -eq 0 ];then\n'
1271 +            txt += '      echo ">>> Verify list of processed files:"\n'
1272 +            txt += '      echo $InputFiles |tr -d \'\\\\\' |tr \',\' \'\\n\'|tr -d \'"\' > input-files.txt\n'
1273 +            txt += '      python $RUNTIME_AREA/parseCrabFjr.py --input $RUNTIME_AREA/crab_fjr_$NJob.xml --lfn > processed-files.txt\n'
1274 +            txt += '      cat input-files.txt  | sort | uniq > tmp.txt\n'
1275 +            txt += '      mv tmp.txt input-files.txt\n'
1276 +            txt += '      echo "cat input-files.txt"\n'
1277 +            txt += '      echo "----------------------"\n'
1278 +            txt += '      cat input-files.txt\n'
1279 +            txt += '      cat processed-files.txt | sort | uniq > tmp.txt\n'
1280 +            txt += '      mv tmp.txt processed-files.txt\n'
1281 +            txt += '      echo "----------------------"\n'
1282 +            txt += '      echo "cat processed-files.txt"\n'
1283 +            txt += '      echo "----------------------"\n'
1284 +            txt += '      cat processed-files.txt\n'
1285 +            txt += '      echo "----------------------"\n'
1286 +            txt += '      diff -q input-files.txt processed-files.txt\n'
1287 +            txt += '      fileverify_status=$?\n'
1288 +            txt += '      if [ $fileverify_status -ne 0 ]; then\n'
1289 +            txt += '         executable_exit_status=30001\n'
1290 +            txt += '         echo "ERROR ==> not all input files processed"\n'
1291 +            txt += '         echo "      ==> list of processed files from crab_fjr.xml differs from list in pset.cfg"\n'
1292 +            txt += '         echo "      ==> diff input-files.txt processed-files.txt"\n'
1293 +            txt += '      fi\n'
1294 +            txt += '    fi\n'
1295 +            txt += '\n'
1296 +        txt += 'else\n'
1297 +        txt += '    echo "CRAB FrameworkJobReport crab_fjr.xml is not available, using exit code of executable from command line."\n'
1298          txt += 'fi\n'
1299          txt += '\n'
1300 +        txt += 'echo "ExeExitCode=$executable_exit_status" | tee -a $RUNTIME_AREA/$repo\n'
1301 +        txt += 'echo "EXECUTABLE_EXIT_STATUS = $executable_exit_status"\n'
1302 +        txt += 'job_exit_code=$executable_exit_status\n'
1303 +
1304          return txt
1305  
1306      def setParam_(self, param, value):
# Line 1485 | Line 1309 | class Cmssw(JobType):
1309      def getParams(self):
1310          return self._params
1311  
1488    def setTaskid_(self):
1489        self._taskId = self.cfg_params['taskId']
1490        
1491    def getTaskid(self):
1492        return self._taskId
1493
1312      def uniquelist(self, old):
1313          """
1314          remove duplicates from a list
# Line 1500 | Line 1318 | class Cmssw(JobType):
1318              nd[e]=0
1319          return nd.keys()
1320  
1321 <
1504 <    def checkOut(self, limit):
1321 >    def outList(self):
1322          """
1323          check the dimension of the output files
1324          """
1325 <        txt = 'echo "*****************************************"\n'
1326 <        txt += 'echo "** Starting output sandbox limit check **"\n'
1510 <        txt += 'echo "*****************************************"\n'
1511 <        allOutFiles = ""
1325 >        txt = ''
1326 >        txt += 'echo ">>> list of expected files on output sandbox"\n'
1327          listOutFiles = []
1328 <        for fileOut in (self.output_file+self.output_file_sandbox):
1329 <             if fileOut.find('crab_fjr') == -1:
1330 <                 allOutFiles = allOutFiles + " " + self.numberFile_(fileOut, '$NJob')
1331 <                 listOutFiles.append(self.numberFile_(fileOut, '$NJob'))
1332 <        txt += 'echo "OUTPUT files: '+str(allOutFiles)+'";\n'
1333 <        txt += 'ls -gGhrta;\n'
1334 <        txt += 'sum=0;\n'
1335 <        txt += 'for file in '+str(allOutFiles)+' ; do\n'
1336 <        txt += '    if [ -e $file ]; then\n'
1337 <        txt += '        tt=`ls -gGrta $file | awk \'{ print $3 }\'`\n'
1338 <        txt += '        sum=`expr $sum + $tt`\n'
1339 <        txt += '    else\n'
1340 <        txt += '        echo "WARNING: output file $file not found!"\n'
1341 <        txt += '    fi\n'
1342 <        txt += 'done\n'
1528 <        txt += 'echo "Total Output dimension: $sum";\n'
1529 <        txt += 'limit='+str(limit)+';\n'
1530 <        txt += 'echo "OUTPUT FILES LIMIT SET TO: $limit";\n'
1531 <        txt += 'if [ $limit -lt $sum ]; then\n'
1532 <        txt += '    echo "WARNING: output files have to big size - something will be lost;"\n'
1533 <        txt += '    echo "         checking the output file sizes..."\n'
1534 <        """
1535 <        txt += '    dim=0;\n'
1536 <        txt += '    exclude=0;\n'
1537 <        txt += '    for files in '+str(allOutFiles)+' ; do\n'
1538 <        txt += '        sumTemp=0;\n'
1539 <        txt += '        for file2 in '+str(allOutFiles)+' ; do\n'
1540 <        txt += '            if [ $file != $file2 ]; then\n'
1541 <        txt += '                tt=`ls -gGrta $file2 | awk \'{ print $3 }\';`\n'
1542 <        txt += '                sumTemp=`expr $sumTemp + $tt`;\n'
1543 <        txt += '            fi\n'
1544 <        txt += '        done\n'
1545 <        txt += '        if [ $sumTemp -lt $limit ]; then\n'
1546 <        txt += '            if [ $dim -lt $sumTemp ]; then\n'
1547 <        txt += '                dim=$sumTemp;\n'
1548 <        txt += '                exclude=$file;\n'
1549 <        txt += '            fi\n'
1550 <        txt += '        fi\n'
1551 <        txt += '    done\n'
1552 <        txt += '    echo "Dimension calculated: $dim"; echo "File to exclude: $exclude";\n'
1553 <        """
1554 <        txt += '    tot=0;\n'
1555 <        txt += '    for file2 in '+str(allOutFiles)+' ; do\n'
1556 <        txt += '        tt=`ls -gGrta $file2 | awk \'{ print $3 }\';`\n'
1557 <        txt += '        tot=`expr $tot + $tt`;\n'
1558 <        txt += '        if [ $limit -lt $tot ]; then\n'
1559 <        txt += '            tot=`expr $tot - $tt`;\n'
1560 <        txt += '            fileLast=$file;\n'
1561 <        txt += '            break;\n'
1562 <        txt += '        fi\n'
1563 <        txt += '    done\n'
1564 <        txt += '    echo "Dimension calculated: $tot"; echo "First file to exclude: $file";\n'
1565 <        txt += '    flag=0;\n'    
1566 <        txt += '    for filess in '+str(allOutFiles)+' ; do\n'
1567 <        txt += '        if [ $fileLast = $filess ]; then\n'
1568 <        txt += '            flag=1;\n'
1569 <        txt += '        fi\n'
1570 <        txt += '        if [ $flag -eq 1 ]; then\n'
1571 <        txt += '            rm -f $filess;\n'
1572 <        txt += '        fi\n'
1573 <        txt += '    done\n'
1574 <        txt += '    ls -agGhrt;\n'
1575 <        txt += '    echo "WARNING: output files are too big in dimension: can not put in the output_sandbox.";\n'
1576 <        txt += '    echo "JOB_EXIT_STATUS = 70000";\n'
1577 <        txt += '    exit_status=70000;\n'
1578 <        txt += 'else'
1579 <        txt += '    echo "Total Output dimension $sum is fine.";\n'
1580 <        txt += 'fi\n'
1581 <        txt += 'echo "*****************************************"\n'
1582 <        txt += 'echo "*** Ending output sandbox limit check ***"\n'
1583 <        txt += 'echo "*****************************************"\n'
1328 >        stdout = 'CMSSW_$NJob.stdout'
1329 >        stderr = 'CMSSW_$NJob.stderr'
1330 >        if (self.return_data == 1):
1331 >            for file in (self.output_file+self.output_file_sandbox):
1332 >                listOutFiles.append(numberFile(file, '$NJob'))
1333 >            listOutFiles.append(stdout)
1334 >            listOutFiles.append(stderr)
1335 >        else:
1336 >            for file in (self.output_file_sandbox):
1337 >                listOutFiles.append(numberFile(file, '$NJob'))
1338 >            listOutFiles.append(stdout)
1339 >            listOutFiles.append(stderr)
1340 >        txt += 'echo "output files: '+string.join(listOutFiles,' ')+'"\n'
1341 >        txt += 'filesToCheck="'+string.join(listOutFiles,' ')+'"\n'
1342 >        txt += 'export filesToCheck\n'
1343          return txt

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines