pandaserver/test/testG4sim.py

0001 import sys
0002 import time
0003 import uuid
0004
0005 from pandaserver.taskbuffer.FileSpec import FileSpec
0006 from pandaserver.taskbuffer.JobSpec import JobSpec
0007 from pandaserver.userinterface import Client
0008
0009 if len(sys.argv) > 1:
0010     site = sys.argv[1]
0011 else:
0012     site = None
0013
0014 dataset_name = f"panda.destDB.{str(uuid.uuid4())}"
0015 destination_se = "BNL_ATLAS_2"
0016 # destination_se    = 'BU_ATLAS_Tier2'
0017
0018 files = {
0019     "mc11.007204.singlepart_mu4.evgen.EVNT.v11000302._00037.pool.root.1": None,
0020     "mc11.007204.singlepart_mu4.evgen.EVNT.v11000302._00038.pool.root.1": None,
0021 }
0022
0023 job_list = []
0024
0025 for lfn in files:
0026     job = JobSpec()
0027     job.jobDefinitionID = int(time.time()) % 10000
0028     job.jobName = str(uuid.uuid4())
0029     job.AtlasRelease = "Atlas-11.0.3"
0030     job.homepackage = "JobTransforms-11-00-03-02"
0031     job.transformation = "share/csc.simul.trf"
0032     job.destinationDBlock = dataset_name
0033     job.destinationSE = destination_se
0034     job.computingSite = site
0035     job.prodDBlock = "mc11.007204.singlepart_mu4.evgen.EVNT.v11000302"
0036     job.cmtConfig = "i686-slc4-gcc34-opt"
0037
0038     job.prodSourceLabel = "test"
0039     job.currentPriority = 1000
0040
0041     fileI = FileSpec()
0042     fileI.dataset = job.prodDBlock
0043     fileI.prodDBlock = job.prodDBlock
0044     fileI.lfn = lfn
0045     fileI.type = "input"
0046     job.addFile(fileI)
0047
0048     fileOE = FileSpec()
0049     fileOE.lfn = f"{str(uuid.uuid4())}.HITS.pool.root"
0050     fileOE.destinationDBlock = job.destinationDBlock
0051     fileOE.destinationSE = job.destinationSE
0052     fileOE.dataset = job.destinationDBlock
0053     fileOE.destinationDBlockToken = "ATLASDATADISK"
0054     fileOE.type = "output"
0055     job.addFile(fileOE)
0056
0057     fileOA = FileSpec()
0058     fileOA.lfn = f"{str(uuid.uuid4())}.RDO.pool.root"
0059     fileOA.destinationDBlock = job.destinationDBlock
0060     fileOA.destinationSE = job.destinationSE
0061     fileOA.dataset = job.destinationDBlock
0062     fileOA.destinationDBlockToken = "ATLASDATADISK"
0063     fileOA.type = "output"
0064     job.addFile(fileOA)
0065
0066     fileOL = FileSpec()
0067     fileOL.lfn = f"{str(uuid.uuid4())}.job.log.tgz"
0068     fileOL.destinationDBlock = job.destinationDBlock
0069     fileOL.destinationSE = job.destinationSE
0070     fileOL.dataset = job.destinationDBlock
0071     fileOL.destinationDBlockToken = "ATLASDATADISK"
0072     fileOL.type = "log"
0073     job.addFile(fileOL)
0074
0075     job.jobParameters = f"{fileI.lfn} {fileOE.lfn} {fileOA.lfn}  100 700 2158"
0076
0077     job_list.append(job)
0078
0079 status, output = Client.submit_jobs(job_list)
0080 print("---------------------")
0081 print(f"Status: {status}. Output: {output}")