Traktor/app/board.py

#!/usr/bin/python3

import random
from images import *

from config import *


class Board:
    def __init__(self):
        self.__fields = []
        self.generate_board()
        self.fill()
        # print(self.__fields)

    def generate_board(self):
        for i in range(HORIZONTAL_NUM_OF_FIELDS):
            self.__fields.append([])
            for j in range(VERTICAL_NUM_OF_FIELDS):
                self.__fields[i].append(None)

    def fill(self):
        # colors = [C_RED, C_GREEN, C_BLUE]
        for i in range(len(self.__fields)):
            for j in range(len(self.__fields[i])):
                self.__fields[i][j] = random.choice(soilType)


    def draw(self, screen: pygame.Surface):
        size = (FIELD_SIZE/HORIZONTAL_NUM_OF_FIELDS, FIELD_SIZE/HORIZONTAL_NUM_OF_FIELDS)
        for x in range(len(self.__fields)):
            for y in range(len(self.__fields[x])):
                obj = self.__fields[x][y]
                i = x * FIELD_SIZE
                j = y * FIELD_SIZE
                if obj == 'grass':
                    grassrect.size = (size)
                    grassrect.center = (i, j)
                    screen.blit(grass, grassrect)
                if obj == 'corn':
                    cornrect.size = (size)
                    cornrect.center = (i, j)
                    screen.blit(corn, cornrect)
                if obj == 'sunflower':
                    sunflowerrect.size = (size)
                    sunflowerrect.center = (i, j)
                    screen.blit(sunflower, sunflowerrect)
                if obj == 'sand':
                    sandrect.size = (size)
                    sandrect.center = (i, j)
                    screen.blit(sand, sandrect)
                if obj == 'clay':
                    clayrect.size = (size)
                    clayrect.center = (i, j)
                    screen.blit(clay, clayrect)