import gtk import gobject import hippo from sugar.graphics.menu import Menu from sugar.graphics.canvasicon import CanvasIcon from sugar.graphics.ClipboardBubble import ClipboardBubble from sugar.graphics import style class ClipboardMenuItem(ClipboardBubble): def __init__(self, percent = 0, stylesheet="clipboard.Bubble"): ClipboardBubble.__init__(self, percent = percent) style.apply_stylesheet(self, stylesheet) class ClipboardMenu(Menu): ACTION_DELETE = 0 ACTION_SHARE = 1 ACTION_STOP_DOWNLOAD = 2 def __init__(self, name, percent): Menu.__init__(self, name) if percent < 100: self._progress_bar = ClipboardMenuItem(percent) self._root.append(self._progress_bar) else: self._progress_bar = None self._remove_icon = None self._stop_icon = None self._update_icons(percent) def _update_icons(self, percent): if percent == 100: if not self._remove_icon: self._remove_icon = CanvasIcon(icon_name='stock-remove') self.add_action(self._remove_icon, ClipboardMenu.ACTION_DELETE) if self._stop_icon: self.remove_action(self._stop_icon) self._stop_icon = None else: if not self._stop_icon: self._stop_icon = CanvasIcon(icon_name='stock-close') self.add_action(self._stop_icon, ClipboardMenu.ACTION_STOP_DOWNLOAD) if self._remove_icon: self.remove_action(self._remove_icon) self._remove_icon = None def set_percent(self, percent): if self._progress_bar: self._progress_bar.set_property('percent', percent) self._update_icons(percent)