ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/COMP/CRAB/python/SchedulerGlite.py
Revision: 1.28
Committed: Wed Nov 14 16:35:21 2007 UTC (17 years, 5 months ago) by ewv
Content type: text/x-python
Branch: MAIN
CVS Tags: CRAB_2_0_3, CRAB_2_0_3_pre1, CRAB_2_0_2, CRAB_2_0_2_pre6, CRAB_2_0_2_pre5
Changes since 1.27: +26 -36 lines
Log Message:
Remove references to GRID3

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 ewv 1.28
34 spiga 1.2 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 ewv 1.28 if n != 0:last.append(n-1)
43 spiga 1.2 if len(first)>len(last) :last.append(common.jobDB.nJobs())
44 ewv 1.28
45 spiga 1.2 req = ''
46     req = req + jbt.getRequirements()
47 ewv 1.28
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 ewv 1.28
79 spiga 1.2 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 ewv 1.28
95 spiga 1.2 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 ewv 1.28 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 ewv 1.28
117 fanzago 1.4 if (self.rb_param_file != ''):
118 ewv 1.28 param_file.write(self.rb_param_file)
119 fanzago 1.4
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 ewv 1.28 param_file.close()
125 spiga 1.2
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 [ $OSG_APP ]; 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 += 'else \n'
164     txt += ' echo "SET_CMS_ENV 10030 ==> middleware not identified" \n'
165     txt += ' echo "JOB_EXIT_STATUS = 10030" \n'
166     txt += ' echo "JobExitCode=10030" | tee -a $RUNTIME_AREA/$repo \n'
167     txt += ' dumpStatus $RUNTIME_AREA/$repo \n'
168     txt += ' exit 1 \n'
169     txt += 'fi \n'
170    
171     txt += 'dumpStatus $RUNTIME_AREA/$repo \n'
172 fanzago 1.26 #txt += 'rm -f $RUNTIME_AREA/$repo \n'
173     #txt += 'echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
174     #txt += 'echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
175 ewv 1.28
176 spiga 1.2 txt += '\n\n'
177    
178 fanzago 1.17 #if int(self.copy_data) == 1:
179     # if self.SE:
180     # txt += 'export SE='+self.SE+'\n'
181     # txt += 'echo "SE = $SE"\n'
182     # if self.SE_PATH:
183     # if ( self.SE_PATH[-1] != '/' ) : self.SE_PATH = self.SE_PATH + '/'
184     # txt += 'export SE_PATH='+self.SE_PATH+'\n'
185     # txt += 'echo "SE_PATH = $SE_PATH"\n'
186 spiga 1.2
187     txt += 'export VO='+self.VO+'\n'
188 ewv 1.28 ### some line for LFC catalog setting
189 fanzago 1.21 #txt += 'if [ $middleware == LCG ]; then \n'
190     #txt += ' if [[ $LCG_CATALOG_TYPE != \''+self.lcg_catalog_type+'\' ]]; then\n'
191     #txt += ' export LCG_CATALOG_TYPE='+self.lcg_catalog_type+'\n'
192     #txt += ' fi\n'
193     #txt += ' if [[ $LFC_HOST != \''+self.lfc_host+'\' ]]; then\n'
194     #txt += ' export LFC_HOST='+self.lfc_host+'\n'
195     #txt += ' fi\n'
196     #txt += ' if [[ $LFC_HOME != \''+self.lfc_home+'\' ]]; then\n'
197     #txt += ' export LFC_HOME='+self.lfc_home+'\n'
198     #txt += ' fi\n'
199     #txt += 'elif [ $middleware == OSG ]; then\n'
200     #txt += ' echo "LFC catalog setting to be implemented for OSG"\n'
201     #txt += 'fi\n'
202 spiga 1.2 #####
203 fanzago 1.21 #if int(self.register_data) == 1:
204     # txt += 'if [ $middleware == LCG ]; then \n'
205     # txt += ' export LFN='+self.LFN+'\n'
206 ewv 1.28 # txt += ' lfc-ls $LFN\n'
207     # txt += ' result=$?\n'
208     # txt += ' echo $result\n'
209     # ### creation of LFN dir in LFC catalog, under /grid/cms dir
210 fanzago 1.21 # txt += ' if [ $result != 0 ]; then\n'
211     # txt += ' lfc-mkdir $LFN\n'
212 ewv 1.28 # txt += ' result=$?\n'
213     # txt += ' echo $result\n'
214 fanzago 1.21 # txt += ' fi\n'
215     # txt += 'elif [ $middleware == OSG ]; then\n'
216     # txt += ' echo " Files registration to be implemented for OSG"\n'
217     # txt += 'fi\n'
218     # txt += '\n'
219     # if self.VO:
220     # txt += 'export VO='+self.VO+'\n'
221     # if self.LFN:
222     # txt += 'if [ $middleware == LCG ]; then \n'
223     # txt += ' export LFN='+self.LFN+'\n'
224     # txt += 'fi\n'
225     # txt += '\n'
226 spiga 1.2
227 ewv 1.28 txt += 'if [ $middleware == LCG ]; then\n'
228 spiga 1.2 txt += ' CloseCEs=`glite-brokerinfo getCE`\n'
229     txt += ' echo "CloseCEs = $CloseCEs"\n'
230     txt += ' CE=`echo $CloseCEs | sed -e "s/:.*//"`\n'
231     txt += ' echo "CE = $CE"\n'
232     txt += 'elif [ $middleware == OSG ]; then \n'
233     txt += ' if [ $OSG_JOB_CONTACT ]; then \n'
234     txt += ' CE=`echo $OSG_JOB_CONTACT | /usr/bin/awk -F\/ \'{print $1}\'` \n'
235     txt += ' else \n'
236     txt += ' echo "SET_CMS_ENV 10099 ==> OSG mode: ERROR in setting CE name from OSG_JOB_CONTACT" \n'
237     txt += ' echo "JOB_EXIT_STATUS = 10099" \n'
238     txt += ' echo "JobExitCode=10099" | tee -a $RUNTIME_AREA/$repo \n'
239     txt += ' dumpStatus $RUNTIME_AREA/$repo \n'
240 fanzago 1.26 #txt += ' rm -f $RUNTIME_AREA/$repo \n'
241     #txt += ' echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
242     #txt += ' echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
243 spiga 1.2 txt += ' exit 1 \n'
244     txt += ' fi \n'
245 ewv 1.28 txt += 'fi \n'
246 spiga 1.2
247     return txt
248 ewv 1.28
249 spiga 1.2 def loggingInfo(self, id):
250     """
251     retrieve the logging info from logging and bookkeeping and return it
252     """
253     self.checkProxy()
254 slacapra 1.8 cmd = 'glite-job-logging-info -v 3 ' + id
255 spiga 1.2 cmd_out = runCommand(cmd)
256     return cmd_out
257    
258     def queryDetailedStatus(self, id):
259     """ Query a detailed status of the job with id """
260     cmd = 'glite-job-status '+id
261     cmd_out = runCommand(cmd)
262     return cmd_out
263    
264     def findSites_(self, n):
265 spiga 1.15 itr4 = []
266 spiga 1.2 sites = common.jobDB.destination(n)
267 slacapra 1.3 if len(sites)>0 and sites[0]=="":
268 spiga 1.15 return itr4
269 ewv 1.28 if sites != [""]:
270 spiga 1.15 ##Addedd Daniele
271 gutsche 1.16 replicas = self.blackWhiteListParser.checkBlackList(sites,n)
272 spiga 1.15 if len(replicas)!=0:
273 gutsche 1.16 replicas = self.blackWhiteListParser.checkWhiteList(replicas,n)
274 ewv 1.28
275 mcinquil 1.24 #if len(replicas)==0:
276     #msg = 'No sites remaining that host any part of the requested data! Exiting... '
277     #raise CrabException(msg)
278 ewv 1.28 itr4 = replicas
279     #####
280 spiga 1.15 return itr4
281 spiga 1.14
282 mcinquil 1.27 def tOut(self, list):
283 mcinquil 1.18 return 180