Palette fixes

1 files changed, 8 insertions, 4 deletions

diff --git a/sugar/graphics/toolbutton.py b/sugar/graphics/toolbutton.py
index a2a6ce2..2f6e9bf 100644
--- a/sugar/graphics/toolbutton.py
+++ b/sugar/graphics/toolbutton.py
@@ -31,10 +31,14 @@ class ToolButton(gtk.ToolButton):
         icon.show()
 
     def set_palette(self, palette):
-        self.connect('clicked', palette.display)
-        palette.props.parent = self
-        palette.props.alignment = ALIGNMENT_BOTTOM_LEFT
-
+        self._palette = palette
+        self._palette.props.parent = self
+        self._palette.props.alignment = ALIGNMENT_BOTTOM_LEFT
+        self.connect('clicked', self._display_palette_cb)
+    
     def set_tooltip(self, text):
         tp = gtk.Tooltips()
         self.set_tooltip(tp, text, text)
+
+    def _display_palette_cb(self, widget):
+        self._palette.popup()