git.maemo.org Git - retroconv/blob - models.py

   1 from PyQt4 import *
   2 from PyQt4.QtCore import *
   3 from PyQt4.QtGui import *
   4 import sys
   5
   6
   7 class SentModel(QAbstractListModel):
   8  def __init__(self, mlist):
   9   QAbstractListModel.__init__(self)
  10
  11
  12   self._items = mlist
  13
  14
  15  def rowCount(self, parent = QModelIndex()):
  16   return len(self._items)
  17
  18
  19
  20  def data(self, index, role = Qt.DisplayRole):
  21   if role == Qt.DisplayRole:
  22         return QVariant(self._items[index.row()].getRecName())
  23   elif role == Qt.DecorationRole:
  24
  25         #img = QImage()
  26         #arr = QByteArray(self._items[index.row()].getImageData())
  27         #img.loadFromData(arr)
  28
  29
  30         if self._items[index.row()].img == "":
  31                 return QVariant(QImage("graphics/contact_trans_50.png"))
  32         else:
  33                 return QVariant(self._items[index.row()].img)
  34   else:
  35         return QVariant()
  36
  37
  38 ############################################################3
  39
  40 class RecvModel(QAbstractListModel):
  41  def __init__(self, mlist):
  42   QAbstractListModel.__init__(self)
  43
  44
  45   self._items = mlist
  46
  47
  48  def rowCount(self, parent = QModelIndex()):
  49   return len(self._items)
  50
  51
  52
  53  def data(self, index, role = Qt.DisplayRole):
  54   if role == Qt.DisplayRole:
  55         return QVariant(self._items[index.row()].getSenderName())
  56   elif role == Qt.DecorationRole:
  57
  58         #img = QImage()
  59         #arr = QByteArray(self._items[index.row()].getImageData())
  60         #img.loadFromData(arr)
  61
  62         if self._items[index.row()].img == "":
  63                 return QVariant(QImage("graphics/contact_trans_50.png"))
  64         else:
  65                 return QVariant(self._items[index.row()].img)
  66   else:
  67         return QVariant()