ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/COMP/CRAB/python/SchedulerGlite.py
Revision: 1.12
Committed: Mon Feb 26 12:02:25 2007 UTC (18 years, 2 months ago) by slacapra
Content type: text/x-python
Branch: MAIN
Changes since 1.11: +1 -1 lines
Log Message:
blah and not *blah*

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     if (req == ' '):
51     req = req + self.EDG_requirements
52     else:
53     req = req + ' && ' + self.EDG_requirements
54     if self.EDG_ce_white_list:
55     ce_white_list = string.split(self.EDG_ce_white_list,',')
56     for i in range(len(ce_white_list)):
57     if i == 0:
58     if (req == ' '):
59 spiga 1.9 req = req + '((RegExp("' + string.strip(ce_white_list[i]) + '", other.GlueCEUniqueId))'
60 spiga 1.2 else:
61 spiga 1.9 req = req + ' && ((RegExp("' + string.strip(ce_white_list[i]) + '", other.GlueCEUniqueId))'
62 spiga 1.2 pass
63     else:
64 spiga 1.9 req = req + ' || (RegExp("' + string.strip(ce_white_list[i]) + '", other.GlueCEUniqueId))'
65 spiga 1.2 req = req + ')'
66    
67     if self.EDG_ce_black_list:
68     ce_black_list = string.split(self.EDG_ce_black_list,',')
69     for ce in ce_black_list:
70     if (req == ' '):
71 spiga 1.9 req = req + '(!RegExp("' + string.strip(ce) + '", other.GlueCEUniqueId))'
72 spiga 1.2 else:
73 spiga 1.9 req = req + ' && (!RegExp("' + string.strip(ce) + '", other.GlueCEUniqueId))'
74 spiga 1.2 pass
75 spiga 1.11
76    
77 spiga 1.2 if self.EDG_clock_time:
78     if (req == ' '):
79     req = req + 'other.GlueCEPolicyMaxWallClockTime>='+self.EDG_clock_time
80     else:
81     req = req + ' && other.GlueCEPolicyMaxWallClockTime>='+self.EDG_clock_time
82    
83     if self.EDG_cpu_time:
84     if (req == ' '):
85     req = req + ' other.GlueCEPolicyMaxCPUTime>='+self.EDG_cpu_time
86     else:
87     req = req + ' && other.GlueCEPolicyMaxCPUTime>='+self.EDG_cpu_time
88    
89     for i in range(len(first)): # Add loop DS
90     self.param='sched_param_'+str(i)+'.clad'
91     param_file = open(common.work_space.shareDir()+'/'+self.param, 'w')
92    
93     itr4=self.findSites_(first[i])
94 slacapra 1.6 reqSites=''
95     reqtmp=[]
96 slacapra 1.3 concString = '||'
97     for arg in itr4:
98     #############
99     # MC Changed matching syntax to avoid gang matching
100     #############
101 slacapra 1.6 reqtmp.append(' Member("'+arg+'" , other.GlueCESEBindGroupSEUniqueID) ')
102 spiga 1.11 if len(reqtmp): reqSites = reqSites + " && (" + concString.join(reqtmp)
103     # requirement added to skip gliteCE
104 slacapra 1.12 reqSites = reqSites + '&& (!RegExp("blah", other.GlueCEUniqueId)));\n'
105 slacapra 1.6 param_file.write('Requirements = ' + req + reqSites )
106 spiga 1.2
107 fanzago 1.4 if (self.rb_param_file != ''):
108     param_file.write(self.rb_param_file)
109    
110 slacapra 1.7 if len(self.EDG_addJdlParam):
111     for p in self.EDG_addJdlParam:
112     param_file.write(p)
113 spiga 1.2
114     param_file.close()
115    
116     def wsSetupEnvironment(self):
117     """
118     Returns part of a job script which does scheduler-specific work.
119     """
120     txt = ''
121     txt += '# strip arguments\n'
122     txt += 'echo "strip arguments"\n'
123     txt += 'args=("$@")\n'
124     txt += 'nargs=$#\n'
125     txt += 'shift $nargs\n'
126     txt += "# job number (first parameter for job wrapper)\n"
127     txt += "NJob=${args[0]}\n"
128    
129     txt += '# job identification to DashBoard \n'
130     txt += 'MonitorJobID=`echo ${NJob}_$GLITE_WMS_JOBID`\n'
131     txt += 'SyncGridJobId=`echo $GLITE_WMS_JOBID`\n'
132     txt += 'MonitorID=`echo ' + self._taskId + '`\n'
133     txt += 'echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
134     txt += 'echo "SyncGridJobId=`echo $SyncGridJobId`" | tee -a $RUNTIME_AREA/$repo \n'
135     txt += 'echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
136    
137     txt += 'echo "middleware discovery " \n'
138     txt += 'if [ $VO_CMS_SW_DIR ]; then \n'
139     txt += ' middleware=LCG \n'
140     txt += ' echo "SyncCE=`glite-brokerinfo getCE`" | tee -a $RUNTIME_AREA/$repo \n'
141     txt += ' echo "GridFlavour=`echo $middleware`" | tee -a $RUNTIME_AREA/$repo \n'
142     txt += ' echo "middleware =$middleware" \n'
143     txt += 'elif [ $GRID3_APP_DIR ]; then\n'
144     txt += ' middleware=OSG \n'
145     txt += ' echo "SyncCE=`echo $EDG_WL_LOG_DESTINATION`" | tee -a $RUNTIME_AREA/$repo \n'
146     txt += ' echo "GridFlavour=`echo $middleware`" | tee -a $RUNTIME_AREA/$repo \n'
147     txt += ' echo "middleware =$middleware" \n'
148     txt += 'elif [ $OSG_APP ]; then \n'
149     txt += ' middleware=OSG \n'
150     txt += ' echo "SyncCE=`echo $EDG_WL_LOG_DESTINATION`" | tee -a $RUNTIME_AREA/$repo \n'
151     txt += ' echo "GridFlavour=`echo $middleware`" | tee -a $RUNTIME_AREA/$repo \n'
152     txt += ' echo "middleware =$middleware" \n'
153     txt += 'else \n'
154     txt += ' echo "SET_CMS_ENV 10030 ==> middleware not identified" \n'
155     txt += ' echo "JOB_EXIT_STATUS = 10030" \n'
156     txt += ' echo "JobExitCode=10030" | tee -a $RUNTIME_AREA/$repo \n'
157     txt += ' dumpStatus $RUNTIME_AREA/$repo \n'
158     txt += ' rm -f $RUNTIME_AREA/$repo \n'
159     txt += ' echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
160     txt += ' echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
161     txt += ' exit 1 \n'
162     txt += 'fi \n'
163    
164     txt += '# report first time to DashBoard \n'
165     txt += 'dumpStatus $RUNTIME_AREA/$repo \n'
166     txt += 'rm -f $RUNTIME_AREA/$repo \n'
167     txt += 'echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
168     txt += 'echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
169    
170     txt += '\n\n'
171    
172     if int(self.copy_data) == 1:
173     if self.SE:
174     txt += 'export SE='+self.SE+'\n'
175     txt += 'echo "SE = $SE"\n'
176     if self.SE_PATH:
177     if ( self.SE_PATH[-1] != '/' ) : self.SE_PATH = self.SE_PATH + '/'
178     txt += 'export SE_PATH='+self.SE_PATH+'\n'
179     txt += 'echo "SE_PATH = $SE_PATH"\n'
180    
181     txt += 'export VO='+self.VO+'\n'
182     ### some line for LFC catalog setting
183     txt += 'if [ $middleware == LCG ]; then \n'
184     txt += ' if [[ $LCG_CATALOG_TYPE != \''+self.lcg_catalog_type+'\' ]]; then\n'
185     txt += ' export LCG_CATALOG_TYPE='+self.lcg_catalog_type+'\n'
186     txt += ' fi\n'
187     txt += ' if [[ $LFC_HOST != \''+self.lfc_host+'\' ]]; then\n'
188     txt += ' export LFC_HOST='+self.lfc_host+'\n'
189     txt += ' fi\n'
190     txt += ' if [[ $LFC_HOME != \''+self.lfc_home+'\' ]]; then\n'
191     txt += ' export LFC_HOME='+self.lfc_home+'\n'
192     txt += ' fi\n'
193     txt += 'elif [ $middleware == OSG ]; then\n'
194     txt += ' echo "LFC catalog setting to be implemented for OSG"\n'
195     txt += 'fi\n'
196     #####
197     if int(self.register_data) == 1:
198     txt += 'if [ $middleware == LCG ]; then \n'
199     txt += ' export LFN='+self.LFN+'\n'
200     txt += ' lfc-ls $LFN\n'
201     txt += ' result=$?\n'
202     txt += ' echo $result\n'
203     ### creation of LFN dir in LFC catalog, under /grid/cms dir
204     txt += ' if [ $result != 0 ]; then\n'
205     txt += ' lfc-mkdir $LFN\n'
206     txt += ' result=$?\n'
207     txt += ' echo $result\n'
208     txt += ' fi\n'
209     txt += 'elif [ $middleware == OSG ]; then\n'
210     txt += ' echo " Files registration to be implemented for OSG"\n'
211     txt += 'fi\n'
212     txt += '\n'
213    
214     if self.VO:
215     txt += 'export VO='+self.VO+'\n'
216     if self.LFN:
217     txt += 'if [ $middleware == LCG ]; then \n'
218     txt += ' export LFN='+self.LFN+'\n'
219     txt += 'fi\n'
220     txt += '\n'
221    
222     txt += 'if [ $middleware == LCG ]; then\n'
223     txt += ' CloseCEs=`glite-brokerinfo getCE`\n'
224     txt += ' echo "CloseCEs = $CloseCEs"\n'
225     txt += ' CE=`echo $CloseCEs | sed -e "s/:.*//"`\n'
226     txt += ' echo "CE = $CE"\n'
227     txt += 'elif [ $middleware == OSG ]; then \n'
228     txt += ' if [ $OSG_JOB_CONTACT ]; then \n'
229     txt += ' CE=`echo $OSG_JOB_CONTACT | /usr/bin/awk -F\/ \'{print $1}\'` \n'
230     txt += ' else \n'
231     txt += ' echo "SET_CMS_ENV 10099 ==> OSG mode: ERROR in setting CE name from OSG_JOB_CONTACT" \n'
232     txt += ' echo "JOB_EXIT_STATUS = 10099" \n'
233     txt += ' echo "JobExitCode=10099" | tee -a $RUNTIME_AREA/$repo \n'
234     txt += ' dumpStatus $RUNTIME_AREA/$repo \n'
235     txt += ' rm -f $RUNTIME_AREA/$repo \n'
236     txt += ' echo "MonitorJobID=`echo $MonitorJobID`" | tee -a $RUNTIME_AREA/$repo \n'
237     txt += ' echo "MonitorID=`echo $MonitorID`" | tee -a $RUNTIME_AREA/$repo\n'
238     txt += ' exit 1 \n'
239     txt += ' fi \n'
240     txt += 'fi \n'
241    
242     return txt
243    
244     def loggingInfo(self, id):
245     """
246     retrieve the logging info from logging and bookkeeping and return it
247     """
248     self.checkProxy()
249 slacapra 1.8 cmd = 'glite-job-logging-info -v 3 ' + id
250 spiga 1.2 cmd_out = runCommand(cmd)
251     return cmd_out
252    
253     def queryDetailedStatus(self, id):
254     """ Query a detailed status of the job with id """
255     cmd = 'glite-job-status '+id
256     cmd_out = runCommand(cmd)
257     return cmd_out
258    
259     def findSites_(self, n):
260     sites = common.jobDB.destination(n)
261 slacapra 1.3 if len(sites)>0 and sites[0]=="":
262     return []
263     return sites