o_block.py

from turtle import Turtle
from tetris_block import TetrisBlock

# to indicate the position of individual segment of the block

POSITION_LIST = [(-20, 270), (2, 270), (2, 292), (-20, 292)]


class OBlock(TetrisBlock):
    def __init__(self):
        super().__init__()
        self.create()
        self.set_up()

    def create(self):
        """
        This method is to create segments of OBlock"""
        for position in POSITION_LIST:
            new_segment = Turtle("square")
            new_segment.penup()
            new_segment.color("white")
            new_segment.speed("fastest")
            new_segment.goto(position)
            self.tetris_block_segments.append(new_segment)

    def set_up(self):
        """
        This method used to set the front of the Block"""
        self.front.extend([self.tetris_block_segments[0], self.tetris_block_segments[1]])
        for segment in self.tetris_block_segments:
            segment.setheading(270)