ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/COMP/CRAB/python/multicrab.py
Revision: 1.32
Committed: Thu Apr 26 14:10:25 2012 UTC (13 years ago) by slacapra
Content type: text/x-python
Branch: MAIN
CVS Tags: CRAB_2_8_2_patch1, CRAB_2_8_2, CRAB_2_8_2_pre5, CRAB_2_8_2_pre4, CRAB_2_8_2_pre3, CRAB_2_8_2_pre2
Changes since 1.31: +49 -32 lines
Log Message:
fix for publication with publish_name in task section

File Contents

# User Rev Content
1 slacapra 1.1 #!/usr/bin/env python
2     import sys, os, time, string, shutil
3     from crab_util import *
4     from crab import *
5     import common
6    
7     ###########################################################################
8     class MultiCrab:
9     def __init__(self, opts):
10     self.prog_name='multicrab'
11     # Configuration file
12     self.cfg_fname = None
13     # Continuation flag
14     self.flag_continue = 0
15     self.continue_dir = None
16     self.processContinueOption_(opts)
17    
18     self.processIniFile_(opts)
19     # configuration
20     self.opts=opts
21    
22     if not self.flag_continue:
23     self.createWorkSpace()
24    
25     print self.prog_name + ' running on ' + time.ctime(time.time())
26     print ' working directory ' + self.continue_dir
27    
28    
29     def processContinueOption_(self,opts):
30    
31     # Look for the '-continue' option.
32    
33     for opt in opts.keys():
34     if ( opt in ('-continue','-c') ):
35     self.flag_continue = 1
36     val = opts[opt]
37     if val:
38     if val[0] == '/': self.continue_dir = val # abs path
39     else: self.continue_dir = os.getcwd() + '/' + val # rel path
40     pass
41     break
42     pass
43    
44     # Look for actions which has sense only with '-continue'
45    
46     if "-create" not in opts.keys() :
47     self.flag_continue = 1
48    
49     if not self.flag_continue: return
50    
51     if not self.continue_dir:
52     prefix = self.prog_name + '_'
53     self.continue_dir = findLastWorkDir(prefix)
54     pass
55    
56     if not self.continue_dir:
57     raise CrabException('Cannot find last working directory.')
58    
59     if not os.path.exists(self.continue_dir):
60     msg = 'Cannot continue because the working directory <'
61     msg += self.continue_dir
62     msg += '> does not exist.'
63     raise CrabException(msg)
64    
65     return
66    
67     def createWorkSpace(self):
68     # create WorkingDir for Multicrab
69 slacapra 1.29 import os
70     if not self.continue_dir:
71     prefix = self.prog_name + '_'
72     self.continue_dir = findLastWorkDir(prefix)
73     pass
74     # if 'MULTICRAB.working_dir' in self.opts.keys():
75     # self.continue_dir = os.path.abspath(self.opts['MULTICRAB.working_dir'])
76     if self.ui_working_dir:
77     self.continue_dir = os.path.abspath(self.ui_working_dir)
78 slacapra 1.1 else:
79     current_time = time.strftime('%y%m%d_%H%M%S', time.localtime(time.time()))
80     self.continue_dir = os.getcwd() + '/' + self.prog_name + '_' + current_time
81    
82     if self.continue_dir and not os.path.exists(self.continue_dir):
83     try:
84     os.mkdir(self.continue_dir)
85     except OSError:
86 slacapra 1.31 msg = 'ERROR: Cannot create '+str(self.continue_dir) +' directory.\n'
87 slacapra 1.1 raise CrabException(msg)
88     pass
89     else:
90 slacapra 1.31 msg = 'ERROR: Directory '+str(self.continue_dir) +' already exist.\n'
91 slacapra 1.1 raise CrabException(msg)
92    
93 slacapra 1.29 os.putenv("MULTICRAB_WORKDIR",self.continue_dir)
94     shutil.copyfile(self.cfg_fname,self.continue_dir+'/multicrab.cfg')
95 slacapra 1.1
96     return
97    
98     def processIniFile_(self, opts):
99     """
100     Processes a configuration INI-file.
101     """
102    
103     # Extract cfg-file name from the cmd-line options.
104    
105     for opt in opts.keys():
106     if ( opt == '-cfg' ):
107     if self.flag_continue:
108     raise CrabException('-continue and -cfg cannot coexist.')
109 slacapra 1.8 if opts[opt] :
110     self.cfg_fname = opts[opt]
111 slacapra 1.11 del opts[opt] # do not pass cfg further on
112 slacapra 1.1 else : processHelpOptions()
113     pass
114     pass
115    
116     # Set default cfg-fname
117    
118     if self.cfg_fname == None:
119     if self.flag_continue:
120     self.cfg_fname = self.continue_dir + '/multicrab.cfg'
121     else:
122     self.cfg_fname = 'multicrab.cfg'
123     pass
124     pass
125    
126     # Load cfg-file
127    
128 slacapra 1.7 cfg_params = {}
129 slacapra 1.1 if self.cfg_fname != None:
130     if os.path.exists(self.cfg_fname):
131 slacapra 1.7 cfg_params = self.loadMultiConfig(self.cfg_fname)
132 slacapra 1.1 pass
133     else:
134 slacapra 1.31 msg = 'ERROR: cfg-file '+self.cfg_fname+' not found.'
135 slacapra 1.1 raise CrabException(msg)
136     pass
137     pass
138    
139     # process the [CRAB] section
140    
141     lhp = len('MULTICRAB.')
142 slacapra 1.7 for k in cfg_params.keys():
143 slacapra 1.1 if len(k) >= lhp and k[:lhp] == 'MULTICRAB.':
144     opt = '-'+k[lhp:]
145     if len(opt) >= 3 and opt[:3] == '-__': continue
146     if opt not in opts.keys():
147 slacapra 1.7 opts[opt] = cfg_params[k]
148 slacapra 1.1 pass
149     pass
150     pass
151    
152     self.cfg_params_dataset = {}
153 slacapra 1.3 common_opts = {}
154 slacapra 1.1 # first get common sections
155 slacapra 1.7 for sec in cfg_params:
156 slacapra 1.1 if sec in ['MULTICRAB']:
157 slacapra 1.8 if 'cfg' in cfg_params[sec]:
158     common_opts['cfg']=cfg_params[sec]['cfg']
159 slacapra 1.1 continue
160     if sec in ['COMMON']:
161 slacapra 1.8 common_opts.update(cfg_params[sec])
162 slacapra 1.1 continue
163     pass
164    
165 slacapra 1.8 crab_cfg='crab.cfg'
166 slacapra 1.2 # read crab.cfg file and search for storage_path
167 slacapra 1.8 crab_cfg_params = loadConfig(crab_cfg,{})
168 slacapra 1.16 # also USER.ui_working_dir USER.outputdir and USER.logdir need special treatment
169 slacapra 1.7 if cfg_params.has_key("COMMON"):
170     self.user_remote_dir = cfg_params["COMMON"].get("user.user_remote_dir", crab_cfg_params.get("USER.user_remote_dir",None))
171 slacapra 1.16 self.outputdir = cfg_params["COMMON"].get("user.outputdir", crab_cfg_params.get("USER.outputdir",None))
172     self.logdir = cfg_params["COMMON"].get("user.logdir", crab_cfg_params.get("USER.logdir",None))
173     self.ui_working_dir = cfg_params["COMMON"].get("user.ui_working_dir", crab_cfg_params.get("USER.ui_working_dir",None))
174 slacapra 1.29 self.publish_data_name = cfg_params["COMMON"].get("user.publish_data_name", crab_cfg_params.get("USER.publish_data_name",None))
175 slacapra 1.6 else:
176 slacapra 1.32 self.user_remote_dir = crab_cfg_params.get("USER.user_remote_dir","./")
177 slacapra 1.16 self.outputdir = crab_cfg_params.get("USER.outputdir",None)
178     self.logdir = crab_cfg_params.get("USER.logdir",None)
179     self.ui_working_dir = crab_cfg_params.get("USER.ui_working_dir",None)
180 slacapra 1.29 self.publish_data_name = crab_cfg_params.get("USER.publish_data_name",None)
181 slacapra 1.32 print "self.publish_data_name ",self.publish_data_name
182    
183     if common_opts.has_key('cfg') : crab_cfg=common_opts['cfg']
184    
185     # then Dataset's specific
186     for sec in cfg_params:
187     if sec in ['MULTICRAB', 'COMMON']: continue
188     # add common to all dataset
189     self.cfg_params_dataset[sec]=cfg_params[sec]
190     # special tratment for some parameter
191     if not self.cfg_params_dataset[sec].has_key("user.publish_data_name") and self.publish_data_name:
192     self.cfg_params_dataset[sec]["user.publish_data_name"]=self.publish_data_name+"_"+sec
193     if not self.cfg_params_dataset[sec].has_key("user.user_remote_dir") and self.user_remote_dir:
194     self.cfg_params_dataset[sec]["user.user_remote_dir"]=self.user_remote_dir+"/"+sec
195     if not self.cfg_params_dataset[sec].has_key("user.ui_working_dir") and self.ui_working_dir:
196     self.cfg_params_dataset[sec]["user.ui_working_dir"]=self.ui_working_dir+"/"+sec
197     if not self.cfg_params_dataset[sec].has_key("user.logdir") and self.logdir:
198     self.cfg_params_dataset[sec]["user.logdir"]=self.logdir+"/"+sec
199     if not self.cfg_params_dataset[sec].has_key("user.outputdir") and self.outputdir:
200     self.cfg_params_dataset[sec]["user.outputdir"]=self.outputdir+"/"+sec
201     for key in common_opts:
202     if not self.cfg_params_dataset[sec].has_key(key):
203     self.cfg_params_dataset[sec][key]=common_opts[key]
204     pass
205    
206 slacapra 1.16
207 slacapra 1.1 return
208    
209     def loadMultiConfig(self, file):
210     """
211     returns a dictionary with keys of the form
212     <section>.<option> and the corresponding values
213     """
214     config={}
215     cp = ConfigParser.ConfigParser()
216     cp.read(file)
217     for sec in cp.sections():
218     # print 'Section',sec
219     config[sec]={}
220     for opt in cp.options(sec):
221 slacapra 1.32 # print 'config['+sec+'.'+opt+'] = '+string.strip(cp.get(sec,opt))
222 slacapra 1.1 config[sec][opt] = string.strip(cp.get(sec,opt))
223     return config
224    
225     def run(self):
226     #run crabs
227 slacapra 1.29 runFileName = self.continue_dir+'/multicrab.exe'
228 slacapra 1.16 runFile = open(runFileName,"w")
229 slacapra 1.1 for sec in self.cfg_params_dataset:
230     options={}
231     if self.flag_continue:
232     options['-c']=sec
233     # DatasetName to be used
234     options['-USER.ui_working_dir']=sec
235     # options from multicrab.cfg
236     for opt in self.cfg_params_dataset[sec]:
237 slacapra 1.8 tmp = "-"+str(opt)
238     if len(opt.split("."))==2:
239     tmp="-"+string.upper(opt.split(".")[0])+"."+opt.split(".")[1]
240    
241 slacapra 1.1 options[tmp]=self.cfg_params_dataset[sec][opt]
242 slacapra 1.16
243 slacapra 1.32 # if ui_working_dir is set, change -c dir accordnigly
244     if not self.cfg_params_dataset.has_key("USER.ui_working_dir") and self.ui_working_dir:
245     if self.flag_continue:
246     options['-c']=self.ui_working_dir+"/"+sec
247    
248 slacapra 1.23 # check if user_remote_dir is set in multicrab.cfg
249 spiga 1.24 # protect against no user_remote_dir
250 slacapra 1.32 # self.user_remote_dir =self.cfg_params_dataset[sec].get("user.user_remote_dir",self.user_remote_dir)
251     # if not self.user_remote_dir:
252     # self.user_remote_dir = "./"
253 slacapra 1.2 # add section to storage_path if exist in crab.cfg
254 slacapra 1.32 # if not self.cfg_params_dataset.has_key("USER.user_remote_dir") and self.user_remote_dir:
255     # options["-USER.user_remote_dir"]=self.user_remote_dir+"/"+sec
256 slacapra 1.27 # print options["-USER.user_remote_dir"]
257 slacapra 1.16 # also for ui_working_dir
258 slacapra 1.32 # if not self.cfg_params_dataset.has_key("USER.ui_working_dir") and self.ui_working_dir:
259     # options["-USER.ui_working_dir"]=self.ui_working_dir+"/"+sec
260 slacapra 1.16 # also for logDir
261 slacapra 1.32 # if not self.cfg_params_dataset.has_key("USER.logdir") and self.logdir:
262     # options["-USER.logdir"]=self.logdir+"/"+sec
263     # # also for outputdir
264     # if not self.cfg_params_dataset.has_key("USER.outputdir") and self.outputdir:
265     # options["-USER.outputdir"]=self.outputdir+"/"+sec
266 slacapra 1.29 # also for publish_data_name
267 slacapra 1.32 # print sec," ",self.cfg_params_dataset[sec], self.cfg_params_dataset[sec].has_key("user.publish_data_name")
268     # if not self.cfg_params_dataset[sec].has_key("user.publish_data_name") and self.publish_data_name:
269     # options["-USER.publish_data_name"]=self.publish_data_name+"_"+sec
270     # print "adding user.publish_data_name", self.cfg_params_dataset.has_key("user.publish_data_name")
271 slacapra 1.16
272 slacapra 1.1 # Input options (command)
273     for opt in self.opts:
274 slacapra 1.29 if opt != '-c':
275     options[opt]=self.opts[opt]
276     # options[opt]=self.opts[opt]
277     if self.flag_continue and options.has_key("-cfg"):
278     del options['-cfg']
279     pass
280 slacapra 1.20
281 slacapra 1.32
282 slacapra 1.20 # write crab command to be executed later...
283     cmd='crab '
284     for o in options:
285     if options[o]==None:
286     cmd+=str(o)+' '
287     else:
288     options[o] = ''.join(options[o].split())
289     cmd+=str(o)+'='+str(options[o])+' '
290     pass
291     cmd+="\n"
292     #print cmd
293    
294     runFile.write(cmd)
295    
296     # SL this does not work for complex, multi include pset.py
297    
298     # crab = Crab()
299     # try:
300     # crab.initialize_(options)
301     # crab.run()
302     # del crab
303     # print 'Log file is %s%s.log'%(common.work_space.logDir(),common.prog_name)
304     # print '\n############################## E N D ####################################\n'
305     # except CrabException, e:
306     # del crab
307     # print '\n' + common.prog_name + ': ' + str(e) + '\n'
308     # pass
309     # pass
310     # if (common.logger): common.logger.delete()
311 slacapra 1.1 pass
312 slacapra 1.29 return self.continue_dir
313 slacapra 1.1
314    
315     ###########################################################################
316     if __name__ == '__main__':
317     ## Get rid of some useless warning
318     try:
319     import warnings
320     warnings.simplefilter("ignore", RuntimeWarning)
321     except ImportError:
322     pass # too bad, you'll get the warning
323    
324     # Parse command-line options and create a dictionary with
325     # key-value pairs.
326     options = parseOptions(sys.argv[1:])
327    
328     # Process "help" options, such as '-help', '-version'
329     if processHelpOptions(options) : sys.exit(0)
330    
331     # Create, initialize, and run a Crab object
332     try:
333     multicrab = MultiCrab(options)
334 slacapra 1.29 continue_dir = multicrab.run()
335     import os
336     sys.exit(continue_dir)
337 slacapra 1.1 except CrabException, e:
338     print '\n' + common.prog_name + ': ' + str(e) + '\n'
339    
340     pass