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