ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/COMP/CRAB/python/SchedulerGlite.py
Revision: 1.27
Committed: Wed Nov 7 18:13:31 2007 UTC (17 years, 5 months ago) by mcinquil
Content type: text/x-python
Branch: MAIN
CVS Tags: CRAB_2_0_2_pre4, CRAB_2_0_2_pre3
Changes since 1.26: +1 -1 lines
Log Message:
Adjustment on getout timeout

File Contents

# User Rev Content
1 spiga 1.2 from SchedulerEdg import SchedulerEdg
2     from crab_logger import Logger
3     from crab_exceptions import *
4     from crab_util import *
5 fanzago 1.4 from GliteConfig import *
6 spiga 1.2 import common
7    
8     import os, sys, time
9    
10     class SchedulerGlite(SchedulerEdg):
11     def __init__(self):
12 slacapra 1.3 SchedulerEdg.__init__(self)
13    
14 fanzago 1.4 def rb_configure(self, RB):
15     self.glite_config = ''
16     self.rb_param_file = ''
17    
18     gliteConfig = GliteConfig(RB)
19     self.glite_config = gliteConfig.config()
20    
21     if (self.glite_config != ''):
22 slacapra 1.5 self.rb_param_file = 'WMSconfig = '+self.glite_config+';\n'
23 fanzago 1.4 #print "rb_param_file = ", self.rb_param_file
24     return self.rb_param_file
25    
26 spiga 1.2 def sched_parameter(self):
27     """
28     Returns file with requirements and scheduler-specific parameters
29     """
30     index = int(common.jobDB.nJobs()) - 1
31     job = common.job_list[index]
32     jbt = job.type()
33    
34     lastDest=''
35     first = []
36     last = []
37     for n in range(common.jobDB.nJobs()):
38     currDest=common.jobDB.destination(n)
39     if (currDest!=lastDest):
40     lastDest = currDest
41     first.append(n)
42     if n != 0:last.append(n-1)
43     if len(first)>len(last) :last.append(common.jobDB.nJobs())
44    
45     req = ''
46     req = req + jbt.getRequirements()
47 spiga 1.11
48    
49 spiga 1.2 if self.EDG_requirements:
50 slacapra 1.13 if (not req == ' '): req = req + ' && '
51     req = req + self.EDG_requirements
52    
53 spiga 1.2 if self.EDG_ce_white_list:
54     ce_white_list = string.split(self.EDG_ce_white_list,',')
55 slacapra 1.13 tmpCe=[]
56     concString = '&&'
57     for ce in ce_white_list:
58     tmpCe.append('RegExp("' + string.strip(ce) + '", other.GlueCEUniqueId)')
59 mcinquil 1.20 ### MATTY' FIX: if more then one CE: && -> ||
60     #print "list CE: " + str(tmpCe)
61     if len(tmpCe) == 1:
62     req += " && (" + concString.join(tmpCe) + ") "
63     elif len(tmpCe) > 1:
64     firstCE = 0
65     for reqTemp in tmpCe:
66     #print reqTemp
67     if firstCE == 0:
68     #print "adding: "+str(" && ( (" + reqTemp + ") ")
69     req += " && ( (" + reqTemp + ") "
70     firstCE = 1
71     elif firstCE > 0:
72     #print "adding: "+str(" || (" + reqTemp + ") ")
73     req += " || (" + reqTemp + ") "
74     if firstCE > 0:
75     req += ") "
76     ## old code
77     # if len(tmpCe): req = req + " && (" + concString.join(tmpCe) + ") "
78 spiga 1.2
79     if self.EDG_ce_black_list:
80     ce_black_list = string.split(self.EDG_ce_black_list,',')
81 slacapra 1.13 tmpCe=[]
82     concString = '&&'
83 spiga 1.2 for ce in ce_black_list:
84 slacapra 1.13 tmpCe.append('(!RegExp("' + string.strip(ce) + '", other.GlueCEUniqueId))')
85     if len(tmpCe): req = req + " && (" + concString.join(tmpCe) + ") "
86 spiga 1.11
87 spiga 1.2 if self.EDG_clock_time:
88 slacapra 1.13 if (not req == ' '): req = req + ' && '
89     req = req + 'other.GlueCEPolicyMaxWallClockTime>='+self.EDG_clock_time
90 spiga 1.2
91     if self.EDG_cpu_time:
92 slacapra 1.13 if (not req == ' '): req = req + ' && '
93     req = req + ' other.GlueCEPolicyMaxCPUTime>='+self.EDG_cpu_time
94 spiga 1.2
95     for i in range(len(first)): # Add loop DS
96     self.param='sched_param_'+str(i)+'.clad'
97     param_file = open(common.work_space.shareDir()+'/'+self.param, 'w')
98    
99     itr4=self.findSites_(first[i])
100 slacapra 1.6 reqSites=''
101     reqtmp=[]
102 slacapra 1.3 concString = '||'
103 slacapra 1.13
104     #############
105     # MC Changed matching syntax to avoid gang matching
106     #############
107 slacapra 1.3 for arg in itr4:
108 slacapra 1.6 reqtmp.append(' Member("'+arg+'" , other.GlueCESEBindGroupSEUniqueID) ')
109 slacapra 1.13
110     if len(reqtmp): reqSites = reqSites + " && (" + concString.join(reqtmp) + ") "
111    
112 spiga 1.11 # requirement added to skip gliteCE
113 slacapra 1.13 reqSites = reqSites + '&& (!RegExp("blah", other.GlueCEUniqueId));\n'
114    
115 slacapra 1.6 param_file.write('Requirements = ' + req + reqSites )
116 spiga 1.2
117 fanzago 1.4 if (self.rb_param_file != ''):
118     param_file.write(self.rb_param_file)
119    
120 slacapra 1.7 if len(self.EDG_addJdlParam):
121     for p in self.EDG_addJdlParam:
122     param_file.write(p)
123 spiga 1.2
124     param_file.close()
125    
126     def wsSetupEnvironment(self):
127     """
128     Returns part of a job script which does scheduler-specific work.
129     """
130     txt = ''
131     txt += '# strip arguments\n'
132     txt += 'echo "strip arguments"\n'
133     txt += 'args=("$@")\n'
134     txt += 'nargs=$#\n'
135     txt += 'shift $nargs\n'
136     txt += "# job number (first parameter for job wrapper)\n"
137     txt += "NJob=${args[0]}\n"
138    
139     txt += '# job identification to DashBoard \n'
140     txt += 'MonitorJobID=`echo ${NJob}_$GLITE_WMS_JOBID`\n'
141     txt += 'SyncGridJobId=`echo $GLITE_WMS_JOBID`\n'
142     txt += 'MonitorID=`echo ' + self._taskId + '`\n'
143     txt += 'echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
144     txt += 'echo "SyncGridJobId=`echo $SyncGridJobId`" | tee -a $RUNTIME_AREA/$repo \n'
145     txt += 'echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
146    
147 fanzago 1.26 txt += 'echo "middleware discovery: " \n'
148 spiga 1.2 txt += 'if [ $VO_CMS_SW_DIR ]; then \n'
149     txt += ' middleware=LCG \n'
150     txt += ' echo "SyncCE=`glite-brokerinfo getCE`" | tee -a $RUNTIME_AREA/$repo \n'
151     txt += ' echo "GridFlavour=`echo $middleware`" | tee -a $RUNTIME_AREA/$repo \n'
152 fanzago 1.26 txt += ' echo ">>> middleware =$middleware" \n'
153 spiga 1.2 txt += 'elif [ $GRID3_APP_DIR ]; then\n'
154     txt += ' middleware=OSG \n'
155 mcinquil 1.25 txt += ' if [ $OSG_JOB_CONTACT ]; then \n'
156     txt += ' SyncCE="$OSG_JOB_CONTACT"; \n'
157     txt += ' echo "SyncCE=$SyncCE" | tee -a $RUNTIME_AREA/$repo ;\n'
158 mcinquil 1.22 txt += ' else\n'
159     txt += ' echo "not reporting SyncCE";\n'
160     txt += ' fi\n';
161 spiga 1.2 txt += ' echo "GridFlavour=`echo $middleware`" | tee -a $RUNTIME_AREA/$repo \n'
162 fanzago 1.26 txt += ' echo ">>> middleware =$middleware" \n'
163 spiga 1.2 txt += 'elif [ $OSG_APP ]; then \n'
164     txt += ' middleware=OSG \n'
165 mcinquil 1.25 txt += ' if [ $OSG_JOB_CONTACT ]; then \n'
166     txt += ' SyncCE="$OSG_JOB_CONTACT"; \n'
167     txt += ' echo "SyncCE=$SyncCE" | tee -a $RUNTIME_AREA/$repo ;\n'
168 mcinquil 1.22 txt += ' else\n'
169     txt += ' echo "not reporting SyncCE";\n'
170     txt += ' fi\n';
171 spiga 1.2 txt += ' echo "GridFlavour=`echo $middleware`" | tee -a $RUNTIME_AREA/$repo \n'
172 fanzago 1.26 txt += ' echo ">>> middleware =$middleware" \n'
173 spiga 1.2 txt += 'else \n'
174     txt += ' echo "SET_CMS_ENV 10030 ==> middleware not identified" \n'
175     txt += ' echo "JOB_EXIT_STATUS = 10030" \n'
176     txt += ' echo "JobExitCode=10030" | tee -a $RUNTIME_AREA/$repo \n'
177     txt += ' dumpStatus $RUNTIME_AREA/$repo \n'
178     txt += ' exit 1 \n'
179     txt += 'fi \n'
180    
181     txt += 'dumpStatus $RUNTIME_AREA/$repo \n'
182 fanzago 1.26 #txt += 'rm -f $RUNTIME_AREA/$repo \n'
183     #txt += 'echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
184     #txt += 'echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
185 spiga 1.2
186     txt += '\n\n'
187    
188 fanzago 1.17 #if int(self.copy_data) == 1:
189     # if self.SE:
190     # txt += 'export SE='+self.SE+'\n'
191     # txt += 'echo "SE = $SE"\n'
192     # if self.SE_PATH:
193     # if ( self.SE_PATH[-1] != '/' ) : self.SE_PATH = self.SE_PATH + '/'
194     # txt += 'export SE_PATH='+self.SE_PATH+'\n'
195     # txt += 'echo "SE_PATH = $SE_PATH"\n'
196 spiga 1.2
197     txt += 'export VO='+self.VO+'\n'
198     ### some line for LFC catalog setting
199 fanzago 1.21 #txt += 'if [ $middleware == LCG ]; then \n'
200     #txt += ' if [[ $LCG_CATALOG_TYPE != \''+self.lcg_catalog_type+'\' ]]; then\n'
201     #txt += ' export LCG_CATALOG_TYPE='+self.lcg_catalog_type+'\n'
202     #txt += ' fi\n'
203     #txt += ' if [[ $LFC_HOST != \''+self.lfc_host+'\' ]]; then\n'
204     #txt += ' export LFC_HOST='+self.lfc_host+'\n'
205     #txt += ' fi\n'
206     #txt += ' if [[ $LFC_HOME != \''+self.lfc_home+'\' ]]; then\n'
207     #txt += ' export LFC_HOME='+self.lfc_home+'\n'
208     #txt += ' fi\n'
209     #txt += 'elif [ $middleware == OSG ]; then\n'
210     #txt += ' echo "LFC catalog setting to be implemented for OSG"\n'
211     #txt += 'fi\n'
212 spiga 1.2 #####
213 fanzago 1.21 #if int(self.register_data) == 1:
214     # txt += 'if [ $middleware == LCG ]; then \n'
215     # txt += ' export LFN='+self.LFN+'\n'
216     # txt += ' lfc-ls $LFN\n'
217     # txt += ' result=$?\n'
218     # txt += ' echo $result\n'
219     # ### creation of LFN dir in LFC catalog, under /grid/cms dir
220     # txt += ' if [ $result != 0 ]; then\n'
221     # txt += ' lfc-mkdir $LFN\n'
222     # txt += ' result=$?\n'
223     # txt += ' echo $result\n'
224     # txt += ' fi\n'
225     # txt += 'elif [ $middleware == OSG ]; then\n'
226     # txt += ' echo " Files registration to be implemented for OSG"\n'
227     # txt += 'fi\n'
228     # txt += '\n'
229     # if self.VO:
230     # txt += 'export VO='+self.VO+'\n'
231     # if self.LFN:
232     # txt += 'if [ $middleware == LCG ]; then \n'
233     # txt += ' export LFN='+self.LFN+'\n'
234     # txt += 'fi\n'
235     # txt += '\n'
236 spiga 1.2
237     txt += 'if [ $middleware == LCG ]; then\n'
238     txt += ' CloseCEs=`glite-brokerinfo getCE`\n'
239     txt += ' echo "CloseCEs = $CloseCEs"\n'
240     txt += ' CE=`echo $CloseCEs | sed -e "s/:.*//"`\n'
241     txt += ' echo "CE = $CE"\n'
242     txt += 'elif [ $middleware == OSG ]; then \n'
243     txt += ' if [ $OSG_JOB_CONTACT ]; then \n'
244     txt += ' CE=`echo $OSG_JOB_CONTACT | /usr/bin/awk -F\/ \'{print $1}\'` \n'
245     txt += ' else \n'
246     txt += ' echo "SET_CMS_ENV 10099 ==> OSG mode: ERROR in setting CE name from OSG_JOB_CONTACT" \n'
247     txt += ' echo "JOB_EXIT_STATUS = 10099" \n'
248     txt += ' echo "JobExitCode=10099" | tee -a $RUNTIME_AREA/$repo \n'
249     txt += ' dumpStatus $RUNTIME_AREA/$repo \n'
250 fanzago 1.26 #txt += ' rm -f $RUNTIME_AREA/$repo \n'
251     #txt += ' echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
252     #txt += ' echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
253 spiga 1.2 txt += ' exit 1 \n'
254     txt += ' fi \n'
255     txt += 'fi \n'
256    
257     return txt
258    
259     def loggingInfo(self, id):
260     """
261     retrieve the logging info from logging and bookkeeping and return it
262     """
263     self.checkProxy()
264 slacapra 1.8 cmd = 'glite-job-logging-info -v 3 ' + id
265 spiga 1.2 cmd_out = runCommand(cmd)
266     return cmd_out
267    
268     def queryDetailedStatus(self, id):
269     """ Query a detailed status of the job with id """
270     cmd = 'glite-job-status '+id
271     cmd_out = runCommand(cmd)
272     return cmd_out
273    
274     def findSites_(self, n):
275 spiga 1.15 itr4 = []
276 spiga 1.2 sites = common.jobDB.destination(n)
277 slacapra 1.3 if len(sites)>0 and sites[0]=="":
278 spiga 1.15 return itr4
279     if sites != [""]:
280     ##Addedd Daniele
281 gutsche 1.16 replicas = self.blackWhiteListParser.checkBlackList(sites,n)
282 spiga 1.15 if len(replicas)!=0:
283 gutsche 1.16 replicas = self.blackWhiteListParser.checkWhiteList(replicas,n)
284 spiga 1.15
285 mcinquil 1.24 #if len(replicas)==0:
286     #msg = 'No sites remaining that host any part of the requested data! Exiting... '
287     #raise CrabException(msg)
288 spiga 1.15 itr4 = replicas
289     #####
290     return itr4
291 spiga 1.14
292 mcinquil 1.27 def tOut(self, list):
293 mcinquil 1.18 return 180