cocluto/ClusterController/JobsState.py

72 lines
2.2 KiB
Python
Raw Normal View History

from Log import *
class JobsState:
"""
represents a snapshot of the state of SGE jobs as seen by the SGE command "qstat -f -u \*"
"""
def __init__( self ):
self.m_jobs = {} # list of jobs
self.m_queueMachines = {} # list of queue machines such as allintel.q@simpatix10
self.m_stateTime = None # the time at which the state was snapshot
def deleteAllJobs( self ):
self.m_jobs = {}
def addJob( self, job ):
self.m_jobs[ job.getId() ] = job
def getJob( self, jobId ):
return self.m_jobs.get( jobId )
def setTime( self, stateTime ):
self.m_stateTime = stateTime
def getTime( self ):
return self.m_stateTime
def getJobsOnMachine( self, machineName ):
jobsOnMachine = {}
for jobId, job in self.m_jobs.iteritems():
if job.getSlots().get(machineName):
jobsOnMachine[ jobId ] = job
return jobsOnMachine
def getNumFreeSlotsOnQueueMachine( self, queueMachine ):
#logInfo('getNumFreeSlotsOnQueueMachine : looking for free slots on queuemachine %s' % queueMachine.getName() )
numUsedSlots = 0
for job in self.m_jobs.itervalues():
numUsedSlotsByThisJob = job.getSlots().get( queueMachine.getMachineName() )
if numUsedSlotsByThisJob != None:
#logInfo('getNumFreeSlotsOnQueueMachine : job %d uses %d slots' % (job.getId().asStr(), numUsedSlotsByThisJob) )
numUsedSlots += numUsedSlotsByThisJob
else:
None
#logInfo('getNumFreeSlotsOnQueueMachine : job %d uses no slot' % job.getId().asStr() )
numFreeSlots = queueMachine.getNumSlots() - numUsedSlots
assert( numFreeSlots >= 0 )
return numFreeSlots
def addQueueMachine( self, queueMachine ):
self.m_queueMachines[ queueMachine.getName() ] = queueMachine
def getQueueMachine( self, machineName ):
"""
finds the queue machine associated with a machine
"""
queueMachine = None
for qmName, qm in self.m_queueMachines.iteritems():
if qm.m_machineName == machineName:
assert( queueMachine == None ) # to be sure that no more than one queue machine is on a given machine
queueMachine = qm
return queueMachine
def getQueueMachines( self ):
return self.m_queueMachines
def getPendingJobs( self ):
pendingJobs = {}
for jobId, job in self.m_jobs.iteritems():
if job.isPending():
pendingJobs[ job.getId() ] = job
return pendingJobs