Fix opening new activities

1 files changed, 24 insertions, 0 deletions

diff --git a/shell/model/BuddyInfo.py b/shell/model/BuddyInfo.py
new file mode 100644
index 0000000..68f0a44
--- /dev/null
+++ b/shell/model/BuddyInfo.py
@@ -0,0 +1,24 @@
+from sugar.presence import PresenceService
+from sugar.canvas.IconColor import IconColor
+
+class BuddyInfo:
+	def __init__(self, buddy=None):
+		if buddy:
+			self.set_name(buddy.get_name())
+			self.set_color(buddy.get_color())
+
+	def set_name(self, name):
+		self._name = name
+
+	def set_color(self, color_string):
+		self._color = IconColor(color_string)
+
+	def get_name(self):
+		return self._name
+
+	def get_color(self):
+		return self._color
+
+	def get_buddy(self):
+		pservice = PresenceService.get_instance()
+		return pservice.get_buddy_by_name(self._name)