diff options
Diffstat (limited to 'extensions/web/twitter/account.py')
| -rw-r--r-- | extensions/web/twitter/account.py | 283 |
1 files changed, 283 insertions, 0 deletions
diff --git a/extensions/web/twitter/account.py b/extensions/web/twitter/account.py new file mode 100644 index 0000000..a2e4bf0 --- /dev/null +++ b/extensions/web/twitter/account.py @@ -0,0 +1,283 @@ +#!/usr/bin/env python +# +# Copyright (c) 2013 Walter Bender, Raul Gutierrez Segales, Martin Abente + +#Permission is hereby granted, free of charge, to any person obtaining a copy +#of this software and associated documentation files (the "Software"), to deal +#in the Software without restriction, including without limitation the rights +#to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +#copies of the Software, and to permit persons to whom the Software is +#furnished to do so, subject to the following conditions: + +#The above copyright notice and this permission notice shall be included in +#all copies or substantial portions of the Software. + +#THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +#IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +#FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +#AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +#LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +#OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +#THE SOFTWARE. + +from gettext import gettext as _ +import logging +import os +import tempfile +import time +import json + +from gi.repository import Gtk +from gi.repository import GdkPixbuf +from gi.repository import GConf +from gi.repository import GObject + +from sugar3.datastore import datastore +from sugar3.graphics.alert import NotifyAlert +from sugar3.graphics.icon import Icon + +from jarabe.journal import journalwindow +from jarabe.web import account + +from twitter.twr_account import TwrAccount +from twitter.twr_status import TwrStatus +from twitter.twr_timeline import TwrTimeline + +ACCOUNT_NEEDS_ATTENTION = 0 +ACCOUNT_ACTIVE = 1 +ACCOUNT_NAME = _('Twitter') +COMMENTS = 'comments' +COMMENT_IDS = 'twr_comment_ids' +COMMENT_LAST_ID = 'last_comment_id' + + +class TwitterAccount(account.Account): + + CONSUMER_TOKEN_KEY = "/desktop/sugar/collaboration/twitter_consumer_token" + CONSUMER_SECRET_KEY = "/desktop/sugar/collaboration/twitter_consumer_secret" + ACCESS_TOKEN_KEY = "/desktop/sugar/collaboration/twitter_access_token" + ACCESS_SECRET_KEY = "/desktop/sugar/collaboration/twitter_access_secret" + + def __init__(self): + self._client = GConf.Client.get_default() + ctoken, csecret, atoken, asecret = self._access_tokens() + TwrAccount.set_secrets(ctoken, csecret, atoken, asecret) + self._alert = None + + def get_description(self): + return ACCOUNT_NAME + + def is_configured(self): + return None not in self._access_tokens() + + def is_active(self): + # No expiration date + return None not in self._access_tokens() + + def get_share_menu(self, journal_entry_metadata): + twr_share_menu = _TwitterShareMenu(journal_entry_metadata, + self.is_active()) + self._connect_transfer_signals(twr_share_menu) + return twr_share_menu + + def get_refresh_menu(self): + twr_refresh_menu = _TwitterRefreshMenu(self.is_active()) + self._connect_transfer_signals(twr_refresh_menu) + return twr_refresh_menu + + def _connect_transfer_signals(self, transfer_widget): + transfer_widget.connect('transfer-state-changed', + self._transfer_state_changed_cb) + + def _transfer_state_changed_cb(self, widget, state_message): + logging.debug('_transfer_state_changed_cb') + + # First, remove any existing alert + if self._alert is None: + self._alert = NotifyAlert() + self._alert.props.title = ACCOUNT_NAME + self._alert.connect('response', self._alert_response_cb) + journalwindow.get_journal_window().add_alert(self._alert) + self._alert.show() + + logging.debug(state_message) + self._alert.props.msg = state_message + + def _alert_response_cb(self, alert, response_id): + journalwindow.get_journal_window().remove_alert(alert) + self._alert = None + + def _access_tokens(self): + return (self._client.get_string(self.CONSUMER_TOKEN_KEY), + self._client.get_string(self.CONSUMER_SECRET_KEY), + self._client.get_string(self.ACCESS_TOKEN_KEY), + self._client.get_string(self.ACCESS_SECRET_KEY)) + + +class _TwitterShareMenu(account.MenuItem): + __gtype_name__ = 'JournalTwitterMenu' + + def __init__(self, metadata, is_active): + account.MenuItem.__init__(self, ACCOUNT_NAME) + + if is_active: + icon_name = 'twitter-share' + else: + icon_name = 'twitter-share-insensitive' + self.set_image(Icon(icon_name=icon_name, + icon_size=Gtk.IconSize.MENU)) + self.show() + self._metadata = metadata + self._comment = '%s: %s' % (self._get_metadata_by_key('title'), + self._get_metadata_by_key('description')) + + self.connect('activate', self._twitter_share_menu_cb) + + def _get_metadata_by_key(self, key, default_value=''): + if key in self._metadata: + return self._metadata[key] + return default_value + + def _status_updated_cb(self, status, data, tmp_file): + if os.path.exists(tmp_file): + os.unlink(tmp_file) + try: + ds_object = datastore.get(self._metadata['uid']) + ds_object.metadata['twr_object_id'] = status._status_id + datastore.write(ds_object, update_mtime=False) + except Exception as e: + logging.debug('_status_updated_cb failed to write: %s', str(e)) + + def _status_updated_failed_cb(self, status, message, tmp_file): + if os.path.exists(tmp_file): + os.unlink(tmp_file) + logging.error('_status_updated_failed_cb %s', message) + + def _twitter_share_menu_cb(self, menu_item): + logging.debug('_twitter_share_menu_cb') + + self.emit('transfer-state-changed', _('Download started')) + tmp_file = tempfile.mktemp() + self._image_file_from_metadata(tmp_file) + + status = TwrStatus() + status.connect('status-updated', self._status_updated_cb, tmp_file) + status.connect('status-updated-failed', + self._status_updated_failed_cb, tmp_file) + status.update_with_media(self._comment, tmp_file) + + def _image_file_from_metadata(self, image_path): + """ Load a pixbuf from a Journal object. """ + pixbufloader = \ + GdkPixbuf.PixbufLoader.new_with_mime_type('image/png') + pixbufloader.set_size(300, 225) + try: + pixbufloader.write(self._metadata['preview']) + pixbuf = pixbufloader.get_pixbuf() + except Exception as ex: + logging.debug("_image_file_from_metadata: %s" % (str(ex))) + pixbuf = None + + pixbufloader.close() + if pixbuf: + pixbuf.savev(image_path, 'png', [], []) + + +class _TwitterRefreshMenu(account.MenuItem): + def __init__(self, is_active): + account.MenuItem.__init__(self, ACCOUNT_NAME) + + self._is_active = is_active + self._metadata = None + + if is_active: + icon_name = 'twitter-refresh' + else: + icon_name = 'twitter-refresh-insensitive' + self.set_image(Icon(icon_name=icon_name, + icon_size=Gtk.IconSize.MENU)) + self.show() + + self.connect('activate', self._twr_refresh_menu_clicked_cb) + + def set_metadata(self, metadata): + self._metadata = metadata + if self._is_active: + if self._metadata: + if 'fb_object_id' in self._metadata: + self.set_sensitive(True) + icon_name = 'twitter-refresh' + else: + self.set_sensitive(False) + icon_name = 'twitter-refresh-insensitive' + self.set_image(Icon(icon_name=icon_name, + icon_size=Gtk.IconSize.MENU)) + + def _twr_refresh_menu_clicked_cb(self, button): + logging.debug('_twr_refresh_menu_clicked_cb') + + if self._metadata is None: + logging.debug('_twr_refresh_menu_clicked_cb called without metadata') + return + + if 'twr_object_id' not in self._metadata: + logging.debug('_twr_refresh_menu_clicked_cb called without twr_object_id in metadata') + return + + self.emit('transfer-state-changed', _('Download started')) + + status_id = self._metadata['twr_object_id'] + + # XXX is there other way to update metadata? + ds_object = datastore.get(self._metadata['uid']) + if COMMENT_LAST_ID in ds_object.metadata: + status_id = ds_object.metadata[COMMENT_LAST_ID] + + timeline = TwrTimeline() + timeline.connect('mentions-downloaded', self._twr_mentions_downloaded_cb) + timeline.mentions_timeline(since_id=status_id) + + def _twr_mentions_downloaded_cb(self, timeline, comments): + logging.debug('_twr_mentions_downloaded_cb') + + ds_object = datastore.get(self._metadata['uid']) + if not COMMENTS in ds_object.metadata: + ds_comments = [] + else: + ds_comments = json.loads(ds_object.metadata[COMMENTS]) + if not COMMENT_IDS in ds_object.metadata: + ds_comment_ids = [] + else: + ds_comment_ids = json.loads(ds_object.metadata[COMMENT_IDS]) + + comment_id = None + new_comment = False + for comment in comments: + # XXX hope for a better API + if comment['in_reply_to_status_id_str'] != self._metadata['twr_object_id']: + continue + + if comment['id_str'] not in ds_comment_ids: + ds_comments.append({'from': comment['user']['name'], + 'message': comment['text'], + 'icon': 'twitter-share'}) + ds_comment_ids.append(comment['id_str']) + + # XXX keep the latest one + if comment['id_str'] > comment_id: + comment_id = comment['id_str'] + new_comment = True + + if new_comment: + ds_object.metadata[COMMENT_LAST_ID] = comment_id + ds_object.metadata[COMMENTS] = json.dumps(ds_comments) + ds_object.metadata[COMMENT_IDS] = json.dumps(ds_comment_ids) + self.emit('comments-changed', ds_object.metadata[COMMENTS]) + + datastore.write(ds_object, update_mtime=False) + + def _twr_comments_download_failed_cb(self, tweet, failed_reason): + logging.debug('_twr_comments_download_failed_cb: %s' % (failed_reason)) + +def get_account(): + return TwitterAccount() |