Viewing file:  treemodel.py (4.23 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
#!/usr/bin/env python
'''Tree View/Generic Tree Model
This test is designed to demonstrate creating a new type of tree model
in python for use with the new tree widget in gtk 2.0.'''
import gtk
import gobject
# to create a new GtkTreeModel from python, you must derive from
# TreeModel.
class MyTreeModel(gtk.GenericTreeModel):
'''This class represents the model of a tree. The iterators used
to represent positions are converted to python objects when passed
to the on_* methods. This means you can use any python object to
represent a node in the tree. The None object represents a NULL
iterator.
In this tree, we use simple tuples to represent nodes, which also
happen to be the tree paths for those nodes. This model is a tree
of depth 3 with 5 nodes at each level of the tree. The values in
the tree are just the string representations of the nodes.'''
TREE_DEPTH = 4
TREE_SIBLINGS = 5
def __init__(self):
'''constructor for the model. Make sure you call
PyTreeModel.__init__'''
gtk.GenericTreeModel.__init__(self)
# the implementations for TreeModel methods are prefixed with on_
def on_get_flags(self):
'''returns the GtkTreeModelFlags for this particular type of model'''
return 0
def on_get_n_columns(self):
'''returns the number of columns in the model'''
return 1
def on_get_column_type(self, index):
'''returns the type of a column in the model'''
return gobject.TYPE_STRING
def on_get_path(self, node):
'''returns the tree path(a tuple of indices at the various
levels) for a particular node.'''
return node
def on_get_iter(self, path):
'''returns the node corresponding to the given path. In our
case, the node is the path'''
return path
def on_get_value(self, node, column):
'''returns the value stored in a particular column for the node'''
assert column == 0
return `node`
def on_iter_next(self, node):
'''returns the next node at this level of the tree'''
if node != None:
if node[-1] == self.TREE_SIBLINGS - 1: # last node at level
return None
return node[:-1] +(node[-1]+1,)
def on_iter_children(self, node):
'''returns the first child of this node'''
if node == None: # top of tree
return(0,)
if len(node) >= self.TREE_DEPTH: # no more levels
return None
return node +(0,)
def on_iter_has_child(self, node):
'''returns true if this node has children'''
return node == None or len(node) < self.TREE_DEPTH
def on_iter_n_children(self, node):
'''returns the number of children of this node'''
if node == None or len(node) < self.TREE_DEPTH:
return self.TREE_SIBLINGS
else:
return 0
def on_iter_nth_child(self, node, n):
'''returns the nth child of this node'''
if node == None:
return(n,)
if len(node) < self.TREE_DEPTH and n < self.TREE_SIBLINGS:
return node +(n,)
else:
return None
def on_iter_parent(self, node):
'''returns the parent of this node'''
assert node != None
if len(node) == 0:
return None
else:
return node[:-1]
class GenericTreeModelDemo(gtk.Window):
def __init__(self, parent=None):
gtk.Window.__init__(self)
try:
self.set_screen(parent.get_screen())
except AttributeError:
self.connect('destroy', lambda *w: gtk.main_quit())
self.set_title(self.__class__.__name__)
scrolled_window = gtk.ScrolledWindow()
scrolled_window.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
self.add(scrolled_window)
model = MyTreeModel()
model = model.filter_new()
tree_view = gtk.TreeView(model)
cell = gtk.CellRendererText()
# the text in the column comes from column 0
column = gtk.TreeViewColumn("tuples", cell, text=0)
tree_view.append_column(column)
scrolled_window.add(tree_view)
self.show_all()
def main():
GenericTreeModelDemo()
gtk.main()
if __name__ == '__main__':
main()
|