python/utils/datasetToSource.py

0001 import os
0002 from .dataset import Dataset, CMSDataset, LocalDataset, createDataset, PrivateDataset, createMyDataset
0003
0004 import FWCore.ParameterSet.Config as cms
0005
0006 def datasetToSource( user, dataset, pattern='.*root', readCache=False):
0007
0008     # print user, dataset, pattern
0009     data = createDataset(user, dataset, pattern, readCache)
0010
0011     source = cms.Source(
0012     "PoolSource",
0013     noEventSort = cms.untracked.bool(True),
0014     duplicateCheckMode = cms.untracked.string("noDuplicateCheck"),
0015     fileNames = cms.untracked.vstring()
0016         )
0017
0018     source.fileNames.extend( data.listOfGoodFiles() )
0019
0020     return source
0021
0022 ### MM
0023 def myDatasetToSource( user, dataset, pattern='.*root', dbsInstance=None, readCache=False):
0024
0025     #print user, dataset, pattern, dbsInstance
0026     data = createMyDataset(user, dataset, pattern, dbsInstance, readCache)
0027
0028     source = cms.Source(
0029         "PoolSource",
0030         noEventSort = cms.untracked.bool(True),
0031         duplicateCheckMode = cms.untracked.string("noDuplicateCheck"),
0032         fileNames = cms.untracked.vstring()
0033         )
0034
0035     #print data.listOfGoodFiles()
0036     source.fileNames.extend( data.listOfGoodFiles() )
0037
0038     return source
0039 ### MM