1 files changed, 147 insertions, 0 deletions
diff --git a/examples/widgets/itemviews/addressbook/tablemodel.py b/examples/widgets/itemviews/addressbook/tablemodel.py
new file mode 100644
index 000000000..24316f981
--- /dev/null
+++ b/examples/widgets/itemviews/addressbook/tablemodel.py
@@ -0,0 +1,147 @@
+#!/usr/bin/python
+
+#############################################################################
+##
+## Copyright (C) 2011 Arun Srinivasan <rulfzid@gmail.com>
+## Copyright (C) 2016 The Qt Company Ltd.
+## Contact: http://www.qt.io/licensing/
+##
+## This file is part of the PySide examples of the Qt Toolkit.
+##
+## $QT_BEGIN_LICENSE:BSD$
+## You may use this file under the terms of the BSD license as follows:
+##
+## "Redistribution and use in source and binary forms, with or without
+## modification, are permitted provided that the following conditions are
+## met:
+##   * Redistributions of source code must retain the above copyright
+##     notice, this list of conditions and the following disclaimer.
+##   * Redistributions in binary form must reproduce the above copyright
+##     notice, this list of conditions and the following disclaimer in
+##     the documentation and/or other materials provided with the
+##     distribution.
+##   * Neither the name of The Qt Company Ltd nor the names of its
+##     contributors may be used to endorse or promote products derived
+##     from this software without specific prior written permission.
+##
+##
+## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+## "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+## LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+## A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+## OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+## SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+## LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+## DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+## THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+## (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+## OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
+##
+## $QT_END_LICENSE$
+##
+#############################################################################
+
+from PySide2.QtCore import (Qt, QAbstractTableModel, QModelIndex)
+
+class TableModel(QAbstractTableModel):
+
+    def __init__(self, addresses=None, parent=None):
+        super(TableModel, self).__init__(parent)
+
+        if addresses is None:
+            self.addresses = []
+        else:
+            self.addresses = addresses
+
+    def rowCount(self, index=QModelIndex()):
+        """ Returns the number of rows the model holds. """
+        return len(self.addresses)
+
+    def columnCount(self, index=QModelIndex()):
+        """ Returns the number of columns the model holds. """
+        return 2
+
+    def data(self, index, role=Qt.DisplayRole):
+        """ Depending on the index and role given, return data. If not
+            returning data, return None (PySide equivalent of QT's
+            "invalid QVariant").
+        """
+        if not index.isValid():
+            return None
+
+        if not 0 <= index.row() < len(self.addresses):
+            return None
+
+        if role == Qt.DisplayRole:
+            name = self.addresses[index.row()]["name"]
+            address = self.addresses[index.row()]["address"]
+
+            if index.column() == 0:
+                return name
+            elif index.column() == 1:
+                return address
+
+        return None
+
+    def headerData(self, section, orientation, role=Qt.DisplayRole):
+        """ Set the headers to be displayed. """
+        if role != Qt.DisplayRole:
+            return None
+
+        if orientation == Qt.Horizontal:
+            if section == 0:
+                return "Name"
+            elif section == 1:
+                return "Address"
+
+        return None
+
+    def insertRows(self, position, rows=1, index=QModelIndex()):
+        """ Insert a row into the model. """
+        self.beginInsertRows(QModelIndex(), position, position + rows - 1)
+
+        for row in range(rows):
+            self.addresses.insert(position + row, {"name":"", "address":""})
+
+        self.endInsertRows()
+        return True
+
+    def removeRows(self, position, rows=1, index=QModelIndex()):
+        """ Remove a row from the model. """
+        self.beginRemoveRows(QModelIndex(), position, position + rows - 1)
+
+        del self.addresses[position:position+rows]
+
+        self.endRemoveRows()
+        return True
+
+    def setData(self, index, value, role=Qt.EditRole):
+        """ Adjust the data (set it to <value>) depending on the given
+            index and role.
+        """
+        if role != Qt.EditRole:
+            return False
+
+        if index.isValid() and 0 <= index.row() < len(self.addresses):
+            address = self.addresses[index.row()]
+            if index.column() == 0:
+                address["name"] = value
+            elif index.column() == 1:
+                address["address"] = value
+            else:
+                return False
+
+            self.dataChanged.emit(index, index, 0)
+            return True
+
+        return False
+
+    def flags(self, index):
+        """ Set the item flags at the given index. Seems like we're
+            implementing this function just to see how it's done, as we
+            manually adjust each tableView to have NoEditTriggers.
+        """
+        if not index.isValid():
+            return Qt.ItemIsEnabled
+        return Qt.ItemFlags(QAbstractTableModel.flags(self, index) |
+                            Qt.ItemIsEditable)