diff options
Diffstat (limited to 'shell')
| -rw-r--r-- | shell/view/BuddyIcon.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/shell/view/BuddyIcon.py b/shell/view/BuddyIcon.py index 0cf90fe..607dec6 100644 --- a/shell/view/BuddyIcon.py +++ b/shell/view/BuddyIcon.py @@ -20,18 +20,17 @@ class BuddyIcon(MenuIcon): def _popup_action_cb(self, popup, action): self.popdown() - model = self._shell.get_model() - if action == BuddyMenu.ACTION_REMOVE_FRIEND: - friends = model.get_friends() - friends.remove(buddy) - buddy = self._friend.get_buddy() if buddy == None: return + model = self._shell.get_model() if action == BuddyMenu.ACTION_INVITE: activity = model.get_current_activity() activity.invite(buddy) elif action == BuddyMenu.ACTION_MAKE_FRIEND: friends = model.get_friends() friends.make_friend(buddy) + elif action == BuddyMenu.ACTION_REMOVE_FRIEND: + friends = model.get_friends() + friends.remove(buddy) |