From 912d3496c973c5f2f1e678f59bba8dd510630d89 Mon Sep 17 00:00:00 2001
From: Romain Bignon <romain@peerfuse.org>
Date: Sun, 28 Nov 2010 11:34:12 +0100
Subject: [PATCH] add flags IS_THREADS and IS_DISCUSSION to Thread to know how
 to display messages

---
 weboob/backends/aum/backend.py  | 2 ++
 weboob/capabilities/messages.py | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/weboob/backends/aum/backend.py b/weboob/backends/aum/backend.py
index 3054e228..0fd08a76 100644
--- a/weboob/backends/aum/backend.py
+++ b/weboob/backends/aum/backend.py
@@ -107,6 +107,7 @@ class AuMBackend(BaseBackend, ICapMessages, ICapMessagesPost, ICapDating, ICapCh
                 self.report_spam(contact.get_id(), contact.get_suppr_id())
                 continue
             thread = Thread(contact.get_id())
+            thread.flags = Thread.IS_DISCUSSION
             thread.title = 'Discussion with %s' % contact.get_name()
             yield thread
 
@@ -123,6 +124,7 @@ class AuMBackend(BaseBackend, ICapMessages, ICapMessagesPost, ICapDating, ICapCh
 
         if not thread:
             thread = Thread(id)
+            thread.flags = Thread.IS_DISCUSSION
             full = False
         else:
             full = True
diff --git a/weboob/capabilities/messages.py b/weboob/capabilities/messages.py
index 3665540a..616a2c44 100644
--- a/weboob/capabilities/messages.py
+++ b/weboob/capabilities/messages.py
@@ -95,6 +95,9 @@ class Message(CapBaseObject):
         return result.encode('utf-8')
 
 class Thread(CapBaseObject):
+    IS_THREADS =    0x001
+    IS_DISCUSSION = 0x002
+
     def __init__(self, id):
         CapBaseObject.__init__(self, id)
         self.add_field('root', Message)
@@ -102,6 +105,7 @@ class Thread(CapBaseObject):
         self.add_field('date', datetime.datetime)
         self.add_field('nb_messages', int)
         self.add_field('nb_unread', int)
+        self.add_field('flags', int, self.IS_THREADS)
 
     def iter_all_messages(self):
         if self.root: