def generateRooms(self): rooms = [] for i in range(self.number_of_rooms): room = Room(i) for j in range(room.number_of_enemies): rand = random.randint(0, 1) if rand == 0: room.enemies.append(Enemy.Bat()) else: room.enemies.append(Enemy.Zombie()) rooms.append(room) return rooms