bluezd
bluezd

Reputation: 53

Pygtk: How do I change the treestore color of a row?

Currently I am developing a tool by using Python with Gtk+ library. And I met an issue here as $subject. Is there any effective way for changing the treestore "1/8"'s background color to red ? Something like the below image:

http://imagebin.ca/v/1pZgJ61cWi9D

I use treestore.set_value to change it, but unfortunately it does not work for me, it can only change the value of the treestore.

No idea about it so I raise a question here. Can anybody keep an eye of it ? Any suggestions that would be appreciated.

Upvotes: 1

Views: 1232

Answers (1)

M.javid
M.javid

Reputation: 6637

you can use documents of pygtk2.0 about TreeViewColumn.set_cell_data_func method and read important page about CellRenderers and special properties of PyGTK2.0 that also is useful on version 3 of PyGTK :)

Below code can helping you:

#!/usr/bin/env python

try:
    import pygtk; pygtk.require('2.0')
except:
    pass
import gtk
import pango

COL_FIRST_NAME = 0
COL_LAST_NAME = 1
COL_YEAR_BORN = 2
COL_COLOR = 3

class TreeStoreExample:

    def createAndFillModel(self):

        treestore = gtk.TreeStore(str, str, str, str)
        toplevel1 = treestore.append(None)
        toplevel2 = treestore.append(None)
        child = treestore.append(toplevel2)
        pre_child = treestore.append(child)

        tree = [(toplevel1,
                      COL_FIRST_NAME, "Maria",
                      COL_LAST_NAME, "Incognito",
                      COL_YEAR_BORN, 1982,
                      COL_COLOR, 'black'),

                (toplevel2,
                      COL_FIRST_NAME, "Jane",
                      COL_LAST_NAME, "Average",
                      COL_YEAR_BORN, 1962,
                      COL_COLOR, 'black'),

                (child,
                      COL_FIRST_NAME, "Janinita",
                      COL_LAST_NAME, "Average",
                      COL_YEAR_BORN, 1985,
                      COL_COLOR, 'black'),

                (pre_child,
                      COL_FIRST_NAME, "ABC",
                      COL_LAST_NAME, "DEF",

                      COL_COLOR, 'black')
        ]

        year_now = 2015
        for item in tree:
            iter = item[0]
            treestore.set(*item)
            if item[5] == COL_COLOR:
                treestore.set_value(iter, COL_YEAR_BORN, "age unknown")
                for x in range(1, len(treestore.get_path(iter)) + 1):
                    niter = treestore.get_iter(treestore.get_path(iter)[:x])
                    treestore.set_value(niter, COL_COLOR, "red")
            else:
                treestore.set_value(iter, COL_YEAR_BORN, year_now - item[6])

        return treestore

    def ageCellDataFunc(self, column, renderer, model, iter, data):

        year_now = 2015
        year_born = model.get_value(iter, COL_YEAR_BORN)

        if year_born and year_born <= year_now:
            age = year_now - year_born
            buf = "%u years old" % age
            renderer.set_property("foreground-set", gtk.FALSE)
        else:
            buf = "age unknown"
            #renderer.set_property("foreground", "red")
            model.set_value(iter, COL_COLOR, "red")
            for x in range(1, len(model.get_path(iter))):
                niter = model.get_iter(model.get_path(iter)[:x])
                node = model.set_value(niter, COL_COLOR, "red")
                print model.get_path(iter)[:x], ":", model.get_value(niter, COL_COLOR)

        #renderer.set_property("text", buf)

    def createViewAndModel(self):

        view = gtk.TreeView(self.createAndFillModel())
        self.view = view
        # --- Column 1 ---
        column = gtk.TreeViewColumn()
        column.set_title("First Name")
        view.append_column(column)
        renderer = gtk.CellRendererText()
        column.pack_start(renderer, gtk.TRUE)
        column.add_attribute(renderer, "text", COL_FIRST_NAME)

        # --- Column 2 ---
        column = gtk.TreeViewColumn()
        column.set_title("Last Name")
        view.append_column(column)
        renderer = gtk.CellRendererText()
        column.pack_start(renderer, gtk.TRUE)
        column.add_attribute(renderer, "text", COL_LAST_NAME)
        renderer.set_property("weight", pango.WEIGHT_BOLD)

        # --- Column 3 ---
        renderer = gtk.CellRendererText()
        renderer.set_property('foreground-set',True)
        column = gtk.TreeViewColumn("age", renderer, foreground=COL_COLOR)
        column.pack_start(renderer, gtk.TRUE)
        column.add_attribute(renderer, "text", COL_YEAR_BORN)
        column.add_attribute(renderer, "foreground", COL_COLOR)
        #column.set_title("age")
        view.append_column(column)
        #column.set_cell_data_func(renderer, self.ageCellDataFunc, None)

        #model = self.createAndFillModel()

        #view.set_model(model)

        view.get_selection().set_mode(gtk.SELECTION_NONE)

        return view

    def __init__(self):

        window = gtk.Window(gtk.WINDOW_TOPLEVEL)
        window.connect("delete_event", gtk.mainquit)

        view = self.createViewAndModel()
        window.add(view)

        window.show_all()

def main():
    gtk.main()
    return 0

if __name__ == "__main__":
    TreeStoreExample()
    main()

The ageCellDataFunc method not run because if set set_cell_data_func then add_attribute method not running.

Upvotes: 1

Related Questions