Piece.py

import click
from math import floor, ceil
from typing import Optional
from dataclasses import dataclass, field

color_map = {
    (1, 1): (0, 0, 255),
    (2, 1): (255, 255, 0),
    (1, 2): (0, 255, 0),
    (2, 2): (255, 0, 0)
}

X_UNIT = 5 # Width of a (1 x 1) piece in characters
Y_UNIT = 3 # Height of a (1 x 1) piece in characters

@dataclass
class Piece(object):
    pos: tuple[int, int]
    dim: tuple[int, int]
    pid: Optional[int] = None
    col: tuple[int, int, int] = field(default=None, init=False, compare=False)

    def __post_init__(self):
        if self.dim[0] < 1 or self.dim[1] < 1:
            raise ValueError(f"All dimensions must be at least 1, received {self.dim}")
        self.col = color_map.get(self.dim, None)
        self.str_list = list()

        char_height = self.dim[0] * Y_UNIT # Height of print out of object
        for i in range(char_height):
            num_middle = self.dim[1] * 5 - 2 # Number of characters between left and right edges
            if i == 0: # Handle case we are first line
                line_str = f" {'_' * num_middle} "
            elif i == char_height - 1: # Handle case we are last line
                line_str = f"|{'_' * num_middle}|"
            elif i == char_height // 2 and self.pid is not None: # Handle case we are on middle line
                pid_str = str(self.pid)
                num_left = floor((num_middle - len(pid_str)) / 2)
                num_right = ceil((num_middle - len(pid_str)) / 2)
                line_str = f"|{' ' * (num_left)}{pid_str}{' ' * num_right}|"
            else:
                line_str = f"|{' ' * num_middle}|"
            self.str_list.append(line_str)
        
        self.term_dim = (len(self.str_list), len(self.str_list[0]))