[Twitter] Add new twitter module

modules/twitter/backend.py

@@ -0,0 +1,99 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2014      Bezleputh
+#
+# This file is part of weboob.
+#
+# weboob is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# weboob 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 Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with weboob. If not, see <http://www.gnu.org/licenses/>.
+
+from datetime import time, datetime, timedelta
+
+from weboob.tools.value import Value, ValueBackendPassword
+from weboob.tools.backend import BaseBackend, BackendConfig
+from weboob.capabilities.messages import ICapMessages, Thread, ICapMessagesPost
+from weboob.capabilities.base import find_object
+from weboob.tools.exceptions import BrowserForbidden
+from .browser import TwitterBrowser
+
+
+__all__ = ['TwitterBackend']
+
+
+class TwitterBackend(BaseBackend, ICapMessages, ICapMessagesPost):
+    NAME = 'twitter'
+    DESCRIPTION = u'twitter website'
+    MAINTAINER = u'Bezleputh'
+    EMAIL = 'carton_ben@yahoo.fr'
+    LICENSE = 'AGPLv3+'
+    VERSION = '0.j'
+    BROWSER = TwitterBrowser
+    STORAGE = {'seen': {}}
+
+    CONFIG = BackendConfig(Value('username',                label='Username', default=''),
+                           ValueBackendPassword('password', label='Password', default=''))
+
+    def create_default_browser(self):
+        return self.create_browser(self.config['username'].get(), self.config['password'].get())
+
+    def iter_threads(self):
+        return self.browser.iter_threads()
+
+    def get_thread(self, _id, thread=None, getseen=True):
+        seen = None
+        if getseen:
+            seen = self.storage.get('seen', default={})
+        return self.browser.get_thread(_id, thread, seen)
+
+    def fill_thread(self, thread, fields, getseen=True):
+        return self.get_thread(thread.id, thread, getseen)
+
+    def set_message_read(self, message):
+        self.storage.set('seen', message.thread.id, 'comments',
+                         self.storage.get('seen',
+                                          message.thread.id,
+                                          'comments', default=[]) + [message.id])
+        self.storage.save()
+        self._purge_message_read()
+
+    def _purge_message_read(self):
+        lastpurge = self.storage.get('lastpurge', default=0)
+
+        if time.time() - lastpurge > 86400:
+            self.storage.set('lastpurge', time.time())
+            self.storage.save()
+
+            # we can't directly delete without a "RuntimeError: dictionary changed size during iteration"
+            todelete = []
+
+            for id in self.storage.get('seen', default={}):
+                date = self.storage.get('date', id, default=0)
+                # if no date available, create a new one (compatibility with "old" storage)
+                if date == 0:
+                    self.storage.set('date', id, datetime.now())
+                elif datetime.now() - date > timedelta(days=60):
+                    todelete.append(id)
+
+            for id in todelete:
+                self.storage.delete('hash', id)
+                self.storage.delete('date', id)
+                self.storage.delete('seen', id)
+            self.storage.save()
+
+    def post_message(self, message):
+        if not self.browser.username:
+            raise BrowserForbidden()
+        self.browser.post(find_object(self.iter_threads(), id=message.full_id.split('.')[0]),
+                          message.content)
+
+    OBJECTS = {Thread: fill_thread}