s_block.py

from turtle import Turtle
from tetris_block import TetrisBlock

# to indicating the correct position of this block segment
POSITION_LIST = [(-20, 270), (-20, 292), (2, 292), (2, 314)]


class SBlock(TetrisBlock):
    
    def __init__(self):
        super().__init__()
        self.create()
        self.set_up()
        self.rotate_index = 0
        self.rotate_positions = [
            [[-22, 22], [0, 22], [0, 0], [22, 0]],
            [[-22, -22], [-22, 0], [0, 0], [0, 22]]
        ]

    def create(self):
        """
        This method is to create segments of SBlock"""
        for position in POSITION_LIST:
            new_segment = Turtle('square')
            new_segment.penup()
            new_segment.color('blue')
            new_segment.speed("fastest")
            new_segment.goto(position)
            self.tetris_block_segments.append(new_segment)
        
        # self.tetris_block_segments[1].color('red')
        # self.tetris_block_segments[2].color('green')
        # self.tetris_block_segments[3].color('yellow')
            
    def set_up(self):
        """
        This method is to assign left, head, front, right segment of LBlock"""
        self.front.extend([self.tetris_block_segments[2], self.tetris_block_segments[0]])
        self.left.extend([self.tetris_block_segments[0], self.tetris_block_segments[1], self.tetris_block_segments[3]])
        self.right.extend([self.tetris_block_segments[0], self.tetris_block_segments[2], self.tetris_block_segments[3]])
        self.head.extend([self.tetris_block_segments[3], self.tetris_block_segments[1]])
        self.center = self.tetris_block_segments[2]
        for segment in self.tetris_block_segments:
            segment.setheading(270)