pandajedi/jedithrottle/AtlasProdJobThrottler.py

0001 from pandacommon.pandalogger.PandaLogger import PandaLogger
0002
0003 from .JobThrottlerBase import JobThrottlerBase
0004
0005 logger = PandaLogger().getLogger(__name__.split(".")[-1])
0006
0007
0008 # class to throttle ATLAS production jobs
0009 class AtlasProdJobThrottler(JobThrottlerBase):
0010     # constructor
0011     def __init__(self, taskBufferIF):
0012         JobThrottlerBase.__init__(self, taskBufferIF)
0013         self.comp_name = "prod_job_throttler"
0014         self.app = "jedi"
0015
0016     # check if throttled
0017     def toBeThrottled(self, vo, prodSourceLabel, cloudName, workQueue, resource_name):
0018         return self.toBeThrottledBase(vo, prodSourceLabel, cloudName, workQueue, resource_name, logger)