powerup.py

from turtle import RawTurtle

from constants import PowerupType, SOUTH, PowerupAttributes


class Powerup(RawTurtle):
    def __init__(self, screen, powerup_type: PowerupType, location, **kwargs):
        super().__init__(screen, **kwargs)

        self.type = powerup_type
        self.image = None

        self.set_default_powerup(location)

    def set_default_powerup(self, location):
        self.penup()
        self.shape(PowerupAttributes.SHAPE)
        self.color(PowerupAttributes.COLOR)
        self.setheading(SOUTH)
        self.setposition(location)
        self.hideturtle()

    def set_image(self, canvas_image):
        self.image = canvas_image

    def get_image(self):
        return self.image

    def move(self):
        self.forward(PowerupAttributes.SPEED)

    def remove(self):
        self.hideturtle()

    def get_type(self):
        return self.type

    def get_location(self):
        return self.xcor(), self.ycor()