File indexing completed on 2026-04-10 08:39:00
0001 import sys
0002
0003 from pandacommon.pandalogger.PandaLogger import PandaLogger
0004
0005 from pandajedi.jedicore.JediTaskBufferInterface import JediTaskBufferInterface
0006 from pandajedi.jedicore.MsgWrapper import MsgWrapper
0007 from pandajedi.jediddm.DDMInterface import DDMInterface
0008 from pandajedi.jediorder.TaskRefiner import TaskRefiner
0009 from pandajedi.jedirefine import RefinerUtils
0010
0011 logger = PandaLogger().getLogger("TaskRefiner")
0012 tmpLog = MsgWrapper(logger)
0013
0014 tbIF = JediTaskBufferInterface()
0015 tbIF.setupInterface()
0016
0017 siteMapper = tbIF.get_site_mapper()
0018
0019 ddmIF = DDMInterface()
0020 ddmIF.setupInterface()
0021
0022
0023 jediTaskID = int(sys.argv[1])
0024
0025 s, taskSpec = tbIF.getTaskWithID_JEDI(jediTaskID)
0026 refiner = TaskRefiner(None, tbIF, ddmIF, taskSpec.vo, taskSpec.prodSourceLabel)
0027 refiner.initializeMods(tbIF, ddmIF)
0028
0029
0030 taskParam = tbIF.getTaskParamsWithID_JEDI(jediTaskID)
0031 taskParamMap = RefinerUtils.decodeJSON(taskParam)
0032
0033 vo = taskParamMap["vo"]
0034 prodSourceLabel = taskParamMap["prodSourceLabel"]
0035 taskType = taskParamMap["taskType"]
0036
0037 cloudName = taskSpec.cloud
0038 vo = taskSpec.vo
0039 prodSourceLabel = taskSpec.prodSourceLabel
0040 queueID = taskSpec.workQueue_ID
0041 gshare_name = taskSpec.gshare
0042
0043 impl = refiner.instantiateImpl(vo, prodSourceLabel, taskType, tbIF, ddmIF)
0044
0045 workQueueMapper = tbIF.getWorkQueueMap()
0046
0047 impl.initializeRefiner(tmpLog)
0048 impl.extractCommon(jediTaskID, taskParamMap, workQueueMapper, taskSpec.splitRule)
0049 impl.doRefine(jediTaskID, taskParamMap)
