removeValsFromFile.py

Go to the documentation of this file.

#!/usr/bin/env python 
 
import uploadTools
 
basepath="/afs/cern.ch/atlas/groups/Generators/CrossSectionInfo/ScriptsCentralPage/"
 
def main():
 
    # parse options from command line
    from optparse import OptionParser
    parser = OptionParser(usage = "usage: %prog arguments", version="%prog")
    parser.add_option("-v", "--verbose",    action="store_true", dest="verbose",    help="Set verbose mode (default: %default)")
    parser.add_option("-y", "--alwaysyes",  action="store_true", dest="alwaysyes",  help="Always answer yes to checks (default: %default)")
    parser.add_option("-i", "--infile",     action="store",      dest="infile",     help="Input file (default: %default)")
    parser.add_option("-x", "--xsreffile",  action="store",      dest="xsreffile",  help="Cross section reference file (default: %default)")
    parser.add_option("-t", "--istest",     action="store",      dest="istest",     help="Is test (no actual upload to AMI) (default: %default)")
    parser.add_option("-d", "--delim",      action="store",      dest="delim",      help="Delimiter for input file (default: %default)")
    parser.add_option("-e", "--explanation",action="store",      dest="explanation",help="Comment for explanation field (default: %default)")
    
    parser.set_defaults(verbose=False,alwaysyes=False,infile=None,xsreffile=basepath+"InputInformation/CrossSectionReference",istest="TRUE",delim=';',explanation=None)
 
    (opts, args) = parser.parse_args()
 
    if not opts.explanation:
        print "ERROR: No explanation found - Should be JIRA ticket or similar"
        return
    elif 'CENTRPAGE' not in opts.explanation:
        print "WARNING: Explanation found, but does not include CENTRPAGE - this should usually be a JIRA ticket"
 
    if opts.istest=="FALSE":
        opts.istest=False
 
    UT=uploadTools.UT(opts=opts)
    cols=UT.colors
 
    if not opts.istest:
        print cols.FAIL+"INFO: THIS IS NOT A TEST - UPLOADING FOR REALS!"+cols.ENDC
    else:
        print cols.OKGREEN+"INFO: THIS IS A TEST - Nothing will be uploaded!"+cols.ENDC
 
 
 
          
    currentfile=UT.getCurrentVals(UT.getDSListFromFile(opts.infile))
    UT.currentVals=UT.getDetailsFromFile(currentfile)
    print UT.currentVals
 
    uploadVals=UT.getDetailsFromFile(opts.infile)
    print uploadVals
 
    for ds in uploadVals:
        if ds in UT.currentVals:
            print cols.OKGREEN+"INFO: Working on datset %s"%(ds)+cols.ENDC
            for param in uploadVals[ds]:
                if param in UT.currentVals[ds]:
                    if len(uploadVals[ds][param].strip()) or uploadVals[ds][param].strip()=="#UNKNOWN#":
                        if uploadVals[ds][param]!=UT.currentVals[ds][param]:
                            print cols.WARNING+"INFO: Upload %s = %s (current val = %s)"%(param,uploadVals[ds][param],UT.currentVals[ds][param])+cols.ENDC
                            UT.uploadValue(ds,param,uploadVals[ds][param],opts.explanation,uploadVals[ds]["crossSectionRef"])
                        else:
                            print "INFO: Inputted %s value is identical to current DB value (%s) - ignoring"%(param,UT.currentVals[ds][param])
                    else:
                        print "INFO: %s value empty or unknown - ignoring"%(param)
                else:
                    print "WARNING: Parameter %s not found in current AMI vals dictionary for %s"%(param,ds)
        else:
            print "WARNING: Parameter dataset %s not found in current AMI val dictionary"%(ds)
 
    #UT.updateXS("test","mc15","1234","Just a test")
 
if __name__ == '__main__':
    main()