python.scripts.check_sg Namespace Reference

Functions
def	main (args)

Variables
	__doc__
	__author__
	name
	nargs
	help
	default

Function Documentation

main()

def python.scripts.check_sg.main

(

args

)

read a POOL file and dump the DataHeader's content

ex:
 $ check-sg aod.pool.root
 $ check-sg /castor/foo.pool
 $ check-sg root://castoratlas//castor/foo.pool
 $ check-sg LFN:ttbar.pool

Definition at line 26 of file check_sg.py.

def main(args):
    """read a POOL file and dump the DataHeader's content
 
    ex:
     $ check-sg aod.pool.root
     $ check-sg /castor/foo.pool
     $ check-sg root://castoratlas//castor/foo.pool
     $ check-sg LFN:ttbar.pool
    """
    files = args.files
    if isinstance(files, str):
        files = [files]
 
    from collections import defaultdict
    from PyUtils.PoolFile import extract_items
    import os
    import sys
 
    for i,f in enumerate(files):
        files[i] = os.path.expandvars(os.path.expanduser(f))
 
    exitcode = 0
    for fname in files:
        try:
            print ("## checking [%s]..." % (fname,))
 
            item_list = defaultdict(list)
            for name, key in extract_items(fname, verbose=False, items_type='eventdata'):
                item_list[name].append(key)
 
            print ("="*80)
            print ("%40s%s%-40s" % ("Container type", " | ","StoreGate keys"))
            print ("%40s%s%-40s" % ("-"*40, "-+-", "-"*(40-3)))
            for name, sgkeys in sorted(item_list.items()):
                # decode b-strings to strings
                sgkeys = [(k.decode("utf-8") if not isinstance(k, str) else k) for k in sgkeys]
                print ("%40s%s%-40s" % (name, " | ", ', '.join(sorted(sgkeys))))
            print ("="*80)
            if args.output:
                outFileName = args.output
                outFileName = os.path.expanduser(outFileName)
                outFileName = os.path.expandvars(outFileName)
                print ("## saving report into [%s]..." % (outFileName,))
                if os.path.splitext(outFileName)[1] in ('.pkl', '.dat'):
                    import shelve
                    if os.path.exists(outFileName):
                        os.remove(outFileName)
                    with shelve.open(outFileName) as db:
                        db['eventdata_items'] = item_list
                else:
                    from pprint import pprint
                    with open( outFileName, 'w' ) as txt:
                        pprint( dict(item_list), txt )
 
        except Exception as e:
            print ("## Caught exception [%s] !!" % str(e.__class__))
            print ("## What:",e)
            print (sys.exc_info()[0])
            print (sys.exc_info()[1])
            import traceback
            traceback.print_exc()
            exitcode = 1
            pass
 
        if len(files) > 1:
            print ()
        pass # loop over fileNames
    
    print ("## Bye.")
    return exitcode
 

Variable Documentation

__author__

python.scripts.check_sg.__author__

private

Definition at line 9 of file check_sg.py.

__doc__

python.scripts.check_sg.__doc__

private

Definition at line 8 of file check_sg.py.

default

python.scripts.check_sg.default

Definition at line 19 of file check_sg.py.

help

python.scripts.check_sg.help

Definition at line 17 of file check_sg.py.

name

python.scripts.check_sg.name

Definition at line 15 of file check_sg.py.

nargs

python.scripts.check_sg.nargs

Definition at line 16 of file check_sg.py.