diff -r 000000000000 -r de9c2efb9d02 Helpviewer/Passwords/PasswordModel.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Helpviewer/Passwords/PasswordModel.py Mon Dec 28 16:03:33 2009 +0000 @@ -0,0 +1,151 @@ +# -*- coding: utf-8 -*- + +# Copyright (c) 2009 Detlev Offenbach <detlev@die-offenbachs.de> +# + +""" +Module implementing a model for password management. +""" + +from PyQt4.QtCore import * + +class PasswordModel(QAbstractTableModel): + """ + Class implementing a model for password management. + """ + def __init__(self, manager, parent = None): + """ + Constructor + + @param manager reference to the password manager (PasswordManager) + @param parent reference to the parent object (QObject) + """ + QAbstractTableModel.__init__(self, parent) + + self.__manager = manager + self.connect(manager, SIGNAL("changed()"), self.__passwordsChanged) + + self.__headers = [ + self.trUtf8("Website"), + self.trUtf8("Username"), + self.trUtf8("Password") + ] + + self.__showPasswords = False + + def setShowPasswords(self, on): + """ + Public methods to show passwords. + + @param on flag indicating if passwords shall be shown (boolean) + """ + self.__showPasswords = on + self.reset() + + def showPasswords(self): + """ + Public method to indicate, if passwords shall be shown. + + @return flag indicating if passwords shall be shown (boolean) + """ + return self.__showPasswords + + def __passwordsChanged(self): + """ + Private slot handling a change of the registered passwords. + """ + self.reset() + + def removeRows(self, row, count, parent = QModelIndex()): + """ + Public method to remove entries from the model. + + @param row start row (integer) + @param count number of rows to remove (integer) + @param parent parent index (QModelIndex) + @return flag indicating success (boolean) + """ + if parent.isValid(): + return False + + if count <= 0: + return False + + lastRow = row + count - 1 + + self.beginRemoveRows(parent, row, lastRow) + + siteList = self.__manager.allSiteNames() + for index in range(row, lastRow + 1): + self.__manager.removePassword(siteList[index]) + + # removeEngine emits changed() + #self.endRemoveRows() + + return True + + def rowCount(self, parent = QModelIndex()): + """ + Public method to get the number of rows of the model. + + @param parent parent index (QModelIndex) + @return number of rows (integer) + """ + if parent.isValid(): + return 0 + else: + return self.__manager.sitesCount() + + def columnCount(self, parent = QModelIndex()): + """ + Public method to get the number of columns of the model. + + @param parent parent index (QModelIndex) + @return number of columns (integer) + """ + if self.__showPasswords: + return 3 + else: + return 2 + + def data(self, index, role): + """ + Public method to get data from the model. + + @param index index to get data for (QModelIndex) + @param role role of the data to retrieve (integer) + @return requested data (QVariant) + """ + if index.row() >= self.__manager.sitesCount() or index.row() < 0: + return QVariant() + + site = self.__manager.allSiteNames()[index.row()] + siteInfo = self.__manager.siteInfo(site) + + if siteInfo is None: + return QVariant() + + if role == Qt.DisplayRole: + if index.column() == 0: + return QVariant(site) + elif index.column() in [1, 2]: + return QVariant(siteInfo[index.column() - 1]) + + return QVariant() + + def headerData(self, section, orientation, role = Qt.DisplayRole): + """ + Public method to get the header data. + + @param section section number (integer) + @param orientation header orientation (Qt.Orientation) + @param role data role (integer) + @return header data (QVariant) + """ + if orientation == Qt.Horizontal and role == Qt.DisplayRole: + try: + return QVariant(self.__headers[section]) + except IndexError: + pass + + return QVariant()