datasetToSource.py

Go to the documentation of this file.

from __future__ import absolute_import
import os
from .dataset import Dataset, CMSDataset, LocalDataset, createDataset, PrivateDataset, createMyDataset
 
import FWCore.ParameterSet.Config as cms
 
def datasetToSource( user, dataset, pattern='.*root', readCache=False):
 
    # print user, dataset, pattern
    data = createDataset(user, dataset, pattern, readCache)
    
    source = cms.Source(
    "PoolSource",
    noEventSort = cms.untracked.bool(True),
    duplicateCheckMode = cms.untracked.string("noDuplicateCheck"),
    fileNames = cms.untracked.vstring()
        )
    
    source.fileNames.extend( data.listOfGoodFiles() )
 
    return source
 
 
def myDatasetToSource( user, dataset, pattern='.*root', dbsInstance=None, readCache=False):
 
    #print user, dataset, pattern, dbsInstance
    data = createMyDataset(user, dataset, pattern, dbsInstance, readCache)
 
    source = cms.Source(
        "PoolSource",
        noEventSort = cms.untracked.bool(True),
        duplicateCheckMode = cms.untracked.string("noDuplicateCheck"),
        fileNames = cms.untracked.vstring()
        )
 
    #print data.listOfGoodFiles()
    source.fileNames.extend( data.listOfGoodFiles() )
 
    return source