pandajedi/jeditest/addTestTaskParamToDEFT.py

0001 import sys
0002
0003 try:
0004     metaID = sys.argv[1]
0005 except Exception:
0006     metaID = None
0007 import json
0008 import uuid
0009
0010 taskParamMap = {}
0011 taskParamMap["taskName"] = str(uuid.uuid4())
0012 taskParamMap["userName"] = "pandasrv1"
0013 taskParamMap["vo"] = "atlas"
0014 taskParamMap["taskPriority"] = 100
0015 taskParamMap["architecture"] = "i686-slc5-gcc43-opt"
0016 taskParamMap["transUses"] = "Atlas-17.2.7"
0017 taskParamMap["transHome"] = "AtlasProduction-17.2.8.10"
0018 taskParamMap["transPath"] = "Reco_trf.py"
0019 taskParamMap["processingType"] = "reco"
0020 taskParamMap["prodSourceLabel"] = "test"
0021 taskParamMap["taskType"] = "prod"
0022 taskParamMap["workingGroup"] = "AP_Higgs"
0023 taskParamMap["coreCount"] = 1
0024 taskParamMap["cloud"] = "US"
0025 logDatasetName = f"panda.jeditest.log.{uuid.uuid4()}"
0026 taskParamMap["log"] = {
0027     "dataset": logDatasetName,
0028     "type": "template",
0029     "param_type": "log",
0030     "token": "ATLASDATADISK",
0031     "value": f"{logDatasetName}.${{SN}}.log.tgz",
0032 }
0033 outDatasetName = f"panda.jeditest.NTUP_EMBLLDN.{uuid.uuid4()}"
0034 taskParamMap["jobParameters"] = [
0035     {
0036         "type": "template",
0037         "param_type": "input",
0038         "value": "inputAODFile=${IN}",
0039         "dataset": "data12_8TeV.00214651.physics_Egamma.merge.AOD.f489_m1261",
0040     },
0041     {
0042         "type": "constant",
0043         "value": 'maxEvents=1000 RunNumber=213816 autoConfiguration=everything preExec="from BTagging.BTaggingFlags import BTaggingFlags;BTaggingFlags.CalibrationTag="BTagCalibALL-07-02""',
0044     },
0045     {
0046         "type": "template",
0047         "param_type": "input",
0048         "value": "DBRelease=${DBR}",
0049         "dataset": "ddo.000001.Atlas.Ideal.DBRelease.v220701",
0050         "attributes": "repeat,nosplit",
0051     },
0052     {"type": "constant", "value": "AMITag=p1462"},
0053     {
0054         "type": "template",
0055         "param_type": "output",
0056         "token": "ATLASDATADISK",
0057         "value": f"outputNTUP_EMBLLDNFile={outDatasetName}.${{SN}}.pool.root",
0058         "dataset": outDatasetName,
0059     },
0060 ]
0061
0062 jonStr = json.dumps(taskParamMap)
0063
0064 from pandajedi.jedicore.JediTaskBufferInterface import (  # noqa: E402
0065     JediTaskBufferInterface,
0066 )
0067
0068 tbIF = JediTaskBufferInterface()
0069 tbIF.setupInterface()
0070 tbIF.insertTaskParams_JEDI(taskParamMap["vo"], taskParamMap["prodSourceLabel"], taskParamMap["userName"], taskParamMap["taskName"], jonStr)