dat.py

import collections
import os
import xml.etree.ElementTree as etree

Dat = collections.namedtuple('Dat', 'name description manufacturer year cloneof isbios node')


def get(i, e):
    ll = i.find(e)
    return ll.text if ll is not None else None


def parseDat(file, logger):
    dats = dict()
    # Old format
    parser = etree.XMLParser(encoding="utf-8")
    games = etree.parse(file, parser=parser).findall(".//game")
    if len(games) > 0:  # remove systems with no games
        for g in games:
            isbios = True if 'isbios' in g.attrib and g.attrib['isbios'] == 'yes' else False
            cloneof = g.attrib['cloneof'] if 'cloneof' in g.attrib else None
            datEntry = Dat(g.attrib['name'], get(g, 'description'), get(g, 'manufacturer'), get(g, 'year'),
                           cloneof, isbios, g)
            dats[g.attrib['name']] = datEntry
    else:
        # New format 
        newparser = etree.XMLParser(encoding="utf-8")
        machines = etree.parse(file, parser=newparser).findall(".//machine")
        if len(machines) > 0:  # remove systems with no machines
            for g in machines:
                isbios = True if 'isbios' in g.attrib and g.attrib['isbios'] == 'yes' else False
                cloneof = g.attrib['cloneof'] if 'cloneof' in g.attrib else None
                datEntry = Dat(g.attrib['name'], get(g, 'description'), get(g, 'manufacturer'), get(g, 'year'),
                               cloneof, isbios, g)
                dats[g.attrib['name']] = datEntry

    logger.log('Dat ' + str(file) + ' : ' + str(len(dats)) + ' entries')
    return dats


def parseDats(scriptDir, dataDir, setDats, usingSystems, logger):
    dats = dict()
    for setKey in usingSystems:
        header = etree.parse(os.path.join(scriptDir, dataDir, setDats[setKey]),
                             etree.XMLParser(encoding="utf-8")).findall(".//header")
        if os.path.exists(setDats[setKey]):
            datPath = setDats[setKey]
        else:
            datPath = os.path.join(scriptDir, dataDir, setDats[setKey])
        dats[setKey] = parseDat(datPath, logger)
        dats[setKey + "Header"] = header[0]
    return dats