animal.py

import os
import random
import game_config as gc

from pygame import image, transform

animals_count = dict((a, 0) for a in gc.asset_files)


def available_animals():
    return [a for (a, c) in animals_count.items() if c < 2]


class Animal:

    def __init__(self, index):
        self.index = index
        self.row = index // gc.num_tiles_side
        self.col = index % gc.num_tiles_side
        self.name = random.choice(available_animals())
        animals_count[self.name] += 1

        self.image_path = os.path.join(gc.asset_dir, self.name)
        self.image = image.load(self.image_path)
        self.image = transform.scale(self.image, (gc.image_size - 2
                * gc.margin, gc.image_size - 2 * gc.margin))
        self.box = self.image.copy()
        t = (200, 200, 200)
        self.box.fill(t)
        self.skip = False