diff frontends/src/quick_frontend/quick_contact_list.py @ 1408:8a7145138330

quick_frontend, primitivus: move code to remove a new message alert from primitivus to quick_frontend
author souliane <souliane@mailoo.org>
date Thu, 16 Apr 2015 13:31:14 +0200
parents 5c3d1d970b94
children 3265a2639182
line wrap: on
line diff
--- a/frontends/src/quick_frontend/quick_contact_list.py	Thu Apr 09 11:27:13 2015 +0200
+++ b/frontends/src/quick_frontend/quick_contact_list.py	Thu Apr 16 13:31:14 2015 +0200
@@ -444,6 +444,22 @@
         self._alerts.add(entity)
         self.update()
 
+    def removeAlert(self, entity, use_bare_jid=True):
+        """Eventually remove an alert on the entity (usually for a waiting message).
+
+        @param entity(jid.JID): entity (resource is significant)
+        @param use_bare_jid (bool): if True, ignore the resource
+        """
+        if use_bare_jid:
+            to_remove = set()
+            for alert_entity in self._alerts:
+                if alert_entity.bare == entity.bare:
+                    to_remove.add(alert_entity)
+            self._alerts.difference_update(to_remove)
+        else:
+            self._alerts.discard(entity)
+        self.update()
+
     def _showOfflineContacts(self, show_str):
         self.showOfflineContacts(C.bool(show_str))