add ui [folder|file]

ctl

@@ -73,6 +73,11 @@ if [ "$1" == "open" ]; then
     fi
     exit 0
 fi
+if [ "$1" == "ui" ]; then
+    shift
+    python2 $NAME/oml/ui.py $@
+    exit $?
+fi
 
 cd $BASE/$NAME
 python2 oml $@

oml/ui.py

@@ -0,0 +1,82 @@
+# encoding: utf-8
+# vi:si:et:sw=4:sts=4:ts=4
+DEBUG=False
+try:
+    from gi.repository import Gtk, GObject
+    GObject.threads_init()
+    use_Gtk = True
+except:
+    from Tkinter import Tk
+    import tkFileDialog
+    use_Gtk = False
+
+class GtkUI:
+    def selectFolder(self, data):
+        dialog = Gtk.FileChooserDialog(data.get("title", "Select Folder"),
+                               None,
+                               Gtk.FileChooserAction.SELECT_FOLDER,
+                               (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL,
+                                Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
+        dialog.set_default_response(Gtk.ResponseType.OK)
+
+        response = dialog.run()
+        if response == Gtk.ResponseType.OK:
+            filename = dialog.get_filename()
+            if DEBUG:
+                print filename, 'selected'
+        elif response == Gtk.ResponseType.CANCEL:
+            if DEBUG:
+                print 'Closed, no files selected'
+            filename = None
+        dialog.destroy()
+        while Gtk.events_pending():
+                Gtk.main_iteration()
+        if DEBUG:
+            print "done"
+        return filename
+
+    def selectFile(self, data):
+        dialog = Gtk.FileChooserDialog(data.get("title", "Select File"),
+                               None,
+                               Gtk.FileChooserAction.OPEN,
+                               (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL,
+                                Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
+        dialog.set_default_response(Gtk.ResponseType.OK)
+
+        response = dialog.run()
+        if response == Gtk.ResponseType.OK:
+            filename = dialog.get_filename()
+            if DEBUG:
+                print filename, 'selected'
+        elif response == Gtk.ResponseType.CANCEL:
+            if DEBUG:
+                print 'Closed, no files selected'
+            filename = None
+        dialog.destroy()
+        while Gtk.events_pending():
+                Gtk.main_iteration()
+        if DEBUG:
+            print "done"
+        return filename
+
+class TkUI:
+    def __init__(self):
+        self.root = Tk()
+        self.root.withdraw() #hiding tkinter window
+    def selectFolder(self, data):
+        return tkFileDialog.askdirectory(title=data.get("title", "Select Folder"))
+
+    def selectFile(self, data):
+        return tkFileDialog.askopenfilename(title=data.get("title", "Select File"))
+
+if use_Gtk:
+    ui = GtkUI()
+else:
+    ui = TkUI()
+
+if __name__ == '__main__':
+    import sys
+    if len(sys.argv) == 2 and sys.argv[1] == 'folder':
+        print ui.selectFolder({})
+    else:
+        print ui.selectFile({})