pickleFileParser.py

Go to the documentation of this file.

__author__="Aurelija"
__date__ ="$2010-08-13 14.28.08$"

import sys
import glob
import pickle
from os.path import join, split

rulesNames = []

if sys.platform[:3] == 'win':
    slash = "\\"
else:
    slash = "/"

def readPicFiles(directory, toSplit = False):

    ruleResult = {}
    rulesResults = {}
    picFiles = glob.glob(join(directory, "cmsCodeRule*.dat"))
    picFiles.sort()

    for file in picFiles:
        head, fileName = split(file)
        ruleName = fileName[11:-4]
        rulesNames.append(ruleName)

        file = open(file)
        ruleResult = pickle.load(file)
        if toSplit:
            ruleResult = splitToPackages(ruleResult)

        rulesResults[ruleName] = ruleResult

    return rulesResults


def splitToPackages(ruleResult):

    packageResult = []
    info = []

    if not ruleResult: return info

    ruleResult = ruleResult.items()
    ruleResult.sort()
    file, lines = ruleResult.pop(0)
    pathList = pathToList(file)
    package = slash.join(pathList[:2])
    packageResult.append((slash.join(pathList[2:]), lines))

    for file, lines in ruleResult:
        pathList = pathToList(file)
        head = slash.join(pathList[:2])
        tail = slash.join(pathList[2:])
        if package == head:
            packageResult.append((tail, lines))
        else:
            info.append((package, packageResult))
            packageResult = []
            package = head
            packageResult.append((tail, lines))
    info.append((package, packageResult))
    return info #list of (package, packageResult)

def pathToList(path):
    list = []
    head, tail = split(path)
    if tail != '':
        list.insert(0, tail)
    while head != '':
        head, tail = split(head)
        if tail != '':
            list.insert(0, tail)
        else:
            break
    return list