root/hodgestar/PythonCode/SutekhBobs/MichaelsCSVParser.py

# MichaelsCSVParser.py
# -*- coding: utf8 -*-
# vim:fileencoding=utf8 ai ts=4 sts=4 et sw=4
# Copyright 2008 Simon Cross <hodgestar@gmail.com>
# GPL - see COPYING for details

"""
Parse cards from a CSV file.
"""

from sutekh.core.CardSetHolder import CardSetHolder
from sutekh.core.CardLookup import DEFAULT_LOOKUP
from sqlobject import sqlhub
import csv

class MichaelsCSVParser(object):
    """Parser cards from a CSV file into a CardSetHolder.

       Cards should be listed in columns and specify at least the card name and
       card count. Each row may optionally include the name of the expansion
       the card comes from.
       """

    def __init__(self):
        self.oCS = CardSetHolder()

    def _commit_holder(self, oCardLookup):
        """Commit contents of the card set holder to
           the database"""
        oOldConn = sqlhub.processConnection
        sqlhub.processConnection = oOldConn.transaction()

        self.oCS.create_physical_cl(oCardLookup)

        sqlhub.processConnection.commit()
        sqlhub.processConnection = oOldConn

    def _process_row(self,iName,iAdv,iExpStart,aExps,aRow):
        sName = aRow[iName].strip()
        if iAdv is not None and aRow[iAdv].strip():
            sName += " (Advanced)"

        if not sName:
            return # skip rows with no name

        for sExp, sCount in zip(aExps,aRow[iExpStart:]):
            try:
                iCount = int(sCount)
                self.oCS.add(iCount, sName, sExp)
            except StandardError, e:
                pass

    def parse(self, fIn, oCardLookup=DEFAULT_LOOKUP):
        oCsvFile = csv.reader(fIn)

        oIter = iter(oCsvFile)

        aCounts = oIter.next()
        aHeaders = oIter.next()

        iName = aHeaders.index('Name')

        try:
            iAdv = aHeaders.index('Adv')
        except ValueError, e:
            iAdv = None

        iExpStart = aHeaders.index('Jyhad')
        aExps = aHeaders[iExpStart:]
        # Jyhad,VTES,DS,AH,Sabbat,SW,FN,BL,CE,Anarchs,BH,Gehenna,Tenth,KMW,LoB,NoR,Third,SoC,LotN,Promo

        for aRow in oIter:
            try:
                self._process_row(iName, iAdv, iExpStart, aExps, aRow)
            except ValueError, e:
                raise ValueError("Line %d in CSV file could not be parsed (%s)" % (oCsvFile.line_num, e))

        self._commit_holder(oCardLookup)