root2sqlite Namespace Reference

Functions
def	columnescape (s)
def	read_objects_root (rootfile)
def	save_keyvalue (dictionary, name)
def	tosqlite (x)
def	treetotable (ttree, name)

Variables
	args
	basic_objects
	db
	default
	description
	f
	help
	inf
	nan
	parser

Function Documentation

columnescape()

def root2sqlite.columnescape

(

s

)

Definition at line 46 of file root2sqlite.py.

def columnescape(s):
    # add whatever is not a valid column name here
    SQLKWDs = ["index"]
    if s.lower() in SQLKWDs:
        return s + "_"
    else:
        return s
 

Referenced by treetotable().

read_objects_root()

def root2sqlite.read_objects_root

(

rootfile

)

Definition at line 70 of file root2sqlite.py.

def read_objects_root(rootfile):
    xml_re = re.compile(r"^<(.+)>(.+)=(.+)<\/\1>$")
    def parse_directory(di):
        directory = rootfile.GetDirectory(di)
        for key in directory.GetListOfKeys():
            entry = key.GetName()
            rtype = key.GetClassName()
            fullpath = "%s/%s" % (di, entry) if di != "" else entry
            if (rtype == "TDirectoryFile"):
                for k, v, t in parse_directory(fullpath):
                    yield (k, v, t)
            else:
                obj = rootfile.Get(fullpath)
                if obj:
                    yield (fullpath, obj, rtype)
                else:
                    # special case to parse the xml abomination
                    m = xml_re.search(entry)
                    if m:
                        name = m.group(1)
                        typecode = m.group(2)
                        value = m.group(3)
                        fp = "%s/%s" % (di, name)
                        yield (fp, value, rtype)
                    else:
                        raise Exception("Invalid xml:" + entry)
    path_fix = re.compile(r"^\/Run \d+")
    for fullname, obj, rtype in parse_directory(""):
        yield fullname, obj, rtype
 

Referenced by save_keyvalue().

save_keyvalue()

def root2sqlite.save_keyvalue	(		dictionary,
			name
	)

Definition at line 100 of file root2sqlite.py.

def save_keyvalue(dictionary, name):
    name = name.replace("/", "_")
    create =  "CREATE TABLE %s(key, value);"  % name
    print(create)
    db.execute(create)
    data = []
    for k, v in dictionary.iteritems():
        vals = (unicode(k), tosqlite(v))
        data.append(vals)
    insert = "INSERT INTO %s(key, value) VALUES (?,?);" % name
    print(insert)
    db.executemany(insert, data)
    db.commit()
 
 

References edm.print(), read_objects_root(), tosqlite(), treetotable(), and betterConfigParser.unicode.

tosqlite()

def root2sqlite.tosqlite

(

x

)

Definition at line 23 of file root2sqlite.py.

def tosqlite(x):
    if isinstance(x, ROOT.string):
        try:
            return unicode(x.data())
        except:
            return buffer(x.data())
    if isinstance(x, int):
          return x
    if isinstance(x, float):
        return x
    if isinstance(x, long):
        return x
    else:
        try: 
            rootobj = unicode(ROOT.TBufferJSON.ConvertToJSON(x))
            # turns out ROOT does not generate valid JSON for NaN/inf
            clean = nan.sub('\\g<1>0\\g<2>', inf.sub('\\g<1>1e38\\g<2>', rootobj))
            obj = json.loads(clean)
            jsonobj = json.dumps(obj, allow_nan=False)
            return jsonobj
        except Exception as e:
            return json.dumps({"root2sqlite_error": e.__repr__(), "root2sqlite_object": x.__repr__()})
 

References edmScanValgrind.buffer, and betterConfigParser.unicode.

Referenced by save_keyvalue(), and treetotable().

treetotable()

def root2sqlite.treetotable	(		ttree,
			name
	)

Definition at line 54 of file root2sqlite.py.

def treetotable(ttree, name):
    name = name.replace("/", "_")
    branches = [b.GetName() for b in ttree.GetListOfBranches()]
    colnames = ", ".join(columnescape(b) for b in branches)
    create =  "CREATE TABLE %s(%s);"  % (name, colnames)
    print(create)
    db.execute(create)
    data = []
    for i in range(ttree.GetEntries()):
        ttree.GetEntry(i)
        vals = tuple([tosqlite(getattr(ttree, b)) for b in branches])
        data.append(vals)
    insert = "INSERT INTO %s(%s) VALUES (%s);" % (name, colnames, ",".join(["?"] * len(branches)))
    print(insert)
    db.executemany(insert, data)
 

References columnescape(), join(), edm.print(), FastTimerService_cff.range, and tosqlite().

Referenced by save_keyvalue().

Variable Documentation

args

root2sqlite.args

Definition at line 13 of file root2sqlite.py.

basic_objects

root2sqlite.basic_objects

Definition at line 18 of file root2sqlite.py.

db

root2sqlite.db

Definition at line 16 of file root2sqlite.py.

default

root2sqlite.default

Definition at line 12 of file root2sqlite.py.

description

root2sqlite.description

Definition at line 9 of file root2sqlite.py.

f

root2sqlite.f

Definition at line 15 of file root2sqlite.py.

help

root2sqlite.help

Definition at line 11 of file root2sqlite.py.

inf

root2sqlite.inf

Definition at line 20 of file root2sqlite.py.

nan

root2sqlite.nan

Definition at line 21 of file root2sqlite.py.

parser

root2sqlite.parser

Definition at line 9 of file root2sqlite.py.