From 99d5e25f9bb9a6cbe51973d8409691ad8e5482a1 Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Wed, 23 Sep 2015 20:28:09 +0200
Subject: Fix various roster count issues

disconnections not clearing the number of connected contacts, leading to
an ever-increasing count, and roster count not always being setup
---
 src/roster.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'src/roster.py')

diff --git a/src/roster.py b/src/roster.py
index 7866655f..ba7da63e 100644
--- a/src/roster.py
+++ b/src/roster.py
@@ -149,9 +149,14 @@ class Roster(object):
             contact = self.get_and_set(key)
             if key != self.jid and (contact and self.exists(contact)):
                 l.append(key)
-        self.length = len(l)
+        self.update_size(l)
         return l
 
+    def update_size(self, jids=None):
+        if jids is None:
+            jids = self.jids()
+        self.length = len(jids)
+
     def get_contacts(self):
         """
         Return a list of all the contacts
-- 
cgit v1.2.3