global s/frontends/applications/

weboob/applications/qhavesex/contacts.py

@@ -0,0 +1,203 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2010  Romain Bignon
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, version 3 of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+
+import urllib2
+import time
+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, HTMLDelegate
+from weboob.capabilities.contact import ICapContact, Contact
+from weboob.capabilities.chat import ICapChat
+from weboob.capabilities.messages import ICapMessages
+
+from .ui.contacts_ui import Ui_Contacts
+from .ui.contact_thread_ui import Ui_ContactThread
+from .ui.thread_message_ui import Ui_ThreadMessage
+
+class ThreadMessage(QFrame):
+    def __init__(self, message, parent=None):
+        QFrame.__init__(self, parent)
+        self.ui = Ui_ThreadMessage()
+        self.ui.setupUi(self)
+
+        self.date = message.get_date()
+
+        self.ui.nameLabel.setText(message.sender)
+        self.ui.headerLabel.setText(time.strftime('%Y-%m-%d %H:%M:%S', message.get_date().timetuple()))
+        if message.is_html:
+            content = message.content
+        else:
+            content = message.content.replace('&', '&amp;').replace('<', '&lt;').replace('>', '&gt;').replace('\n', '<br />')
+        self.ui.contentLabel.setText(content)
+
+class ContactThread(QWidget):
+    def __init__(self, weboob, contact, parent=None):
+        QWidget.__init__(self, parent)
+        self.ui = Ui_ContactThread()
+        self.ui.setupUi(self)
+
+        self.weboob = weboob
+        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_backends(self.contact.backend, command, thread=self.contact.id)
+
+    def gotMessage(self, backend, message):
+        if not message:
+            v = self.ui.scrollArea.verticalScrollBar()
+            print v.minimum(), v.value(), v.maximum(), v.sliderPosition()
+            self.ui.scrollArea.verticalScrollBar().setValue(self.ui.scrollArea.verticalScrollBar().maximum())
+            return
+
+        widget = ThreadMessage(message)
+        for i, m in enumerate(self.messages):
+            if widget.date < m.date:
+                self.ui.scrollAreaContent.layout().insertWidget(i, widget)
+                self.messages.insert(i, widget)
+                return
+
+        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
+        self.name = name
+        self.weboob = weboob
+
+    def iter_contacts(self, cb):
+        raise NotImplementedError()
+
+class MetaGroup(IGroup):
+    def iter_contacts(self, cb):
+        if self.id == 'online':
+            status = Contact.STATUS_ONLINE
+        elif self.id == 'offline':
+            status = Contact.STATUS_OFFLINE
+        else:
+            status = Contact.STATUS_ALL
+
+        self.process = QtDo(self.weboob, lambda b, d: self.cb(cb, b, d))
+        self.process.do_caps(ICapContact, 'iter_contacts', status)
+
+    def cb(self, cb, backend, contact):
+        if contact:
+            contact.backend = backend
+        cb(contact)
+
+class ContactsWidget(QWidget):
+    def __init__(self, weboob, parent=None):
+        QWidget.__init__(self, parent)
+        self.ui = Ui_Contacts()
+        self.ui.setupUi(self)
+
+        self.weboob = weboob
+        self.contact = None
+        self.ui.contactList.setItemDelegate(HTMLDelegate())
+
+        self.ui.groupBox.addItem('All', MetaGroup(self.weboob, 'all', self.tr('All')))
+        self.ui.groupBox.addItem('Onlines', MetaGroup(self.weboob, 'online', self.tr('Online')))
+        self.ui.groupBox.addItem('Offlines', MetaGroup(self.weboob, 'offline', self.tr('Offline')))
+
+        self.connect(self.ui.groupBox, SIGNAL('currentIndexChanged(int)'), self.groupChanged)
+        self.connect(self.ui.contactList, SIGNAL('currentItemChanged(QListWidgetItem*, QListWidgetItem*)'), self.contactChanged)
+        self.connect(self.ui.refreshButton, SIGNAL('clicked()'), self.refreshContactList)
+
+    def load(self):
+        self.ui.groupBox.setCurrentIndex(1)
+
+    def groupChanged(self, i):
+        self.refreshContactList()
+
+    def refreshContactList(self):
+        self.ui.contactList.clear()
+        i = self.ui.groupBox.currentIndex()
+        group = self.ui.groupBox.itemData(i).toPyObject()
+        group.iter_contacts(self.addContact)
+
+    def addContact(self, contact):
+        if not contact:
+            return
+
+        data = urllib2.urlopen(contact.thumbnail_url).read()
+        img = QImage.fromData(data)
+
+        status = ''
+        if contact.status == Contact.STATUS_ONLINE:
+            status = 'Online'
+            status_color = 0x00aa00
+        elif contact.status == Contact.STATUS_OFFLINE:
+            status = 'Offline'
+            status_color = 0xff0000
+
+        item = QListWidgetItem()
+        item.setText('<h2>%s</h2><font color="#%06X">%s</font><br /><i>%s</i>' % (contact.name, status_color, status, contact.backend.name))
+        item.setIcon(QIcon(QPixmap.fromImage(img)))
+        item.setData(Qt.UserRole, contact)
+
+        self.ui.contactList.addItem(item)
+
+    def contactChanged(self, current, previous):
+        self.ui.tabWidget.clear()
+        self.contact = None
+
+        if not current:
+            return
+
+        self.contact = current.data(Qt.UserRole).toPyObject()
+
+        self.ui.tabWidget.addTab(QWidget(), self.tr('Profile'))
+        if self.contact.backend.has_caps(ICapMessages):
+            self.ui.tabWidget.addTab(ContactThread(self.weboob, self.contact), self.tr('Messages'))
+        if self.contact.backend.has_caps(ICapChat):
+            self.ui.tabWidget.addTab(QWidget(), self.tr('Chat'))
+        self.ui.tabWidget.addTab(QWidget(), self.tr('Calendar'))
+        self.ui.tabWidget.addTab(QWidget(), self.tr('Notes'))