ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/COMP/CRAB/python/DBinterface.py
(Generate patch)

Comparing COMP/CRAB/python/DBinterface.py (file contents):
Revision 1.29 by spiga, Tue May 20 20:48:08 2008 UTC vs.
Revision 1.44 by spiga, Sun Aug 31 22:00:21 2008 UTC

# Line 34 | Line 34 | class DBinterface:
34              raise CrabException('Istantiate DB Session : '+str(e))
35  
36          try:
37 <            common.bossSession.installDB('$CRABPRODCOMMONPYTHON/ProdCommon/BossLite/DbObjects/setupDatabase-sqlite.sql')    
37 >            common.bossSession.bossLiteDB.installDB('$CRABPRODCOMMONPYTHON/ProdCommon/BossLite/DbObjects/setupDatabase-sqlite.sql')
38          except Exception, e :
39              raise CrabException('DB Installation error : '+str(e))
40          return
# Line 107 | Line 107 | class DBinterface:
107  
108          return
109  
110 <    def createJobs_(self, jobsL):
110 >    def createJobs_(self, jobsL, isNew=True):
111          """  
112          Fill crab DB with  the jobs filed
113          """
# Line 116 | Line 116 | class DBinterface:
116          jobs = []
117          for id in jobsL:
118              parameters = {}
119 <            parameters['jobId'] =  str(id)
119 >            parameters['jobId'] = int(id)
120 >            parameters['taskId'] = 1
121              parameters['name'] = task['name'] + '_' + 'job' + str(id)
122              job = Job(parameters)
123              jobs.append(job)
124              common.bossSession.getRunningInstance(job)
125              job.runningJob['status'] = 'C'
126 <        task.addJobs(jobs)
126 >        ## added to support second step creation
127 >        ## maybe it is not needed. TO CLARIFY
128 >        if isNew:
129 >            task.addJobs(jobs)
130 >        else:
131 >            task.appendJobs(jobs)
132          try:
133              common.bossSession.updateDB( task )
134          except Exception, e :
# Line 177 | Line 183 | class DBinterface:
183          """
184          task = self.getTask(jobs)
185  
180        Jobs = task.getJobs()
186          print "--------------------------"
187 <        for Job in Jobs:
187 >        for Job in task.jobs:
188              print "Id: ",Job['jobId']
189              print "Dest: ", Job['dlsDestination']
190              print "Output: ", Job['outputFiles']
191              print "Args: ",Job['arguments']
192 +            print "Service: ",Job.runningJob['service']
193              print "--------------------------"
194          return      
195  
# Line 192 | Line 198 | class DBinterface:
198              tmp_task = self.getTask()
199          return common.bossSession.serialize(tmp_task)  
200  
201 <    def queryID(self,server_mode=0):
201 >    def queryID(self,server_mode=0, jid=False):
202          '''
203          Return the taskId if serevr_mode =1
204          Return the joblistId if serevr_mode =0
# Line 201 | Line 207 | class DBinterface:
207          lines=[]
208          task = self.getTask()
209          if server_mode == 1:
210 <            header= "Task Id = %-40s " %(task['name'])
211 <        else:
210 >            headerTask= "Task Id = %-40s " %(task['name'])
211 >            displayReport(self,headerTask,lines)
212 >        if (jid ) or (server_mode == 0):
213              for job in task.jobs:
214                  toPrint=''
215                  common.bossSession.getRunningInstance(job)
216                  toPrint = "%-5s %-50s " % (job['jobId'],job.runningJob['schedulerId'])
217                  lines.append(toPrint)
218              header+= "%-5s %-50s " % ('Job:','ID' )
219 <        displayReport(self,header,lines)
219 >            displayReport(self,header,lines)
220          return  
221  
222      def queryTask(self,attr):
# Line 315 | Line 322 | class DBinterface:
322      def deserXmlStatus(self, reportList):
323  
324          task = self.getTask()
318
325          for job in task.jobs:
326              if not job.runningJob:
327                  raise CrabException( "Missing running object for job %s"%str(job['jobId']) )
328  
329              id = str(job.runningJob['jobId'])
324            # TODO linear search, probably it can be optized with binary search
330              rForJ = None
331 +            nj_list= []
332              for r in reportList:
333                  if r.getAttribute('id') in [ id, 'all']:
334                      rForJ = r
335                      break
336 +            ## Check the submission number and create new running jobs on the client side          
337 +            if rForJ.getAttribute('resubmit') != 'None' and (rForJ.getAttribute('status') not in ['Killed','Killing','Done']) :
338 +                if int(job.runningJob['submission']) < int(rForJ.getAttribute('resubmit')) + 1:
339 +                    nj_list.append(id)
340 +                    self.newRunJobs(nj_list)
341 +
342 +        task_new = self.getTask()
343  
344 +        for job in task_new.jobs:
345 +            id = str(job.runningJob['jobId'])
346 +            # TODO linear search, probably it can be optized with binary search
347 +            rForJ = None
348 +            for r in reportList:
349 +                if r.getAttribute('id') in [ id, 'all']:
350 +                    rForJ = r
351 +                    break
352 +                  
353              # Data alignment
354 <            jobStatus = str(job.runningJob['statusScheduler'])
355 <            if rForJ.getAttribute('status') not in ['Created', 'Submitting', 'Unknown'] and \
356 <                     job.runningJob['statusScheduler'] != 'Cleared':
354 >            if rForJ.getAttribute('status') not in ['Created', 'Unknown'] and not\
355 >                (job.runningJob['statusScheduler'] == 'Killing' and rForJ.getAttribute('status')!='Killed') \
356 >                and not  (job.runningJob['statusScheduler'] in 'Submitting'  and rForJ.getAttribute('status') in ['Killed','Cleared'])  :
357 >                   # update the status  
358 >                common.logger.debug(3,"Updating DB status for job: " + str(id) + " @: " \
359 >                                      + str(rForJ.getAttribute('status')) )
360                  job.runningJob['statusScheduler'] = str( rForJ.getAttribute('status') )
361 <                jobStatus = str(job.runningJob['statusScheduler'])
362 <                job.runningJob['status'] = str( rForJ.getAttribute('sched_status') )
363 <
364 <            job.runningJob['destination'] = str( rForJ.getAttribute('site') )
365 <            dest = str(job.runningJob['destination']).split(':')[0]
366 <
367 <            job.runningJob['applicationReturnCode'] = str( rForJ.getAttribute('exe_exit') )
368 <            exe_exit_code = str(job.runningJob['applicationReturnCode'])
369 <
370 <            job.runningJob['wrapperReturnCode'] = str( rForJ.getAttribute('job_exit') )
371 <            job_exit_code = str(job.runningJob['wrapperReturnCode'])
372 <
361 >                if (rForJ.getAttribute('status') == 'Done' or rForJ.getAttribute('status') == 'Done (Failed)')\
362 >                  and rForJ.getAttribute('sched_status') == 'E' :
363 >                    job.runningJob['status'] = 'SD'
364 >                else:
365 >                    job.runningJob['status'] = str( rForJ.getAttribute('sched_status') )
366 >          
367 >                job.runningJob['schedulerId'] = str( rForJ.getAttribute('sched_id') )
368 >
369 >                job.runningJob['destination'] = str( rForJ.getAttribute('site') )
370 >                dest = str(job.runningJob['destination']).split(':')[0]
371 >              
372 >                job.runningJob['applicationReturnCode'] = str( rForJ.getAttribute('exe_exit') )
373 >                exe_exit_code = str(job.runningJob['applicationReturnCode'])
374 >              
375 >                job.runningJob['wrapperReturnCode'] = str( rForJ.getAttribute('job_exit') )
376 >                job_exit_code = str(job.runningJob['wrapperReturnCode'])
377 >          
378              #if str( rForJ.getAttribute('resubmit') ).isdigit():
379              #    job['submissionNumber'] = int(rForJ.getAttribute('resubmit'))
380              #    job.runningJob['submission'] =  int(rForJ.getAttribute('resubmit'))
381 <
381 >          
382              # TODO cleared='0' field, how should it be handled/mapped in BL? #Fabio
383  
384 <        common.bossSession.updateDB( task )
355 <
384 >        common.bossSession.updateDB( task_new )
385          return
386 +
387 +    # FIXME temporary method to verify what kind of submission to perform towards the server
388 +    def checkIfNeverSubmittedBefore(self):
389 +        for j in self.getTask().jobs:
390 +            if j.runningJob['submission'] > 1 or j.runningJob['status'] != 'C' or \
391 +                    j.runningJob['statusScheduler'] != 'Created':
392 +                return False
393 +        return True
394 +
395 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines