parsePQSLOutput.py

Go to the documentation of this file.

#!/bin/env python
print "run  : --- parsePSQL output"
# This is output got from the database, an example query would be:
# psql -U nova_reader -d nova_prod -h ifdbprod.fnal.gov -p 5433 -c "SELECT * FROM fardet.runs WHERE run > 13800 AND partition = 1;" -o log.txt
########################### Option parser
from optparse import OptionParser
parser = OptionParser()
parser.add_option("-i", "--input",   help="input log file",       action="store", type=str, dest="input",    default=False)
parser.add_option("-v", "--verbose", help="turn on verbose mode", action="store_true",      dest="verbose",  default=False)
(options, args) = parser.parse_args()
if options.verbose:
    print "run  : --- Options"
    print "run  :     input:        ",options.input
    print "run  :     verbose mode: ",options.verbose
assert (options.input), "No input file provided, specify one with -i MYFILE"
########################### Analyse
lines = open(options.input, "rb").readlines()
print "run  : Got %i lines"%len(lines)
good_lines  = 0
good_events = 0
good_subruns = 0
for line in lines:
    tokens = line.split("|")
    if len(tokens) != 19:  continue
    if "run" in tokens[0]:  continue
    if tokens[0].isspace(): continue
    bad_run = False
    for i in range(4):
        if tokens[i].isspace(): bad_run = True
    if bad_run: continue
    #print tokens[:4]
    run       = int(tokens[0])
    nsubruns  = int(tokens[1])
    n_events  = int(tokens[2])
    partition = int(tokens[3])
    assert(partition == 1)
    #print tokens[:4]
    good_lines+=1
    good_events+=n_events
    good_subruns+=nsubruns
print "run  : Got %i good lines"%good_lines
print "run  : Got %i good events"%good_events
print "run  : Got %i good subruns"%good_subruns
########################### Done
print "run  : --- Done\n"