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

Comparing COMP/CRAB/python/multicrab.py (file contents):
Revision 1.14 by slacapra, Mon Jun 8 16:39:34 2009 UTC vs.
Revision 1.28 by slacapra, Fri Jul 2 15:42:15 2010 UTC

# Line 25 | Line 25 | class MultiCrab:
25          print self.prog_name + ' running on ' +  time.ctime(time.time())
26          print '  working directory   ' + self.continue_dir
27  
28        crabs=[]
28  
29      def processContinueOption_(self,opts):
30  
# Line 169 | Line 168 | class MultiCrab:
168  
169          # read crab.cfg file and search for storage_path
170          crab_cfg_params = loadConfig(crab_cfg,{})
171 +        # also USER.ui_working_dir USER.outputdir and USER.logdir need special treatment
172          if cfg_params.has_key("COMMON"):
173              self.user_remote_dir = cfg_params["COMMON"].get("user.user_remote_dir", crab_cfg_params.get("USER.user_remote_dir",None))
174 +            self.outputdir = cfg_params["COMMON"].get("user.outputdir", crab_cfg_params.get("USER.outputdir",None))
175 +            self.logdir = cfg_params["COMMON"].get("user.logdir", crab_cfg_params.get("USER.logdir",None))
176 +            self.ui_working_dir = cfg_params["COMMON"].get("user.ui_working_dir", crab_cfg_params.get("USER.ui_working_dir",None))
177          else:
178              self.user_remote_dir = crab_cfg_params.get("USER.user_remote_dir",None)
179 +            self.outputdir = crab_cfg_params.get("USER.outputdir",None)
180 +            self.logdir = crab_cfg_params.get("USER.logdir",None)
181 +            self.ui_working_dir = crab_cfg_params.get("USER.ui_working_dir",None)
182 +
183          return
184  
185      def loadMultiConfig(self, file):
# Line 193 | Line 200 | class MultiCrab:
200  
201      def run(self):
202          #run crabs
203 +        runFileName = 'multicrab.exe'
204 +        runFile = open(runFileName,"w")
205          for sec in self.cfg_params_dataset:
206              options={}
207              if self.flag_continue:
# Line 206 | Line 215 | class MultiCrab:
215                      tmp="-"+string.upper(opt.split(".")[0])+"."+opt.split(".")[1]
216                  
217                  options[tmp]=self.cfg_params_dataset[sec][opt]
218 +
219 +            # check if user_remote_dir is set in multicrab.cfg
220 +            # protect against no user_remote_dir
221 +            self.user_remote_dir =self.cfg_params_dataset[sec].get("user.user_remote_dir",self.user_remote_dir)
222 +            if not self.user_remote_dir:
223 +                self.user_remote_dir = "./"
224              # add section to storage_path if exist in crab.cfg
225              if not self.cfg_params_dataset.has_key("USER.user_remote_dir") and self.user_remote_dir:
226                  options["-USER.user_remote_dir"]=self.user_remote_dir+"/"+sec
227 +            # print options["-USER.user_remote_dir"]
228 +            # also for ui_working_dir
229 +            if not self.cfg_params_dataset.has_key("USER.ui_working_dir") and self.ui_working_dir:
230 +                options["-USER.ui_working_dir"]=self.ui_working_dir+"/"+sec
231 +            # if ui_working_dir is set, change -c dir accordnigly
232 +                if self.flag_continue:
233 +                    options['-c']=self.ui_working_dir+"/"+sec
234 +
235 +            # also for logDir
236 +            if not self.cfg_params_dataset.has_key("USER.logdir") and self.logdir:
237 +                options["-USER.logdir"]=self.logdir+"/"+sec
238 +            # also for outputdir
239 +            if not self.cfg_params_dataset.has_key("USER.outputdir") and self.outputdir:
240 +                options["-USER.outputdir"]=self.outputdir+"/"+sec
241 +
242              # Input options (command)
243              for opt in self.opts:
244                  options[opt]=self.opts[opt]
245                  if self.flag_continue and options.has_key("-cfg"):
246                      del options['-cfg']
247                  pass
248 <            crab = None
249 <            try:
250 <                crab = Crab(options)
251 <                crab.run()
252 <                crab.delete()
253 <                print 'Log file is %s%s.log'%(common.work_space.logDir(),common.prog_name)  
254 <                print '\n##############################  E N D  ####################################\n'
255 <            except CrabException, e:
256 <                crab.delete()
227 <                print '\n' + common.prog_name + ': ' + str(e) + '\n'
248 >
249 >            # write crab command to be executed later...
250 >            cmd='crab '
251 >            for o in options:
252 >                if options[o]==None:
253 >                    cmd+=str(o)+' '
254 >                else:
255 >                    options[o] = ''.join(options[o].split())
256 >                    cmd+=str(o)+'='+str(options[o])+' '
257                  pass
258 <            pass
258 >            cmd+="\n"
259 >            #print cmd
260 >
261 >            runFile.write(cmd)
262 >
263 >            # SL this does not work for complex, multi include pset.py
264 >
265 >            # crab = Crab()
266 >            # try:
267 >            #     crab.initialize_(options)
268 >            #     crab.run()
269 >            #     del crab
270 >            #     print 'Log file is %s%s.log'%(common.work_space.logDir(),common.prog_name)  
271 >            #     print '\n##############################  E N D  ####################################\n'
272 >            # except CrabException, e:
273 >            #     del crab
274 >            #     print '\n' + common.prog_name + ': ' + str(e) + '\n'
275 >            #     pass
276 >            # pass
277 >            # if (common.logger): common.logger.delete()
278          pass
279          
280  

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines