| 5 |
|
from SchedulerGrid import SchedulerGrid |
| 6 |
|
from crab_exceptions import CrabException |
| 7 |
|
from crab_util import runCommand |
| 8 |
+ |
from crab_util import gethnUserNameFromSiteDB |
| 9 |
|
from ServerConfig import * |
| 9 |
– |
from WMCore.SiteScreening.BlackWhiteListParser import SEBlackWhiteListParser |
| 10 |
|
import Scram |
| 11 |
|
|
| 12 |
|
import common |
| 40 |
|
|
| 41 |
|
self.environment_unique_identifier = None |
| 42 |
|
self.submissionDay = time.strftime("%y%m%d",time.localtime()) |
| 43 |
< |
|
| 43 |
> |
|
| 44 |
|
return |
| 45 |
|
|
| 46 |
|
|
| 48 |
|
""" |
| 49 |
|
Configure the scheduler with the config settings from the user |
| 50 |
|
""" |
| 51 |
< |
|
| 51 |
> |
|
| 52 |
> |
# this line needs to be before the call to SchedulerGrid.configure |
| 53 |
> |
# because that calls SchedulerRemoteglidin in turn and |
| 54 |
> |
# sshControlPersist needs to be defined then :-( |
| 55 |
> |
self.sshControlPersist = cfg_params.get('USER.ssh_control_persist','3600') |
| 56 |
> |
if self.sshControlPersist.lower() == "no" or \ |
| 57 |
> |
self.sshControlPersist.lower() == "yes" or \ |
| 58 |
> |
self.sshControlPersist.isdigit() : |
| 59 |
> |
pass |
| 60 |
> |
else: |
| 61 |
> |
msg = "Error: invalid value '%s' for USER.ssh_control_persist " % \ |
| 62 |
> |
self.sshControlPersist |
| 63 |
> |
raise CrabException(msg) |
| 64 |
> |
|
| 65 |
|
SchedulerGrid.configure(self, cfg_params) |
| 66 |
|
|
| 67 |
|
self.proxyValid=0 |
| 68 |
< |
self.dontCheckProxy=int(cfg_params.get("GRID.dont_check_proxy",0)) |
| 68 |
> |
self.dontCheckProxy=int(cfg_params.get("GRID.dont_check_proxy",'0')) |
| 69 |
|
self.space_token = cfg_params.get("USER.space_token",None) |
| 70 |
|
self.proxyServer= 'myproxy.cern.ch' |
| 71 |
|
self.group = cfg_params.get("GRID.group", None) |
| 72 |
|
self.role = cfg_params.get("GRID.role", None) |
| 73 |
|
self.VO = cfg_params.get('GRID.virtual_organization','cms') |
| 74 |
< |
self.allowOverflow = cfg_params.get('GRID.allow_overflow', 1) |
| 74 |
> |
self.allowOverflow = cfg_params.get('GRID.allow_overflow', '1') |
| 75 |
> |
self.max_rss = cfg_params.get('GRID.max_rss','2000') |
| 76 |
|
|
| 77 |
|
self.checkProxy() |
| 78 |
|
|
| 121 |
|
raise CrabException(msg) |
| 122 |
|
|
| 123 |
|
return |
| 110 |
– |
|
| 111 |
– |
def userName(self): |
| 112 |
– |
""" return the user name """ |
| 113 |
– |
tmp=runCommand("voms-proxy-info -identity 2>/dev/null") |
| 114 |
– |
return tmp.strip() |
| 124 |
|
|
| 125 |
+ |
# |
| 126 |
|
def envUniqueID(self): |
| 127 |
|
taskHash = sha1(common._db.queryTask('name')).hexdigest() |
| 128 |
|
id = "https://" + socket.gethostname() + '/' + taskHash + "/${NJob}" |
| 134 |
|
by $CRABPYTHON/Scheduler.py |
| 135 |
|
""" |
| 136 |
|
|
| 137 |
< |
#SB paste from crab ScheduerGlidein |
| 137 |
> |
#SB paste from crab SchedulerGlidein |
| 138 |
|
|
| 139 |
|
jobParams = "" |
| 140 |
|
|
| 156 |
|
cmsver=re.split('_', cmsVersion) |
| 157 |
|
numericCmsVersion = "%s%.2d%.2d" %(cmsver[1], int(cmsver[2]), int(cmsver[3])) |
| 158 |
|
|
| 159 |
+ |
jobParams += '+DESIRED_CMSDataset ="' +self.datasetPath+'";' |
| 160 |
|
jobParams += '+DESIRED_CMSVersion ="' +cmsVersion+'";' |
| 161 |
|
jobParams += '+DESIRED_CMSVersionNr ="' +numericCmsVersion+'";' |
| 162 |
|
jobParams += '+DESIRED_CMSScramArch ="' +scramArch+'";' |
| 163 |
+ |
|
| 164 |
+ |
userName = gethnUserNameFromSiteDB() |
| 165 |
+ |
jobParams += '+AccountingGroup ="' + userName+'";' |
| 166 |
|
|
| 167 |
|
myscheddName = self.remoteHost |
| 168 |
+ |
|
| 169 |
|
jobParams += '+Glidein_MonitorID = "https://'+ myscheddName + \ |
| 170 |
|
'//' + self.submissionDay + '//$(Cluster).$(Process)"; ' |
| 171 |
|
|
| 172 |
|
if (self.EDG_clock_time): |
| 173 |
< |
jobParams += '+MaxWallTimeMins = '+self.EDG_clock_time+'; ' |
| 173 |
> |
glideinTime = "%d" % (int(self.EDG_clock_time)+20) # 20 min to wrapup |
| 174 |
> |
jobParams += '+MaxWallTimeMins = '+ glideinTime + '; ' |
| 175 |
|
else: |
| 176 |
< |
jobParams += '+MaxWallTimeMins = %d; ' % (60*24) |
| 176 |
> |
jobParams += '+MaxWallTimeMins = %d; ' % (21*60+55) # 21:55h (unit = min) |
| 177 |
> |
|
| 178 |
> |
|
| 179 |
> |
if self.max_rss : |
| 180 |
> |
jobParams += 'request_memory = '+self.max_rss+';' |
| 181 |
|
|
| 182 |
|
if self.allowOverflow == "0": |
| 183 |
|
jobParams += '+CMS_ALLOW_OVERFLOW = False; ' |
| 184 |
|
|
| 185 |
< |
common._db.updateTask_({'jobType':jobParams}) |
| 185 |
> |
if self.EDG_addJdlParam: |
| 186 |
> |
if self.EDG_addJdlParam[-1] == '': |
| 187 |
> |
self.EDG_addJdlParam = self.EDG_addJdlParam[:-1] |
| 188 |
> |
for p in self.EDG_addJdlParam: |
| 189 |
> |
jobParams += p.strip()+';\n' |
| 190 |
|
|
| 191 |
+ |
common._db.updateTask_({'jobType':jobParams}) |
| 192 |
|
|
| 193 |
|
return jobParams |
| 194 |
|
|
| 198 |
|
Return dictionary with specific parameters, to use with real scheduler |
| 199 |
|
is called when scheduler is initialized in Boss, i.e. at each crab command |
| 200 |
|
""" |
| 201 |
< |
#SB this method is used to pass directory names to Boss Scheduler |
| 201 |
> |
#SB this method is used to pass informatinos to Boss Scheduler |
| 202 |
|
# via params dictionary |
| 203 |
|
|
| 204 |
|
jobDir = common.work_space.jobDir() |
| 208 |
|
params = {'shareDir':shareDir, |
| 209 |
|
'jobDir':jobDir, |
| 210 |
|
'taskDir':taskDir, |
| 211 |
< |
'submissionDay':self.submissionDay} |
| 211 |
> |
'submissionDay':self.submissionDay, |
| 212 |
> |
'sshControlPersist':self.sshControlPersist} |
| 213 |
|
|
| 214 |
|
return params |
| 215 |
|
|
| 254 |
|
|
| 255 |
|
txt += 'func_exit() { \n' |
| 256 |
|
txt += self.wsExitFunc_common() |
| 231 |
– |
|
| 232 |
– |
txt += ' tar zcvf ${out_files}.tgz ${final_list}\n' |
| 233 |
– |
txt += ' tmp_size=`ls -gGrta ${out_files}.tgz | awk \'{ print $3 }\'`\n' |
| 234 |
– |
txt += ' rm ${out_files}.tgz\n' |
| 235 |
– |
txt += ' size=`expr $tmp_size`\n' |
| 236 |
– |
txt += ' echo "Total Output dimension: $size"\n' |
| 237 |
– |
txt += ' limit='+str(self.OSBsize) +' \n' |
| 238 |
– |
txt += ' echo "WARNING: output files size limit is set to: $limit"\n' |
| 239 |
– |
txt += ' if [ "$limit" -lt "$size" ]; then\n' |
| 240 |
– |
txt += ' exceed=1\n' |
| 241 |
– |
txt += ' job_exit_code=70000\n' |
| 242 |
– |
txt += ' echo "Output Sanbox too big. Produced output is lost "\n' |
| 243 |
– |
txt += ' else\n' |
| 244 |
– |
txt += ' exceed=0\n' |
| 245 |
– |
txt += ' echo "Total Output dimension $size is fine."\n' |
| 246 |
– |
txt += ' fi\n' |
| 247 |
– |
|
| 248 |
– |
txt += ' echo "JOB_EXIT_STATUS = $job_exit_code"\n' |
| 249 |
– |
txt += ' echo "JobExitCode=$job_exit_code" >> $RUNTIME_AREA/$repo\n' |
| 250 |
– |
txt += ' dumpStatus $RUNTIME_AREA/$repo\n' |
| 251 |
– |
txt += ' if [ $exceed -ne 1 ]; then\n' |
| 252 |
– |
txt += ' tar zcvf ${out_files}.tgz ${final_list}\n' |
| 253 |
– |
txt += ' else\n' |
| 254 |
– |
txt += ' tar zcvf ${out_files}.tgz CMSSW_${NJob}.stdout CMSSW_${NJob}.stderr\n' |
| 255 |
– |
txt += ' fi\n' |
| 256 |
– |
txt += ' python $RUNTIME_AREA/fillCrabFjr.py $RUNTIME_AREA/crab_fjr_$NJob.xml --errorcode $job_exit_code \n' |
| 257 |
– |
|
| 257 |
|
txt += ' exit $job_exit_code\n' |
| 258 |
|
txt += '}\n' |
| 259 |
|
|
| 281 |
|
remoteUserHost=str(task['serverName']) |
| 282 |
|
common.logger.info("serverName from Task DB is %s" % |
| 283 |
|
remoteUserHost) |
| 285 |
– |
if '@' in remoteUserHost: |
| 286 |
– |
remoteHost = remoteUserHost.split('@')[1] |
| 287 |
– |
else: |
| 288 |
– |
remoteHost = remoteUserHost |
| 284 |
|
else: |
| 285 |
|
if self.cfg_params.has_key('CRAB.submit_host'): |
| 286 |
|
# get a remote submission host from crab config file |
| 287 |
|
srvCfg=ServerConfig(self.cfg_params['CRAB.submit_host']).config() |
| 288 |
< |
remoteHost=srvCfg['serverName'] |
| 289 |
< |
common.logger.info("remotehost from crab.cfg = %s" % remoteHost) |
| 288 |
> |
remoteUserHost=srvCfg['serverName'] |
| 289 |
> |
common.logger.info("remotehost from crab.cfg = %s" % remoteUserHost) |
| 290 |
|
else: |
| 291 |
|
# pick from Available Servers List |
| 292 |
|
srvCfg=ServerConfig('default').config() |
| 293 |
< |
remoteHost = srvCfg['serverName'] |
| 294 |
< |
common.logger.info("remotehost from Avail.List = %s" % remoteHost) |
| 293 |
> |
remoteUserHost = srvCfg['serverName'] |
| 294 |
> |
common.logger.info("remotehost from Avail.List = %s" % remoteUserHost) |
| 295 |
|
|
| 296 |
< |
if not remoteHost: |
| 296 |
> |
if not remoteUserHost: |
| 297 |
|
raise CrabException('FATAL ERROR: remoteHost not defined') |
| 303 |
– |
|
| 304 |
– |
#common.logger.info("try to find out username for remote Host via uberftp ...") |
| 305 |
– |
#command="uberftp %s pwd|grep User|awk '{print $3}'" % remoteHost |
| 306 |
– |
#(status, output) = commands.getstatusoutput(command) |
| 307 |
– |
#if status == 0: |
| 308 |
– |
# remoteUser = output |
| 309 |
– |
# common.logger.info("remoteUser set to %s" % remoteUser) |
| 310 |
– |
# if remoteUser==None: |
| 311 |
– |
# raise CrabException('FATAL ERROR: REMOTE USER not defined') |
| 298 |
|
|
| 299 |
< |
#remoteUserHost = remoteUser + '@' + remoteHost |
| 300 |
< |
remoteUserHost = remoteHost |
| 299 |
> |
if '@' in remoteUserHost: |
| 300 |
> |
remoteHost = remoteUserHost.split('@')[1] |
| 301 |
> |
else: |
| 302 |
> |
remoteHost = remoteUserHost |
| 303 |
|
|
| 304 |
|
common._db.updateTask_({'serverName':remoteUserHost}) |
| 305 |
|
|
