From 9ce768ca394551a623ac7e36ea5ae2f9ca57ce8a Mon Sep 17 00:00:00 2001
From: Dan Williams <dcbw@localhost.localdomain>
Date: Wed, 27 Sep 2006 15:35:34 +0000
Subject: Add and remove activities to/from Buddy objects at appropriate times

---
(limited to 'services')

diff --git a/services/presence/PresenceService.py b/services/presence/PresenceService.py
index b43506e..9898b2b 100644
--- a/services/presence/PresenceService.py
+++ b/services/presence/PresenceService.py
@@ -451,15 +451,13 @@ class PresenceService(object):
 		if activity:
 			activity.add_service(service)
 
-		# Add the activity to its buddy
-		# FIXME: use something other than name to attribute to buddy
-		name = service.get_name()
-		buddy = None
-		try:
-			buddy = self._buddies[name]
-			buddy.add_activity(activity)
-		except KeyError:
-			pass		
+			# Add the activity to its buddy
+			# FIXME: use something other than name to attribute to buddy
+			try:
+				buddy = self._buddies[service.get_name()]
+				buddy.add_activity(activity)
+			except KeyError:
+				pass		
 
 		if not was_valid and activity.is_valid():
 			self._dbus_helper.ActivityAppeared(activity.object_path())
@@ -473,18 +471,16 @@ class PresenceService(object):
 
 		activity = self._activities[actid]
 
-		# Remove the activity from its buddy
-		# FIXME: use something other than name to attribute to buddy
-		name = service.get_name()
-		buddy = None
-		try:
-			buddy = self._buddies[name]
-			buddy.remove_activity(activity)
-		except KeyError:
-			pass		
-
 		activity.remove_service(service)
 		if len(activity.get_services()) == 0:
+			# Remove the activity from its buddy
+			# FIXME: use something other than name to attribute to buddy
+			try:
+				buddy = self._buddies[service.get_name()]
+				buddy.remove_activity(activity)
+			except KeyError:
+				pass		
+
 			# Kill the activity
 			self._dbus_helper.ActivityDisappeared(activity.object_path())
 			del self._activities[actid]
--
cgit v0.9.1