# HG changeset patch
# User Stefano Rivera <stefano@rivera.za.net>
# Date 1316271520 -7200
# Node ID fefb37fb9c6d1465faa2f08c137fa8ac94265cac
# Parent  546b68aedf0c21c1d2210355b719ea21aaf3d241
Focus the level menu first item again

diff -r 546b68aedf0c -r fefb37fb9c6d mamba/habitats/levelmenu.py
--- a/mamba/habitats/levelmenu.py	Sat Sep 17 16:52:52 2011 +0200
+++ b/mamba/habitats/levelmenu.py	Sat Sep 17 16:58:40 2011 +0200
@@ -22,7 +22,8 @@
     def __init__(self):
         super(LevelMenu, self).__init__()
         self.level_buttons = {}
-        self.container.add(GridContainer(width=self.WIDTH))
+        self.grid_container = GridContainer(width=self.WIDTH)
+        self.container.add(self.grid_container)
         self.container.add_callback(KEYDOWN, self.keydown_event)
 
     def on_enter(self):
@@ -38,6 +39,8 @@
     def _setup_levels(self, ev=None):
         if ev is not None:
             self.setup_levels()
+            # We now have widgets, let's give them focus
+            self.container.adjust_focus(1)
         else:
             HabitatSetupEvent.post()
 
@@ -48,7 +51,7 @@
                              level)
         button.add_callback('clicked', self.level_selected, name)
         button.done = level.unique_name() in done_levels
-        self.container.add(button)
+        self.grid_container.add(button)
         self.level_buttons[name] = button
 
     def setup_levels(self):