Julio Biason
14 years ago
2 changed files with 401 additions and 354 deletions
@ -0,0 +1,364 @@
|
||||
#!/usr/bin/env python |
||||
# -*- coding: utf-8 -*- |
||||
|
||||
# Mitter, a micro-blogging client with multiple interfaces. |
||||
# Copyright (C) 2007-2010 Mitter Contributors |
||||
# |
||||
# 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, either version 3 of the License, or |
||||
# (at your option) any later version. |
||||
# |
||||
# 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, see <http://www.gnu.org/licenses/>. |
||||
|
||||
import gobject |
||||
import gtk |
||||
import logging |
||||
import pango |
||||
|
||||
from cgi import escape as html_escape |
||||
|
||||
# ---------------------------------------------------------------------- |
||||
# String with the format to the message |
||||
# ---------------------------------------------------------------------- |
||||
MESSAGE_FORMAT = ('%(favourite_star)s' |
||||
'<b>%(full_name)s</b> ' |
||||
'<small>(%(username)s' |
||||
'%(message_type)s' |
||||
')</small>:' |
||||
'%(read_status)s ' |
||||
'%(protected_status)s ' |
||||
'\n' |
||||
'%(message)s\n' |
||||
'<small>%(message_age)s</small>') |
||||
|
||||
# ---------------------------------------------------------------------- |
||||
# Logging |
||||
# ---------------------------------------------------------------------- |
||||
_log = logging.getLogger('ui.pygtk.messagegrid') |
||||
|
||||
# ---------------------------------------------------------------------- |
||||
# MessageGrid class |
||||
# ---------------------------------------------------------------------- |
||||
class MessageGrid(gtk.ScrolledWindow, gobject.GObject): |
||||
"""Custom message grid.""" |
||||
|
||||
__gsignals__ = { |
||||
"count-changed": ( |
||||
gobject.SIGNAL_RUN_LAST, |
||||
gobject.TYPE_NONE, |
||||
(gobject.TYPE_INT, ))} # The exception |
||||
|
||||
@property |
||||
def count(self): |
||||
"""Number of unread messages.""" |
||||
return self._message_count |
||||
|
||||
@count.setter |
||||
def count(self, messages): |
||||
"""Update the number of unread messages.""" |
||||
self._message_count = messages |
||||
self.emit('count-changed', messages) |
||||
return |
||||
|
||||
def __init__(self): |
||||
super(MessageGrid, self).__init__() |
||||
|
||||
self.avatars = None |
||||
self.link_color = '#0000ff' |
||||
self.unread_char = '(o)' |
||||
self.favorite_char = '(F)' |
||||
self.unfavorite_char = '( )' |
||||
self.protected_char = '(protected)' |
||||
|
||||
self._message_count = 0 |
||||
|
||||
store = gtk.ListStore(object) |
||||
store.set_sort_column_id(0, gtk.SORT_ASCENDING) |
||||
store.set_sort_func(0, self._order_datetime) |
||||
|
||||
self._grid = gtk.TreeView(store) |
||||
self._grid.set_property('headers-visible', False) |
||||
self._grid.set_rules_hint(True) # change color for each row |
||||
|
||||
user_renderer = gtk.CellRendererPixbuf() |
||||
user_column = gtk.TreeViewColumn('User', user_renderer) |
||||
user_column.set_fixed_width(48) # avatar size (we resize to 48x48) |
||||
user_column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED) |
||||
user_column.set_cell_data_func(user_renderer, |
||||
self._cell_renderer_user) |
||||
|
||||
message_renderer = gtk.CellRendererText() |
||||
message_renderer.set_property('wrap-mode', pango.WRAP_WORD) |
||||
message_renderer.set_property('wrap-width', 200) |
||||
message_column = gtk.TreeViewColumn('Message', |
||||
message_renderer) |
||||
message_column.set_cell_data_func(message_renderer, |
||||
self._cell_renderer_message) |
||||
|
||||
self._grid.append_column(user_column) |
||||
self._grid.append_column(message_column) |
||||
|
||||
self._grid.set_resize_mode(gtk.RESIZE_IMMEDIATE) |
||||
self._grid.connect('cursor-changed', self._message_selected) |
||||
self._grid.connect('button-press-event', self._click_message) |
||||
self._grid.connect('popup-menu', self._message_popup) # Menu button |
||||
self._grid.connect('cursor-changed', self._mark_message_read) |
||||
|
||||
self.set_policy(gtk.POLICY_NEVER, gtk.POLICY_ALWAYS) |
||||
self.add(self._grid) |
||||
return |
||||
|
||||
#----------------------------------------------------------------------- |
||||
# Renderers |
||||
#----------------------------------------------------------------------- |
||||
def _cell_renderer_user(self, column, cell, store, position): |
||||
"""Callback for the user column. Used to created the pixbuf of the |
||||
userpic.""" |
||||
|
||||
data = store.get_value(position, 0) |
||||
pic = data.author.avatar |
||||
cell.set_property('pixbuf', self.avatars[pic]) |
||||
|
||||
return |
||||
|
||||
def _cell_renderer_message(self, column, cell, store, position): |
||||
"""Callback for the message column. We need this to adjust the markup |
||||
property of the cell, as setting it as text won't do any markup |
||||
processing.""" |
||||
|
||||
data = store.get_value(position, 0) |
||||
time = timesince.timesince(data.message_time) |
||||
|
||||
message_values = {} |
||||
|
||||
# unescape escaped entities that pango is not okay with |
||||
message_values['message'] = html_escape(data.message) |
||||
message_values['username'] = html_escape(data.author.username) |
||||
message_values['full_name'] = html_escape(data.author.name) |
||||
message_values['message_age'] = time |
||||
|
||||
# highlight URLs |
||||
mask = r'<span foreground="%s">\1</span>' % (self.link_color) |
||||
message_values['message'] = URL_RE.sub(mask, |
||||
message_values['message']) |
||||
|
||||
# use a different highlight for the current user |
||||
# TODO: How to handle this with several networks? |
||||
#message = re.sub(r'(@'+self.twitter.username+')', |
||||
# r'<span foreground="#FF6633">\1</span>', |
||||
# message) |
||||
|
||||
if not data.read: |
||||
message_values['read_status'] = (' ' + self.unread_char) |
||||
else: |
||||
message_values['read_status'] = '' |
||||
|
||||
if data.favourite: |
||||
message_values['favourite_star'] = (self.unfavourite_char) |
||||
else: |
||||
message_values['favourite_star'] = (self.favourite_char) |
||||
|
||||
info = [] |
||||
if data.reposted_by: |
||||
info.append(_(' — <i>reposted by %s</i>') % |
||||
(data.reposted_by.username)) |
||||
|
||||
if data.parent_owner: |
||||
info.append(_(' — <i>in reply to %s</i>') % |
||||
(data.parent_owner.username)) |
||||
|
||||
if data.protected: |
||||
message_values['protected_status'] = (self.protected_char) |
||||
else: |
||||
message_values['protected_status'] = '' |
||||
|
||||
message_values['message_type'] = ''.join(info) |
||||
markup = MESSAGE_FORMAT % (message_values) |
||||
|
||||
cell.set_property('markup', markup) |
||||
|
||||
return |
||||
|
||||
#----------------------------------------------------------------------- |
||||
# Widget callbacks |
||||
#----------------------------------------------------------------------- |
||||
def _message_selected(self, view, user_data=None): |
||||
"""Callback when a row in the list is selected. Mostly, we'll check |
||||
if the message is from the logged user and we change de "sensitive" |
||||
property of the message actions to reflect what the user can and |
||||
cannot do.""" |
||||
|
||||
(model, iter) = view.get_selection().get_selected() |
||||
if not iter: |
||||
return |
||||
data = model.get_value(iter, 0) |
||||
data.read = True |
||||
|
||||
self._delete_action.set_property('sensitive', data.deletable) |
||||
self._reply_action.set_property('sensitive', data.replyable) |
||||
self._repost_action.set_property('sensitive', data.repostable) |
||||
self._favourite_action.set_property('sensitive', data.favoritable) |
||||
|
||||
return 0 |
||||
|
||||
def _click_message(self, widget, event, user_data=None): |
||||
"""Check the click on the message and, if it's a right click, call the |
||||
_message_popup function to show the popup.""" |
||||
if event.button != 3: |
||||
# not right click |
||||
return False |
||||
|
||||
path = widget.get_path_at_pos(event.x, event.y) |
||||
if not path: |
||||
return False |
||||
|
||||
(path, _, _, _) = path |
||||
return self._url_popup(widget, path, event) |
||||
|
||||
def _message_popup(self, widget, user_data=None): |
||||
"""Builds the popup with the URLs in the message.""" |
||||
_log.debug('Popup') |
||||
(path, _) = widget.get_cursor() |
||||
if not path: |
||||
return True |
||||
|
||||
# create a syntetic event (the popup requires one) |
||||
event = gtk.gdk.Event(gtk.gdk.BUTTON_PRESS) |
||||
event.button = 3 |
||||
return self._url_popup(widget, path, event) |
||||
|
||||
def _url_popup(self, widget, path, event): |
||||
"""Builds the popup with URLs in the cell pointed by *path*. Requires |
||||
the *event* that the widget received.""" |
||||
iter = widget.get_model().get_iter(path) |
||||
message = widget.get_model().get_value(iter, 0) |
||||
|
||||
items = [] |
||||
if message.link: |
||||
items.append((_('Open on %s') % (message.network_name), |
||||
message.link)) |
||||
|
||||
urls = URL_RE.findall(message.message) |
||||
for url in urls: |
||||
if len(url) > 20: |
||||
title = url[:20] + '...' |
||||
else: |
||||
title = url |
||||
items.append((title, url)) |
||||
|
||||
if len(items) == 0: |
||||
return |
||||
|
||||
popup = gtk.Menu() |
||||
for url in items: |
||||
(title, url) = url |
||||
item = gtk.MenuItem(title) |
||||
item.connect('activate', self._open_url, url) |
||||
popup.append(item) |
||||
|
||||
popup.show_all() |
||||
popup.popup(None, None, None, event.button, event.time) |
||||
|
||||
return True |
||||
|
||||
def _mark_message_read(self, widget, user_data=None): |
||||
"""Mark a message as read when it's selected.""" |
||||
(path, _) = widget.get_cursor() |
||||
if not path: |
||||
return True |
||||
|
||||
iter = widget.get_model().get_iter(path) |
||||
message = widget.get_model().get_value(iter, 0) |
||||
|
||||
if message.read: |
||||
return True |
||||
|
||||
self.count -= 1 |
||||
return True |
||||
|
||||
def update(self, messages): |
||||
"""Update the grid with new messages.""" |
||||
self._grid.freeze() |
||||
store = self._grid.get_model() |
||||
for message in messages: |
||||
message.read = False |
||||
store.prepend([message]) |
||||
store.sort_column_changed() |
||||
self._grid.thaw() |
||||
return |
||||
|
||||
def _order_datetime(self, model, iter1, iter2, user_data=None): |
||||
"""Used by the ListStore to sort the columns (in our case, "column") |
||||
by date.""" |
||||
message1 = model.get_value(iter1, 0) |
||||
message2 = model.get_value(iter2, 0) |
||||
|
||||
if (not message1) or \ |
||||
(not message1.message_time) or \ |
||||
(message1.message_time > message2.message_time): |
||||
return -1 |
||||
|
||||
if (not message2) or \ |
||||
(not message2.message_time) or \ |
||||
(message2.message_time > message1.message_time): |
||||
return 1 |
||||
return 0 |
||||
|
||||
def clear_posts(self, widget, user_data=None): |
||||
"""Clear the list of posts from the grid.""" |
||||
self._grid.get_model().clear() |
||||
self.count = 0; |
||||
return |
||||
|
||||
def mark_all_read(self, widget, user_data=None): |
||||
"""Mark all messages as read.""" |
||||
model = self._grid.get_model() |
||||
if len(model) == 0: |
||||
# no messages, so we don't worry (if I recall correctly, |
||||
# get_iter_first will fail if the model is empty) |
||||
return |
||||
|
||||
iter = model.get_iter_first() |
||||
while iter: |
||||
message = model.get_value(iter, 0) |
||||
message.read = True |
||||
iter = model.iter_next(iter) |
||||
|
||||
self.count = 0 |
||||
return |
||||
|
||||
def _open_url(self, widget, user_data=None): |
||||
"""Opens an URL (used mostly from popup menu items.)""" |
||||
webbrowser.open_new_tab(user_data) |
||||
return |
||||
|
||||
def update_window_size(self, size): |
||||
"""Called when the window size changes, so the grid needs to change |
||||
its word-wrapping size.""" |
||||
model = self._grid.get_model() |
||||
if len(model) == 0: |
||||
return |
||||
|
||||
column = self._grid.get_column(1) # 0=avatar, 1=message |
||||
iter = model.get_iter_first() |
||||
path = model.get_path(iter) |
||||
|
||||
width = win_width - 85 # 48 = icon size |
||||
# TODO: Find out where those 37 pixels came from and/or if they |
||||
# are platform specific. |
||||
|
||||
for renderer in column.get_cell_renderers(): |
||||
renderer.set_property('wrap-width', width) |
||||
|
||||
while iter: |
||||
path = model.get_path(iter) |
||||
model.row_changed(path, iter) |
||||
iter = model.iter_next(iter) |
||||
return |
||||
Loading…
Reference in new issue