from CRABClient.UserUtilities import config
config = config()
config.General.workArea = 'crab_projects'
config.General.transferLogs = True
config.JobType.pluginName = 'Analysis'
config.Data.inputDBS = 'global'
config.Data.splitting = 'FileBased'
config.Data.unitsPerJob = 1
config.Data.outLFNDirBase = '/store/group/lpcgem/'
config.Data.publication = True
config.Site.storageSite = 'T3_US_FNALLPC'

directory = '/afs/cern.ch/work/d/dildick/public/GEM/BackPortGEMUpdatesL1T/CMSSW_10_6_1_patch2/src/'

paramsets = {
    'ILT' : 'DigiFullPU_2023tiltedPU_CSConly.py',
    'NoILT' : 'DigiFullPU_2023tiltedPU_CSConly_noILT.py',
    'NoSLHC' : 'DigiFullPU_2023tiltedPU_CSConly_noSLHC.py'
}

samples = {
    'PU200' : '/Mu_FlatPt2to100-pythia8-gun/PhaseIITDRSpring19DR-PU200_106X_upgrade2023_realistic_v3-v2/GEN-SIM-DIGI-RAW',
    'PU0' : '/Mu_FlatPt2to100-pythia8-gun/PhaseIITDRSpring19DR-NoPU_106X_upgrade2023_realistic_v3-v1/GEN-SIM-DIGI-RAW'
}


if __name__ == '__main__':
    from CRABAPI.RawCommand import crabCommand
    for p in paramsets:
        for s in samples:
            print p, paramsets[p], s, samples[s]
            config.Data.inputDataset = samples[s]
            config.JobType.psetName = directory + paramsets[p]
            config.General.requestName = config.Data.inputDataset.split('/')[1] + '_SD20191002_' + s + '_CSCTP_' + p
            print config
            #crabCommand('submit', config = config)