shapeGenerator.py

import random
BLOCK = "■"
EMPTY = " "

def genereate(blockCount=4, areaSize=4):
    maxI = areaSize-1

    def getValidCords(cord: tuple[int, int]) -> list[tuple[int, int]]:
        y = cord[0]
        x = cord[1]
        validCords = []
        if y > 0:
            if (y-1, x) not in blocks:
                validCords.append((y-1, x))

        if y < maxI:
            if (y+1, x) not in blocks:
                validCords.append((y+1, x))

        if x > 0:
            if (y, x-1) not in blocks:
                validCords.append((y, x-1))

        if x < maxI:
            if (y, x+1) not in blocks:
                validCords.append((y, x+1))

        return validCords

    if blockCount > areaSize**2:
        raise Exception("blockCount is greater than content of the area")

    blocks = [(areaSize//2, areaSize//2)]
    choices = [(areaSize//2, areaSize//2)]
    for _ in range(blockCount-1):
        while True:
            choisedCord = random.choice(choices)
            validCords = getValidCords(choisedCord)
            if len(validCords) == 0:
                choices.remove(choisedCord)
            else:
                if len(validCords) == 1:
                    choices.remove(choisedCord)
                newcords = random.choice(validCords)
                blocks.append(newcords)
                choices.append(newcords)
                break

    minY = min([block[0] for block in blocks])
    maxY = max([block[0] for block in blocks])
    minX = min([block[1] for block in blocks])
    maxX = max([block[1] for block in blocks])

    shape = [[BLOCK if (y, x) in blocks else EMPTY for x in range(minX, maxX+1)] for y in range(minY, maxY+1)]
    return shape