diff options
author | Walter Bender <walter.bender@gmail.com> | 2013-04-03 03:50:36 (GMT) |
---|---|---|
committer | Walter Bender <walter.bender@gmail.com> | 2013-04-03 03:50:36 (GMT) |
commit | d0110fcdeb9c0c5d83e3730215410ba720e21702 (patch) | |
tree | e9a42bbaddb25fa588a47bb9bf2fe193e661a12f /extensions/web | |
parent | 567769dc3139caf39ab1d4cac057f77255d1d2e3 (diff) |
cleanup based on feedback from dnarvaez
Diffstat (limited to 'extensions/web')
-rw-r--r-- | extensions/web/facebook/Makefile.am | 2 | ||||
-rw-r--r-- | extensions/web/facebook/facebookaccount.py | 281 | ||||
-rw-r--r-- | extensions/web/twitter/Makefile.am | 2 | ||||
-rw-r--r-- | extensions/web/twitter/twitteraccount.py | 283 |
4 files changed, 566 insertions, 2 deletions
diff --git a/extensions/web/facebook/Makefile.am b/extensions/web/facebook/Makefile.am index e2e6d1f..f9bac63 100644 --- a/extensions/web/facebook/Makefile.am +++ b/extensions/web/facebook/Makefile.am @@ -3,4 +3,4 @@ SUBDIRS = facebook icons sugardir = $(pkgdatadir)/extensions/web/facebook sugar_PYTHON = \ __init__.py \ - facebook_online_account.py + facebookaccount.py diff --git a/extensions/web/facebook/facebookaccount.py b/extensions/web/facebook/facebookaccount.py new file mode 100644 index 0000000..f105fdb --- /dev/null +++ b/extensions/web/facebook/facebookaccount.py @@ -0,0 +1,281 @@ +#!/usr/bin/env python +# +# Copyright (c) 2013 Walter Bender, Raul Gutierrez Segales + +#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 facebook import facebook + +ACCOUNT_NEEDS_ATTENTION = 0 +ACCOUNT_ACTIVE = 1 +ACCOUNT_NAME = _('Facebook') +COMMENTS = 'comments' +COMMENT_IDS = 'fb_comment_ids' + +class FacebookAccount(account.Account): + + ACCESS_TOKEN_KEY = "/desktop/sugar/collaboration/facebook_access_token" + ACCESS_TOKEN_KEY_EXPIRATION_DATE = \ + "/desktop/sugar/collaboration/facebook_access_token_expiration_date" + + def __init__(self): + self._client = GConf.Client.get_default() + facebook.FbAccount.set_access_token(self._access_token()) + self._alert = None + + def get_description(self): + return ACCOUNT_NAME + + def is_configured(self): + return self._access_token() is not None + + def is_active(self): + expiration_date = \ + self._client.get_int(self.ACCESS_TOKEN_KEY_EXPIRATION_DATE) + return expiration_date != 0 and expiration_date > time.time() + + def get_share_menu(self, journal_entry_metadata): + fb_share_menu = _FacebookShareMenu(journal_entry_metadata, + self.is_active()) + self._connect_transfer_signals(fb_share_menu) + return fb_share_menu + + def get_refresh_menu(self): + fb_refresh_menu = _FacebookRefreshMenu(self.is_active()) + self._connect_transfer_signals(fb_refresh_menu) + return fb_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: + logging.debug('creating new alert') + 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_token(self): + return self._client.get_string(self.ACCESS_TOKEN_KEY) + + +class _FacebookShareMenu(account.MenuItem): + __gtype_name__ = 'JournalFacebookMenu' + + def __init__(self, metadata, is_active): + account.MenuItem.__init__(self, ACCOUNT_NAME) + + if is_active: + icon_name = 'facebook-share' + else: + icon_name = 'facebook-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._facebook_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 _facebook_share_menu_cb(self, menu_item): + logging.debug('_facebook_share_menu_cb') + + self.emit('transfer-state-changed', _('Upload started')) + tmp_file = tempfile.mktemp() + self._image_file_from_metadata(tmp_file) + + photo = facebook.FbPhoto() + photo.connect('photo-created', self._photo_created_cb, tmp_file) + photo.connect('photo-create-failed', + self._photo_create_failed_cb, + tmp_file) + photo.connect('transfer-state-changed', + self._transfer_state_changed_cb) + + GObject.idle_add(photo.create, tmp_file) + + def _photo_created_cb(self, fb_photo, fb_object_id, tmp_file): + logging.debug("_photo_created_cb") + + if os.path.exists(tmp_file): + os.unlink(tmp_file) + + fb_photo.connect('comment-added', self._comment_added_cb) + fb_photo.connect('comment-add-failed', self._comment_add_failed_cb) + fb_photo.add_comment(self._comment) + + try: + ds_object = datastore.get(self._metadata['uid']) + ds_object.metadata['fb_object_id'] = fb_object_id + datastore.write(ds_object, update_mtime=False) + except Exception as ex: + logging.debug("_photo_created_cb failed to write to datastore: " % \ + str(ex)) + + def _photo_create_failed_cb(self, fb_photo, failed_reason, tmp_file): + logging.debug("_photo_create_failed_cb") + + if os.path.exists(tmp_file): + os.unlink(tmp_file) + + def _comment_added_cb(self, fb_photo, fb_comment_id): + logging.debug("_comment_added_cb") + + def _comment_add_failed_cb(self, fb_photo, failed_reason): + logging.debug("_comment_add_failed_cb") + + 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 _FacebookRefreshMenu(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 = 'facebook-refresh' + else: + icon_name = 'facebook-refresh-insensitive' + self.set_image(Icon(icon_name=icon_name, + icon_size=Gtk.IconSize.MENU)) + self.show() + + self.connect('activate', self._fb_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 = 'facebook-refresh' + else: + self.set_sensitive(False) + icon_name = 'facebook-refresh-insensitive' + self.set_image(Icon(icon_name=icon_name, + icon_size=Gtk.IconSize.MENU)) + + def _fb_refresh_menu_clicked_cb(self, button): + logging.debug('_fb_refresh_menu_clicked_cb') + + if self._metadata is None: + logging.debug('_fb_refresh_menu_clicked_cb called without metadata') + return + + if 'fb_object_id' not in self._metadata: + logging.debug('_fb_refresh_menu_clicked_cb called without fb_object_id in metadata') + return + + self.emit('transfer-state-changed', _('Download started')) + fb_photo = facebook.FbPhoto(self._metadata['fb_object_id']) + fb_photo.connect('comments-downloaded', + self._fb_comments_downloaded_cb) + fb_photo.connect('comments-download-failed', + self._fb_comments_download_failed_cb) + fb_photo.connect('transfer-state-changed', + self._transfer_state_changed_cb) + GObject.idle_add(fb_photo.refresh_comments) + + def _fb_comments_downloaded_cb(self, fb_photo, comments): + logging.debug('_fb_comments_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]) + new_comment = False + for comment in comments: + if comment['id'] not in ds_comment_ids: + # TODO: get avatar icon and add it to icon_theme + ds_comments.append({'from': comment['from'], + 'message': comment['message'], + 'icon': 'facebook-share'}) + ds_comment_ids.append(comment['id']) + new_comment = True + if new_comment: + 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 _fb_comments_download_failed_cb(self, fb_photo, failed_reason): + logging.debug('_fb_comments_download_failed_cb: %s' % (failed_reason)) + +def get_account(): + return FacebookAccount() diff --git a/extensions/web/twitter/Makefile.am b/extensions/web/twitter/Makefile.am index 8346b41..438d3fa 100644 --- a/extensions/web/twitter/Makefile.am +++ b/extensions/web/twitter/Makefile.am @@ -3,4 +3,4 @@ SUBDIRS = twitter icons sugardir = $(pkgdatadir)/extensions/web/twitter sugar_PYTHON = \ __init__.py \ - twitter_online_account.py + twitteraccount.py diff --git a/extensions/web/twitter/twitteraccount.py b/extensions/web/twitter/twitteraccount.py new file mode 100644 index 0000000..a2e4bf0 --- /dev/null +++ b/extensions/web/twitter/twitteraccount.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() |