65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
import random
|
|
|
|
import goocanvas
|
|
|
|
from sugar.canvas.IconItem import IconItem
|
|
import Theme
|
|
|
|
class FriendIcon(IconItem):
|
|
def __init__(self, friend):
|
|
IconItem.__init__(self, icon_name='stock-buddy',
|
|
color=friend.get_color(), size=48,
|
|
x=random.random() * 1100,
|
|
y=random.random() * 800)
|
|
self._friend = friend
|
|
|
|
def get_friend(self):
|
|
return self._friend
|
|
|
|
class FriendsGroup(goocanvas.Group):
|
|
def __init__(self, data_model):
|
|
goocanvas.Group.__init__(self)
|
|
self._friend_to_child = {}
|
|
|
|
self._theme = Theme.get_instance()
|
|
self._theme.connect("theme-changed", self.__theme_changed_cb)
|
|
|
|
color = self._theme.get_home_mesh_color()
|
|
self._mesh_rect = goocanvas.Rect(width=1200, height=900,
|
|
line_width=0, fill_color=color)
|
|
self.add_child(self._mesh_rect)
|
|
|
|
color = self._theme.get_home_friends_color()
|
|
self._friends_rect = goocanvas.Rect(x=50, y=50, width=1100, height=800,
|
|
line_width=0, fill_color=color,
|
|
radius_x=30, radius_y=30)
|
|
self.add_child(self._friends_rect)
|
|
|
|
for friend in data_model:
|
|
self.add_friend(friend)
|
|
|
|
data_model.connect('friend-added', self.__friend_added_cb)
|
|
data_model.connect('friend-removed', self.__friend_removed_cb)
|
|
|
|
def __theme_changed_cb(self, theme):
|
|
color = self._theme.get_home_friends_color()
|
|
self._friends_rect.set_property("fill-color", color)
|
|
color = self._theme.get_home_mesh_color()
|
|
self._mesh_rect.set_property("fill-color", color)
|
|
|
|
def add_friend(self, friend):
|
|
icon = FriendIcon(friend)
|
|
self.add_child(icon)
|
|
self._friend_to_child[friend] = icon
|
|
|
|
def remove_friend(self, friend):
|
|
icon = self._friend_to_child[friend]
|
|
self.remove_child(self._root.find_child(icon))
|
|
del self._friend_to_child[friend]
|
|
|
|
def __friend_added_cb(self, data_model, friend):
|
|
self.add_friend(friend)
|
|
|
|
def __friend_removed_cb(self, data_model, friend):
|
|
self.remove_friend(friend)
|