X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Fscenes%2Fnight.py;h=330530b029f947edc26729e0cf99c3dae4913d78;hb=26b52ca902b0a22f30009fb61bc222601e7775fb;hp=5d3721bd99e9c9a7cc195acb9f444e2e5736f01e;hpb=17c77c62d4679e2002ec49a01e8a3bc39829451b;p=tabakrolletjie.git

diff --git a/tabakrolletjie/scenes/night.py b/tabakrolletjie/scenes/night.py
index 5d3721b..330530b 100644
--- a/tabakrolletjie/scenes/night.py
+++ b/tabakrolletjie/scenes/night.py
@@ -5,27 +5,29 @@ import pygame.locals as pgl
 import pymunk
 
 from .base import BaseScene
-from ..obstacles import Wall
+from ..lights import BaseLight
+from ..obstacles import BaseObstacle
 from ..events import SceneChangeEvent
 
 
 class NightScene(BaseScene):
     def enter(self, gamestate):
-        import pprint
-        pprint.pprint(gamestate.station)
-
         self._space = pymunk.Space()
-
-        self._obstacles = []
-        self._lights = []
-        for obs in gamestate.station['obstacles']:
-            wall = Wall(obs['vertices'], self._space)
-            self._obstacles.append(wall)
+        self._obstacles = [
+            BaseObstacle.load(cfg) for cfg in gamestate.station["obstacles"]]
+        self._lights = [
+            BaseLight.load(cfg) for cfg in gamestate.station["lights"]]
+        for obs in self._obstacles:
+            obs.add(self._space)
+        for light in self._lights:
+            light.add(self._space)
 
     def render(self, surface, gamestate):
         surface.fill((0, 0, 255))
         for obs in self._obstacles:
             obs.render(surface)
+        for light in self._lights:
+            light.render(surface)
 
     def event(self, ev, gamestate):
         if ev.type == pgl.KEYDOWN: