1 files changed, 136 insertions, 0 deletions
diff --git a/filedialog.py b/filedialog.py
new file mode 100644
index 0000000..ea6da02
--- /dev/null
+++ b/filedialog.py
@@ -0,0 +1,136 @@
+from gi.repository import Gtk, Gdk
+from sugar3.graphics.toolbarbox import ToolbarBox
+from sugar3.graphics.toolbutton import ToolButton
+from gettext import gettext as _
+import os
+
+
+class FileDialog(Gtk.Dialog):
+    """
+    Simple file dialog:
+
+    from filedialog import FileDialog
+    dialog = FileDialog(['Graphics', '/path/of/graphics'])
+    dialog.run()
+    path = dialog.get_path()
+    """
+
+    def __init__(self, dirs, window=None, button=None):
+        Gtk.Dialog.__init__(self, flags=Gtk.DialogFlags.DESTROY_WITH_PARENT)
+
+        self.example_path = None
+        self.expanders = []
+        self.dirs = dirs
+        self.button = button
+
+        x, y = (Gdk.Screen.width() / 1.5, Gdk.Screen.height() / 1.5)
+        self.set_size_request(x, y)
+
+        toolbox = self.build_toolbar()
+        expands = self.build_scroll()
+
+        self.vbox.pack_start(toolbox, False, False, 0)
+
+        self.vbox.pack_start(expands, True, True, 5)
+
+        self.set_decorated(False)
+        self.set_skip_pager_hint(True)
+        self.set_skip_taskbar_hint(True)
+        self.set_resizable(False)
+        self.set_modal(True)
+
+        if window:
+            self.set_parent(window)
+
+        self.modify_bg(Gtk.StateType.NORMAL, Gdk.color_parse('#F3EEEE'))
+        self.show_all()
+
+    def get_path(self):
+        return self.example_path
+
+    def _destroy(self, widget, sample=False):
+        if sample:
+            self.example_path = widget.get_tooltip_text()
+
+        if self.button:
+            self.button.set_icon_name("pippy-openoff")
+
+        self.destroy()
+
+    def build_toolbar(self):
+        toolbox = ToolbarBox()
+        toolbar = toolbox.toolbar
+
+        label = Gtk.Label(_('Open an example bundle'))
+        label.modify_fg(Gtk.StateType.NORMAL,
+            Gdk.color_parse('white'))
+
+        item = Gtk.ToolItem()
+        item.add(label)
+
+        close = ToolButton('entry-cancel')
+        close.connect('clicked', self._destroy)
+
+        separator = Gtk.SeparatorToolItem()
+        separator.props.draw = False
+        separator.set_expand(True)
+
+        toolbar.insert(item, -1)
+        toolbar.insert(separator, -1)
+        toolbar.insert(close, -1)
+
+        toolbox.set_size_request(-1, 35)
+
+        return toolbox
+
+    def build_scroll(self):
+        scroll = Gtk.ScrolledWindow()
+        scroll.set_policy(Gtk.PolicyType.AUTOMATIC,
+                Gtk.PolicyType.AUTOMATIC)
+
+        vbox = Gtk.VBox()
+        scroll.add_with_viewport(vbox)
+
+        dirs = self.dirs
+        for dir_ in dirs:
+            dir_path = dir_[1]
+            dir_name = dir_[0]
+            expand = self.build_expand(dir_path, dir_name)
+            if not expand:
+                continue
+            vbox.pack_start(expand, False, False, 2)
+            expand.show()
+            self.expanders.append(expand)
+
+        return scroll
+
+    def build_expand(self, path, name):
+        if not os.path.exists(path):
+            return None
+
+        expander = Gtk.Expander()
+        expander.set_label(name)
+        expander.modify_fg(Gtk.StateType.NORMAL,
+            Gdk.color_parse('black'))
+
+        vbox = Gtk.VBox()
+        files = sorted(os.listdir(path))
+
+        if not files:
+            return None
+
+        for _file in files:
+            if _file.endswith('~'):
+                continue
+            entry = {"name": _(_file.capitalize()),
+                     "path": os.path.join(path, _file)}
+
+            button = Gtk.Button(entry['name'])
+            button.set_tooltip_text(entry['path'])
+            button.set_has_tooltip(False)
+            button.connect('clicked', self._destroy, True)
+            vbox.pack_start(button, False, False, 1)
+
+        expander.add(vbox)
+
+        return expander