pandajedi/jeditest/jobThrottlerTest.py

0001 from pandajedi.jedicore.JediTaskBufferInterface import JediTaskBufferInterface
0002 from pandajedi.jediorder.JobThrottler import JobThrottler
0003
0004 tbIF = JediTaskBufferInterface()
0005 tbIF.setupInterface()
0006
0007 vo = "atlas"
0008 prodSourceLabel = "managed"
0009 cloud = "WORLD"
0010
0011 # get SiteMapper
0012 siteMapper = tbIF.get_site_mapper()
0013 wqMap = tbIF.getWorkQueueMap()
0014
0015 jt = JobThrottler(vo, prodSourceLabel)
0016 jt.initializeMods(tbIF)
0017
0018 workQueues = wqMap.getAlignedQueueList(vo, prodSourceLabel)
0019 resource_types = tbIF.load_resource_types()
0020 for workQueue in workQueues:
0021     for resource_type in resource_types:
0022         print(jt.toBeThrottled(vo, prodSourceLabel, cloud, workQueue, resource_type.resource_name))