extractor.py

import os
from defusedxml import ElementTree


def has_curef(infile):
    with open(infile, "r") as afile:
        data = afile.read()
    sentinel = True if "CUREF" in data else False
    return sentinel


def searcher(indir):
    candidates = [x for x in os.listdir(indir) if x.endswith(".xml")]
    passes = [y for y in candidates if has_curef(y)]
    return passes


def extractor(xmlfile):
    with open(xmlfile, "r") as afile:
        data = afile.read()
    root = ElementTree.fromstring(data)
    curef = root.find("CUREF")
    tvver = root.find("VERSION").find("TV")
    return curef.text, tvver.text


def main():
    passes = searcher(os.getcwd())
    verdict = {}
    for passfile in passes:
        curef, tvver = extractor(passfile)
        verdict[curef] = tvver
    for prd in sorted(verdict.keys()):
        print("PRD-{0}: {1}".format(prd, verdict[prd]))


if __name__ == "__main__":
    main()