lumiData2.py

Go to the documentation of this file.

#!/usr/bin/env python
VERSION='2.00'
import os,sys
import coral
from RecoLuminosity.LumiDB import argparse,lumiCalcAPI,sessionManager

def listRemoveDuplicate(inlist):
    d={}
    for x in inlist:
        d[x]=x
    return d.values()

def main():
    parser = argparse.ArgumentParser(prog=os.path.basename(sys.argv[0]),description="list lumi data availability")
    # add required arguments
    parser.add_argument('-c',dest='connect',action='store',required=True,help='connect string to lumiDB')
    # add optional arguments
    parser.add_argument('-P',dest='authpath',action='store',help='path to authentication file')
    parser.add_argument('-siteconfpath',dest='siteconfpath',action='store',
            default=None,
                        required=False,
            help='specific path to site-local-config.xml file, optional. If path undefined, fallback to cern proxy&server')
    parser.add_argument('action',choices=['listrun'],help='command actions')
    parser.add_argument('--minrun',
            dest='minrun',
            action='store',
            type=int,
            help='min run number')
    parser.add_argument('--maxrun',
            dest='maxrun',
            action='store',
            type=int,
            help='max run number')
    parser.add_argument('--minfill',
            dest='minfill',
            type=int,
            action='store',
            help='min fill number')
    parser.add_argument('--maxfill',
            dest='maxfill',
            type=int,
            action='store',
            help='max fill number')
    parser.add_argument('--verbose',
            dest='verbose',
            action='store_true',
            help='verbose mode for printing' )
    parser.add_argument('--debug',dest='debug',action='store_true',help='debug')
    # parse arguments
    options=parser.parse_args()
    if options.authpath:
        os.environ['CORAL_AUTH_PATH'] = options.authpath
    svc=sessionManager.sessionManager(options.connect,authpath=options.authpath,debugON=options.debug)
    session=svc.openSession(isReadOnly=True,cpp2sqltype=[('unsigned int','NUMBER(10)'),('unsigned long long','NUMBER(20)')])
    reqTrg=True
    reqHLT=False
    session.transaction().start(True)
    schema=session.nominalSchema()
    runlist=lumiCalcAPI.runList(schema,None,runmin=options.minrun,runmax=options.maxrun,startT=None,stopT=None,l1keyPattern=None,hltkeyPattern=None,amodetag=None,nominalEnergy=None,energyFlut=None,requiretrg=reqTrg,requirehlt=reqHLT)
    session.transaction().commit()
    if options.action == 'listrun':
        if not runlist:
            print '[]'
        sys.exit(0)
        singlelist=sorted(listRemoveDuplicate(runlist))
        print singlelist
    del session
    del svc
if __name__=='__main__':
    main()