Initial Project Commit

[retroconv] / models.py

diff --git a/models.py b/models.py
new file mode 100644 (file)
index 0000000..f700add
--- /dev/null
+++ b/models.py
@@ -0,0 +1,67 @@
+from PyQt4 import *
+from PyQt4.QtCore import *
+from PyQt4.QtGui import *
+import sys
+
+
+class SentModel(QAbstractListModel):
+ def __init__(self, mlist):
+  QAbstractListModel.__init__(self)
+
+ 
+  self._items = mlist
+
+
+ def rowCount(self, parent = QModelIndex()):
+  return len(self._items)
+
+ 
+ 
+ def data(self, index, role = Qt.DisplayRole):
+  if role == Qt.DisplayRole:
+       return QVariant(self._items[index.row()].getRecName())
+  elif role == Qt.DecorationRole:
+       
+       #img = QImage()
+       #arr = QByteArray(self._items[index.row()].getImageData())
+       #img.loadFromData(arr)
+       
+       
+       if self._items[index.row()].img == "":
+               return QVariant(QImage("graphics/contact_trans_50.png"))
+       else:
+               return QVariant(self._items[index.row()].img)
+  else:
+       return QVariant()
+
+
+############################################################3
+       
+class RecvModel(QAbstractListModel):
+ def __init__(self, mlist):
+  QAbstractListModel.__init__(self)
+
+ 
+  self._items = mlist
+
+
+ def rowCount(self, parent = QModelIndex()):
+  return len(self._items)
+
+ 
+ 
+ def data(self, index, role = Qt.DisplayRole):
+  if role == Qt.DisplayRole:
+       return QVariant(self._items[index.row()].getSenderName())
+  elif role == Qt.DecorationRole:
+       
+       #img = QImage()
+       #arr = QByteArray(self._items[index.row()].getImageData())
+       #img.loadFromData(arr)
+       
+       if self._items[index.row()].img == "":
+               return QVariant(QImage("graphics/contact_trans_50.png"))
+       else:
+               return QVariant(self._items[index.row()].img)
+  else:
+       return QVariant()