CMSSW_4_4_3_patch1/src/CondCore/TagCollection/python/CommonUtils.py

Go to the documentation of this file.

def inventoryTableName():
        return 'TAGINVENTORY_TABLE'
def inventoryIDTableName():
        return 'TAGINVENTORY_IDS'
def treeTableName(treename):
        return 'TAGTREE_TABLE_'+str.upper(treename)
def treeIDTableName(treename):
        return 'TAGTREE_'+str.upper(treename)+'_IDS'
def commentTableName():
        return 'ENTRYCOMMENT_TABLE'

import coral
def dropAllTreeTables(dbsession):
    """drop all tagtree related tables
    """
    try:
        dbsession.transaction().start(False)
        tablelist = dbsession.nominalSchema().listTables()
        for tname in tablelist:
           if tname.find('TAGTREE_') != -1:
              dbsession.nominalSchema().dropTable(tname)
        dbsession.transaction().commit()
    except Exception, e:
        raise Exception, str(e)
            
def tagInTrees(dbsession,tagname,pfn=''):
    """returns the tree names which contain the given tag
       select tagid from taginventory_table where tagname=tagname
       select count(*) from tablename where tablename.tagid=tagid
    """
    try:
        dbsession.transaction().start(True)    
        invquery = dbsession.nominalSchema().tableHandle(inventoryTableName()).newQuery()
        conditionbindDict=coral.AttributeList()
        conditionbindDict.extend('tagname','string')
        conditionbindDict['tagname'].setData(tagname)
        condition='tagname = :tagname'
        if len(pfn) !=0 :
           condition+=' AND pfn = :pfn'
           conditionbindDict.extend('pfn','string')
           conditionbindDict['pfn'].setData(pfn)
        invquery.setCondition(condition,conditionbindDict)
        invquery.addToOutputList('tagid')
        invquery.addToOutputList('pfn')
        cursor = invquery.execute()
        tagidmap={}
        while ( cursor.next() ):
            tagid=cursor.currentRow()['tagid'].data()
            pfn=cursor.currentRow()['pfn'].data()
            tagidmap[pfn]=tagid
        cursor.close()
        dbsession.transaction().commit()
        del invquery
        if len(tagidmap)==0:
           return tagidmap
   
        result={}
        treetablelist=[]
        dbsession.transaction().start(True)    
        tablelist = dbsession.nominalSchema().listTables()
        for t in tablelist:
           if t.find('TAGTREE_TABLE_')!= -1:
                   treetablelist.append(t)
        for (pfn,tagid) in tagidmap.items():
           result[pfn]=[]
           condition='tagid = :tagid'
           for t in treetablelist:
              conditionBind=coral.AttributeList()
              conditionBind.extend('tagid','unsigned long')
              conditionBind['tagid'].setData(tagid)
              q=dbsession.nominalSchema().tableHandle(t).newQuery()
              q.addToOutputList('count(*)','count')
              myresult=coral.AttributeList() 
              myresult.extend('count','unsigned long')
              q.defineOutput(myresult)
              q.setCondition(condition,conditionBind)
              cr=q.execute()
              while (cr.next()):
                if cr.currentRow()['count'].data()!=0:
                  result[pfn].append(t[len('TAGTREE_TABLE_'):])
              cr.close()
              del q
        dbsession.transaction().commit()        
        return result    
    except Exception, e:
        raise Exception, str(e)

if __name__ == "__main__":
    #context = coral.Context()
    #context.setVerbosityLevel( 'ERROR' )
    svc = coral.ConnectionService()
    session = svc.connect( 'sqlite_file:testInventory.db',
                           accessMode = coral.access_Update )
    try:
        print 'TEST 1'
        intrees=tagInTrees(session,'Tracker_Geometry_CRUZET3')
        print intrees
        print 'TEST 2'
        hello=tagInTrees(session,'Tracker_Geometry_CRUZ3')
        print hello
        print 'TEST 3'
        kitty=tagInTrees(session,'Tracker_Geometry_CRUZET3','pfnme')
        print kitty
        print 'TEST 4'
        mikey=tagInTrees(session,'Tracker_Geometry_CRUZET3','frontier://(proxyurl=https://localhost:3128)(serverurl=https://frontier1.cms:8000/FrontierOnProd)(serverurl=https://frontier2.cms:8000/FrontierOnProd)(retrieve-ziplevel=0)/CMS_COND_20X_ALIGNMENT')
        print mikey
        del session
    except Exception, e:
        print "Failed in unit test"
        print str(e)
        del session