root/hodgestar/PythonCode/SutekhBobs/MichaelImporter.py

# CSVImporter.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

import gtk
from sutekh.io.MichaelsCSVParser import MichaelsCSVParser
from sutekh.core.SutekhObjects import PhysicalCard, AbstractCardSet, PhysicalCardSet
from sutekh.core.CardLookup import LookupFailed
from sutekh.gui.PluginManager import CardListPlugin
from sutekh.gui.SutekhDialog import SutekhDialog, do_complaint_error

class MichaelImporter(CardListPlugin):
    dTableVersions = {
        AbstractCardSet: [3],
        PhysicalCardSet: [4],
        PhysicalCard: [2],
    }
    aModelsSupported = [AbstractCardSet, PhysicalCardSet, PhysicalCard]

    def __init__(self,*args,**kws):
        super(MichaelImporter,self).__init__(*args,**kws)

    def get_menu_item(self):
        """
        Overrides method from base class.
        """
        if not self.check_versions() or not self.check_model_type():
            return None
        iDF = gtk.MenuItem("Import Michael's CSV File")
        iDF.connect("activate", self.activate)
        return iDF

    def get_desired_menu(self):
        return "Plugins"

    def activate(self,oWidget):
        oDlg = self.make_dialog()
        oDlg.run()

    def make_dialog(self):
        self.oDlg = SutekhDialog("Choose CSV File",None,
                gtk.DIALOG_MODAL | gtk.DIALOG_DESTROY_WITH_PARENT,
                (gtk.STOCK_OK, gtk.RESPONSE_OK,
                 gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))

        self.oFileChooser = gtk.FileChooserWidget(gtk.FILE_CHOOSER_ACTION_OPEN)
        self.oDlg.vbox.pack_start(self.oFileChooser)

        self.oDlg.connect("response", self.handle_response)
        self.oDlg.set_size_request(400,400)
        self.oDlg.show_all()

        return self.oDlg

    def handle_response(self,oWidget,oResponse):
        if oResponse == gtk.RESPONSE_OK:

            oParser = MichaelsCSVParser()

            sFile = self.oFileChooser.get_filename()
            fIn = file(sFile,"rb")

            try:
                oParser.parse(fIn, oCardLookup=self.cardlookup)
            finally:
                fIn.close()

        self.oDlg.destroy()

# pylint: disable-msg=C0103
# accept plugin name
plugin = MichaelImporter