tree for svenne

diff --git a/src/tree.py b/src/tree.py
new file mode 100644 (file)
index 0000000..ff5e28f
--- /dev/null
+++ b/src/tree.py
@@ -0,0 +1,117 @@
+#!/usr/bin/env python
+# vim: ts=4:sw=4:tw=78:nowrap
+import pygtk
+pygtk.require("2.0")
+import gtk, gobject
+
+tasks =  {
+    "Buy groceries": "Go to Asda after work",
+    "Do some programming": "Remember to update your software",
+    "Power up systems": "Turn on the client but leave the server",
+    "Watch some tv": "Remember to catch ER"
+    } 
+
+class tree:
+    def __init__(self):
+        Store = InfoModel()    
+        Display = DisplayModel()
+
+        self.mdl = Store.get_model()
+        self.view = Display.make_view( self.mdl )
+        self.view.show_all()
+
+    def get_tree(self):
+        return self.view
+      
+
+class GUI_Controller:
+    """ The GUI class is the controller for our application """
+    def __init__(self):
+        # setup the main window
+        self.root = gtk.Window(type=gtk.WINDOW_TOPLEVEL)
+        self.root.set_title("CellRenderer Example")
+        self.root.connect("destroy", self.destroy_cb)
+
+        myTree = tree()
+        self.root.add(myTree.get_tree())
+
+        self.root.show_all()
+        
+        return
+    def destroy_cb(self, *kw):
+        """ Destroy callback to shutdown the app """
+        gtk.main_quit()
+        return
+    def run(self):
+        """ run is called to set off the GTK mainloop """
+        gtk.main()
+       return  
+
+class InfoModel:
+    """ The model class holds the information we want to display """
+    def __init__(self):
+        """ Sets up and populates our gtk.TreeStore """
+        self.tree_store = gtk.TreeStore( gobject.TYPE_STRING,
+                                         gobject.TYPE_BOOLEAN )
+        # places the global people data into the list
+        # we form a simple tree.
+        for item in tasks.keys():
+            parent = self.tree_store.append( None, (item, None) )
+            self.tree_store.append( parent, (tasks[item],None) )
+        return
+    def get_model(self):
+        """ Returns the model """
+        if self.tree_store:
+            return self.tree_store 
+        else:
+            return None
+
+class DisplayModel:
+    """ Displays the Info_Model model in a view """
+    def make_view( self, model ):
+        """ Form a view for the Tree Model """
+        self.view = gtk.TreeView( model )
+        # setup the text cell renderer and allows these
+        # cells to be edited.
+        self.renderer = gtk.CellRendererText()
+        self.renderer.set_property( 'editable', True )
+        self.renderer.connect( 'edited', self.col0_edited_cb, model )
+
+        # The toggle cellrenderer is setup and we allow it to be
+        # changed (toggled) by the user.
+        self.renderer1 = gtk.CellRendererToggle()
+        self.renderer1.set_property('activatable', True)
+        self.renderer1.connect( 'toggled', self.col1_toggled_cb, model )
+               
+        # Connect column0 of the display with column 0 in our list model
+        # The renderer will then display whatever is in column 0 of
+        # our model .
+        self.column0 = gtk.TreeViewColumn("Name", self.renderer, text=0)
+               
+        # The columns active state is attached to the second column
+        # in the model.  So when the model says True then the button
+        # will show as active e.g on.
+        self.column1 = gtk.TreeViewColumn("Complete", self.renderer1 )
+        self.column1.add_attribute( self.renderer1, "active", 1)
+        self.view.append_column( self.column0 )
+        self.view.append_column( self.column1 )
+        return self.view
+    def col0_edited_cb( self, cell, path, new_text, model ):
+        """
+        Called when a text cell is edited.  It puts the new text
+        in the model so that it is displayed properly.
+        """
+        print "Change '%s' to '%s'" % (model[path][0], new_text)
+        model[path][0] = new_text
+        return
+    def col1_toggled_cb( self, cell, path, model ):
+        """
+        Sets the toggled state on the toggle button to true or false.
+        """
+        model[path][1] = not model[path][1]
+        print "Toggle '%s' to: %s" % (model[path][0], model[path][1],)
+        return
+
+if __name__ == '__main__':
+    myGUI = GUI_Controller()
+    myGUI.run()