[IMP] mail: Add counter in starred

2 files changed, 15 insertions, 4 deletions

diff --git a/addons/mail/models/mail_message.py b/addons/mail/models/mail_message.py
index 949cef23138..c054b6e10a3 100644
--- a/addons/mail/models/mail_message.py
+++ b/addons/mail/models/mail_message.py
@@ -237,14 +237,13 @@ class Message(models.Model):
         self.env['bus.bus'].sendone((self._cr.dbname, 'res.partner', self.env.user.partner_id.id), notification)
 
     @api.multi
-    def set_message_starred(self, starred):
-        """ Set messages as (un)starred. Technically, the notifications related
+    def toggle_message_starred(self):
+        """ Toggle messages as (un)starred. Technically, the notifications related
             to uid are set to (un)starred.
-
-            :param bool starred: set notification as (un)starred
         """
         # a user should always be able to star a message he can read
         self.check_access_rule('read')
+        starred = not self.starred
         if starred:
             self.sudo().write({'starred_partner_ids': [(4, self.env.user.partner_id.id)]})
         else:
diff --git a/addons/mail/models/res_partner.py b/addons/mail/models/res_partner.py
index 5e3f4685282..46012ce2447 100644
--- a/addons/mail/models/res_partner.py
+++ b/addons/mail/models/res_partner.py
@@ -228,6 +228,18 @@ class Partner(models.Model):
         return 0
 
     @api.model
+    def get_starred_count(self):
+        """ compute the number of starred of the current user """
+        if self.env.user.partner_id:
+            self.env.cr.execute("""
+                SELECT count(*) as starred_count
+                FROM mail_message_res_partner_starred_rel R
+                WHERE R.res_partner_id = %s """, (self.env.user.partner_id.id,))
+            return self.env.cr.dictfetchall()[0].get('starred_count')
+        _logger.error('Call to starred_count without partner_id')
+        return 0
+
+    @api.model
     def get_static_mention_suggestions(self):
         """ To be overwritten to return the id, name and email of partners used as static mention
             suggestions loaded once at webclient initialization and stored client side. """