| 1 |
import os, sys, warnings
|
| 2 |
from optparse import OptionParser
|
| 3 |
from BetterConfigParser import BetterConfigParser
|
| 4 |
from samplesclass import sample
|
| 5 |
|
| 6 |
def parse_info(samples_config,samples_path):
|
| 7 |
try:
|
| 8 |
os.stat(samples_config)
|
| 9 |
except:
|
| 10 |
raise Exception('config file is wrong/missing')
|
| 11 |
|
| 12 |
if '/pnfs/psi.ch/cms/' in samples_path:
|
| 13 |
T3 = True
|
| 14 |
_,p2=samples_path.split('/pnfs/')
|
| 15 |
t3_path = '/pnfs/'+p2.strip('\n')
|
| 16 |
else:
|
| 17 |
T3 = False
|
| 18 |
|
| 19 |
config = BetterConfigParser()
|
| 20 |
config.read(samples_config)
|
| 21 |
|
| 22 |
newprefix=config.get('General','newprefix')
|
| 23 |
lumi=float(config.get('General','lumi'))
|
| 24 |
weightexpression=config.get('General','weightexpression')
|
| 25 |
|
| 26 |
info = []
|
| 27 |
|
| 28 |
fileslist=[]
|
| 29 |
if T3:
|
| 30 |
ls = os.popen("lcg-ls -b -D srmv2 -l srm://t3se01.psi.ch:8443/srm/managerv2?SFN="+t3_path)
|
| 31 |
else:
|
| 32 |
ls = os.popen("ls -l "+samples_path)
|
| 33 |
|
| 34 |
for line in ls.readlines():
|
| 35 |
fileslist.append(line)
|
| 36 |
|
| 37 |
|
| 38 |
for Sample in config.sections():
|
| 39 |
|
| 40 |
if not config.has_option(Sample,'infile'): continue
|
| 41 |
#if not SamplesList == [''] and not config.get(Sample,'sampleName') in SamplesList: continue
|
| 42 |
infile = config.get(Sample,'infile')
|
| 43 |
sampleName = config.get(Sample,'sampleName')
|
| 44 |
|
| 45 |
if any(infile in file for file in fileslist):
|
| 46 |
print 'Sample %s is present'%(sampleName)
|
| 47 |
else:
|
| 48 |
warnings.warn('Sample %s is NOT! present'%(sampleName))
|
| 49 |
warnings.warn("File %s not present"%(infile))
|
| 50 |
#Initialize samplecalss element
|
| 51 |
sampleType = config.get(Sample,'sampleType')
|
| 52 |
cut = config.get(Sample, 'cut')
|
| 53 |
info.append(sample(sampleName,sampleType))
|
| 54 |
|
| 55 |
info[-1].addtreecut(cut)
|
| 56 |
info[-1].path=samples_path
|
| 57 |
info[-1].identifier=infile
|
| 58 |
info[-1].weightexpression=weightexpression
|
| 59 |
info[-1].lumi=lumi
|
| 60 |
info[-1].prefix=newprefix
|
| 61 |
|
| 62 |
if eval(config.get(Sample,'subsamples')):
|
| 63 |
info[-1].subsamples=True
|
| 64 |
info[-1].group = eval((config.get(Sample,'sampleGroup')))
|
| 65 |
info[-1].subcuts = eval((config.get(Sample, 'subcuts')))
|
| 66 |
info[-1].subnames = eval((config.get(Sample, 'subnames')))
|
| 67 |
if sampleType != 'DATA':
|
| 68 |
info[-1].sf = eval((config.get(Sample, 'SF')))
|
| 69 |
info[-1].xsec = eval((config.get(Sample,'xSec')))
|
| 70 |
else:
|
| 71 |
info[-1].group = config.get(Sample,'sampleGroup')
|
| 72 |
if sampleType != 'DATA':
|
| 73 |
info[-1].sf = config.get(Sample, 'SF')
|
| 74 |
info[-1].xsec = config.get(Sample,'xSec')
|
| 75 |
|
| 76 |
return info
|
