from Feed import Feed
from PyQt4 import QtCore, QtGui
from PyQt4.QtCore import QAbstractListModel, QModelIndex, QVariant, Qt
from Ui_MainWindow import Ui_MainWindow

class MainWindowController(QtGui.QMainWindow):
    def __init__(self, session=None):
        QtGui.QWidget.__init__(self, None)
        self.session = session
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        self.setupWidgets()
        
    def setupWidgets(self):
        self.setupFeedList()
        
    def setupFeedList(self):
        self.allFeeds = Feed.all(self.session)
        feedModel = DisplayModel(self, self.allFeeds, Feed.userPresentableString)
        self.ui.feedList.setModel(feedModel)
        self.connect(self.ui.feedList, QtCore.SIGNAL("clicked(QModelIndex)"), self.feedSelected)
        
    def feedSelected(self, index):
        feed = self.allFeeds[index.row()]
        print("feed selected: " + str(feed))

class DisplayModel(QAbstractListModel):
    def __init__(self, parent=None, list=None, displayFunction=None, **args):
        QAbstractListModel.__init__(self, parent, *args)
        self.list = list
        self.displayFunction = displayFunction
                
    def rowCount(self, parent=QModelIndex()):
        return len(self.list)
    
    def data(self, index, role): 
        if index.isValid() and role == Qt.DisplayRole:
            row = index.row()
            object = self.list[row]
            displayString = self.displayFunction(object)
            return QVariant(displayString)
        else: 
            return QVariant()