X-Git-Url: https://git.ctpug.org.za/?p=koperkapel.git;a=blobdiff_plain;f=koperkapel%2Fgamelib%2Fenemy_generator.py;fp=koperkapel%2Fgamelib%2Fenemy_generator.py;h=d399f19f12af367492e6664c5731cb634b7202f0;hp=b5a1dd40969f363f9ab25dec2f1f6b1c0a6a8dbc;hb=11ff8390096b6205f4f7aaa38a8922775fabd1d4;hpb=aee40d4ecdbdcb4879a9e5136d75ae25d9e90239

diff --git a/koperkapel/gamelib/enemy_generator.py b/koperkapel/gamelib/enemy_generator.py
index b5a1dd4..d399f19 100644
--- a/koperkapel/gamelib/enemy_generator.py
+++ b/koperkapel/gamelib/enemy_generator.py
@@ -4,7 +4,7 @@ import os
 from pgzero.actor import Actor
 from pgzero.clock import each_tick
 
-from .enemy_roach import get_enemy_roach
+from .enemy_roach import get_enemy_roach, get_rat
 
 class EnemyGenerator(Actor):
     """Generators are currently invisble, but we want the update hook."""
@@ -35,6 +35,10 @@ class EnemyGenerator(Actor):
             roach = get_enemy_roach(self.level)
             self._made_enemies.append(roach)
             self.level.add_enemy(roach, *self.gen_pos)
+        elif self.enemy_type == 'rat':
+            roach = get_rat(self.level)
+            self._made_enemies.append(roach)
+            self.level.add_enemy(roach, *self.gen_pos)
 
     def killed(self, enemy):
         if enemy in self._made_enemies: