Constructing multi-level GtkMenu by example of time zone hierarchy.

gtk_menu_hierarchy.py

""" Constructing multi-level GtkMenu by example of time zone hierarchy.
The example below is missing a GtkWindow and a gtk.main() to work."""

import gtk
from collections import defaultdict
from commands import getoutput

def autovivified():
    return defaultdict(autovivified)

def cb_selected(menuitem, data):
    pass

hierarchy = autovivified()

for line in getoutput("awk '/^[^#]/{ print $3,$1 }' /usr/share/zoneinfo/zone.tab").split('\n'):
    timezone, country_code = line.split()
    submenu, parts = hierarchy, timezone.split('/')
    for i, part in enumerate(parts, 1):
        if i != len(parts): submenu = submenu[part]
        else: submenu[part] = country_code

def _build_menu(d):
    menu = gtk.Menu()
    for k, v in d.items():
        menuitem = gtk.MenuItem(k)
        if isinstance(v, dict):
            menuitem.set_submenu(_build_menu(v))
        else:
            menuitem.connect('activate', cb_selected, v)
        menuitem.show()
        menu.append(menuitem)
    menu.show()
    return menu

menu = _build_menu(hierarchy)