diff mamba/widgets/levelbutton.py @ 314:fba74abad7b7

Pass Level() objects to LevelHabitat and LevelEditorHabitat, in preparation for generating thumbnails from levels in levelmenu
author Stefano Rivera <stefano@rivera.za.net>
date Fri, 16 Sep 2011 19:07:51 +0200
parents b371c2ea309e
children 86c5c6afdac0
line wrap: on
line diff
--- a/mamba/widgets/levelbutton.py	Fri Sep 16 18:56:42 2011 +0200
+++ b/mamba/widgets/levelbutton.py	Fri Sep 16 19:07:51 2011 +0200
@@ -9,9 +9,10 @@
 
 class LevelButton(Button):
 
-    def __init__(self, rect, text, done=False):
+    def __init__(self, rect, level, done=False):
         super(LevelButton, self).__init__(rect)
-        self.text = text
+        self.level = level
+        self.text = level.name
         self.done = done
         self.focussable = True
         self.border = 3
@@ -23,10 +24,8 @@
         self.surface = pygame.Surface(self.rect.size, SRCALPHA)
         self.surface.fill(0)
 
-        image = None
         if self.done:
             image = load_image('menus/tick.png')
-        if image:
             self.surface.blit(image, image.get_rect())
 
         self._text = TextWidget((0, 0), self.text, fontsize=24)