Added some methods to Palette.

2 files changed, 22 insertions, 0 deletions

diff --git a/sugar/graphics/palette.py b/sugar/graphics/palette.py
index 760f702..2179663 100644
--- a/sugar/graphics/palette.py
+++ b/sugar/graphics/palette.py
@@ -83,6 +83,23 @@ class Palette(gobject.GObject):
         self._separator.show()
         self._menu.insert(item, len(self._menu.get_children()) - 2)
 
+    def insert_menu_item(self, item, index=-1):
+        self._separator.show()
+        if index < 0:
+            self._menu.insert(item, len(self._menu.get_children()) - 2)
+        else:
+            self._menu.insert(item, index + 2)
+
+    def remove_menu_item(self, index):
+        if index > len(self._menu.get_children()) - 4:
+            raise ValueError('index %i out of range' % index)
+        self._menu.remove(self._menu.get_children()[index + 2])
+        if len(self._menu.get_children()) == 0:
+            self._separator.hide()
+
+    def menu_item_count(self):
+        return len(self._menu.get_children()) - 4
+        
     def set_content(self, widget):
         self._content.set_widget(widget)
         self._content.show()
diff --git a/sugar/graphics/toolbutton.py b/sugar/graphics/toolbutton.py
index 2c85432..79e70a6 100644
--- a/sugar/graphics/toolbutton.py
+++ b/sugar/graphics/toolbutton.py
@@ -35,6 +35,9 @@ class ToolButton(gtk.ToolButton):
         self.set_icon_widget(icon)
         icon.show()
 
+    def get_palette(self):
+        return self._palette
+    
     def set_palette(self, palette):
         self._palette = palette
         self._palette.props.invoker = WidgetInvoker(self.child)
@@ -42,3 +45,5 @@ class ToolButton(gtk.ToolButton):
     def set_tooltip(self, text):
         self._palette = Palette(text)
         self._palette.props.invoker = WidgetInvoker(self.child)
+    
+    palette = property(get_palette, set_palette)