# HG changeset patch
# User Neil Muller <drnlmuller@gmail.com>
# Date 1316301032 -7200
# Node ID dee11e9a37b03a026dfdaef20db6975b474362ab
# Parent  111c3a6274ecdd011559a7c9b4236603c66943aa
Caption fiddling

diff -r 111c3a6274ec -r dee11e9a37b0 mamba/habitats/level.py
--- a/mamba/habitats/level.py	Sun Sep 18 01:10:06 2011 +0200
+++ b/mamba/habitats/level.py	Sun Sep 18 01:10:32 2011 +0200
@@ -1,8 +1,9 @@
 """Where a level object hides."""
 
+import pygame.display
 from pygame.locals import KEYDOWN
 
-from mamba.constants import ESCAPE_KEYS
+from mamba.constants import ESCAPE_KEYS, NAME
 from mamba.engine import Habitat, LevelCompletedEvent, SnakeDiedEvent
 from mamba.gamestate import level_done
 from mamba.widgets.game import GameWidget
@@ -16,6 +17,7 @@
         self.go_menu = go_menu
         self.go_next_level = go_next_level
         self.world = World(level)
+        self.level_name = level.name
         self.game_widget = GameWidget(self.world)
         self.container.add(self.game_widget)
         self.container.add_callback(KEYDOWN, self.keydown_event)
@@ -28,6 +30,16 @@
             self.go_menu()
             return True
 
+    def on_enter(self):
+        """Set the caption to the level name"""
+        super(LevelHabitat, self).on_enter()
+        pygame.display.set_caption('%s: %s' % (NAME, self.level_name))
+
+    def on_exit(self):
+        """Fix name"""
+        super(LevelHabitat, self).on_exit()
+        pygame.display.set_caption(NAME)
+
     def level_completed(self, ev, widget):
         if not self.world.pause():
             return False