unread messages count

src/list_items.py

@@ -169,6 +169,8 @@ class ContactItem(QtGui.QWidget):
         self.status_message.setFont(font)
         self.connection_status = StatusCircle(self)
         self.connection_status.setGeometry(QtCore.QRect(243, 5, 32, 32))
+        self.messages = UnreadMessagesCount(self)
+        self.messages.setGeometry(QtCore.QRect(52, 50, 30, 20))
 
 
 class StatusCircle(QtGui.QWidget):
@@ -201,6 +203,30 @@ class StatusCircle(QtGui.QWidget):
         self.label.setPixmap(pixmap)
 
 
+class UnreadMessagesCount(QtGui.QWidget):
+
+    def __init__(self, parent=None):
+        super(UnreadMessagesCount, self).__init__(parent)
+        self.resize(30, 20)
+        self.label = QtGui.QLabel(self)
+        self.label.setGeometry(QtCore.QRect(0, 0, 30, 20))
+        self.label.setVisible(False)
+        font = QtGui.QFont()
+        font.setFamily("Times New Roman")
+        font.setPointSize(12)
+        font.setBold(True)
+        self.label.setFont(font)
+        self.label.setAlignment(QtCore.Qt.AlignVCenter | QtCore.Qt.AlignCenter)
+        self.label.setStyleSheet('QLabel { color: white; background-color: red; border-radius: 10; }')
+
+    def update(self, messages_count):
+        if messages_count:
+            self.label.setVisible(True)
+            self.label.setText(str(messages_count))
+        else:
+            self.label.setVisible(False)
+
+
 class FileTransferItem(QtGui.QListWidget):
 
     def __init__(self, file_name, size, time, user, friend_number, file_number, state, width, parent=None):
@@ -356,6 +382,9 @@ class FileTransferItem(QtGui.QListWidget):
                 self.setStyleSheet('QListWidget { border: 1px solid green; }')
                 self.state = state
 
+    def mark_as_sent(self):
+        return False
+
 
 class UnsentFileItem(FileTransferItem):
 
@@ -391,6 +420,9 @@ class InlineImageItem(QtGui.QWidget):
             self._image_label.setPixmap(pixmap)
             self.resize(QtCore.QSize(max_size, pixmap.height()))
 
+    def mark_as_sent(self):
+        return False
+