From 7233fca84db2da56167b096e1add2ed3ddc1b1d0 Mon Sep 17 00:00:00 2001
From: Romain Bignon <romain@peerfuse.org>
Date: Sun, 4 Jul 2010 17:23:20 +0200
Subject: [PATCH] ability to answer to messages

---
 weboob/frontends/qhavesex/contacts.py | 37 ++++++++++++++++++++++++---
 weboob/ouiboube.py                    |  3 +++
 2 files changed, 37 insertions(+), 3 deletions(-)

diff --git a/weboob/frontends/qhavesex/contacts.py b/weboob/frontends/qhavesex/contacts.py
index 6d2c0d37..6f9a9279 100644
--- a/weboob/frontends/qhavesex/contacts.py
+++ b/weboob/frontends/qhavesex/contacts.py
@@ -17,7 +17,8 @@
 
 import urllib2
 import time
-from PyQt4.QtGui import QWidget, QListWidgetItem, QImage, QIcon, QPixmap, QFrame
+import logging
+from PyQt4.QtGui import QWidget, QListWidgetItem, QImage, QIcon, QPixmap, QFrame, QMessageBox
 from PyQt4.QtCore import SIGNAL, Qt
 
 from weboob.tools.application.qt import QtDo
@@ -55,9 +56,18 @@ class ContactThread(QWidget):
         self.contact = contact
         self.messages = []
 
+        self.connect(self.ui.sendButton, SIGNAL('clicked()'), self.postReply)
+
+        self.refreshMessages()
+
+    def refreshMessages(self):
+        if self.ui.scrollAreaContent.layout().count() > 0:
+            command = 'iter_new_messages'
+        else:
+            command = 'iter_messages'
+
         self.process = QtDo(self.weboob, self.gotMessage)
-        self.process.do('iter_messages', thread=self.contact.id)
-        print self.contact.id
+        self.process.do_backends(self.contact.backend, command, thread=self.contact.id)
 
     def gotMessage(self, backend, message):
         if not message:
@@ -76,6 +86,27 @@ class ContactThread(QWidget):
         self.ui.scrollAreaContent.layout().addWidget(widget)
         self.messages.append(widget)
 
+    def postReply(self):
+        text = unicode(self.ui.textEdit.toPlainText())
+        self.ui.textEdit.setEnabled(False)
+        self.ui.sendButton.setEnabled(False)
+        self.process = QtDo(self.weboob, self.replyPosted, self.replyNotPosted)
+        self.process.do_backends(self.contact.backend, 'post_reply', self.contact.id, 0, '', text)
+
+    def replyPosted(self, backend, ignored):
+        self.ui.textEdit.clear()
+        self.ui.textEdit.setEnabled(True)
+        self.ui.sendButton.setEnabled(True)
+        self.refreshMessages()
+
+    def replyNotPosted(self, backend, error, backtrace):
+        content = unicode(self.tr('Unable to send message:\n%s\n')) % error
+        if logging.root.level == logging.DEBUG:
+            content += '\n%s\n' % backtrace
+        QMessageBox.critical(self, self.tr('Error while posting reply'),
+                             content, QMessageBox.Ok)
+
+
 class IGroup(object):
     def __init__(self, weboob, id, name):
         self.id = id
diff --git a/weboob/ouiboube.py b/weboob/ouiboube.py
index 962b881c..eb13b9a2 100644
--- a/weboob/ouiboube.py
+++ b/weboob/ouiboube.py
@@ -24,6 +24,7 @@ import sys
 
 from weboob.bcall import BackendsCall, CallErrors
 from weboob.modules import ModulesLoader, BackendsConfig
+from weboob.backend import BaseBackend
 from weboob.scheduler import Scheduler
 
 if sys.version_info[:2] <= (2, 5):
@@ -147,6 +148,8 @@ class Weboob(object):
     def do_backends(self, backends, function, *args, **kwargs):
         if backends is None:
             backends = list(self.iter_backends())
+        elif isinstance(backends, BaseBackend):
+            backends = [backends]
         elif isinstance(backends, (str,unicode)):
             backends = [backend for backend in self.iter_backends() if backend.name == backends]
         elif isinstance(backends, (list,tuple)):