player.py

from constants import *
import circleshape
import pygame
from shot import Shot

class Player(circleshape.CircleShape):
    def __init__(self, x, y):
        super().__init__(x, y, PLAYER_RADIUS)
        self.rotation = 0
        self.cooldown = 0
# in the player class

    def triangle(self):
        forward = pygame.Vector2(0, 1).rotate(self.rotation)
        right = pygame.Vector2(0, 1).rotate(
            self.rotation + 90) * self.radius / 1.5
        a = self.position + forward * self.radius  # pyright: ignore
        b = self.position - forward * self.radius - right  # pyright: ignore
        c = self.position - forward * self.radius + right  # pyright: ignore
        return [a, b, c]

    def draw(self, screen):
        pygame.draw.polygon(screen, "white", self.triangle(), 2)

    def rotate(self, dt):
        self.rotation += PLAYER_TURN_SPEED * dt

    def move(self, dt):
        fw = pygame.Vector2(0, 1).rotate(self.rotation)
        self.position += fw * PLAYER_SPEED * dt
    def shoot(self,dt):
        if self.cooldown <= 0:
            shot = Shot(self.position.x,self.position.y,SHOT_RADIUS)
            shot.velocity = pygame.Vector2(0,1).rotate(self.rotation) * PLAYER_SHOT_SPEED
            self.cooldown = PLAYER_SHOOT_COOLDOWN
            return shot

    def update(self, dt):
        self.cooldown -= dt
        keys = pygame.key.get_pressed()
        if keys[pygame.K_a]:
            self.rotate(-dt)
        if keys[pygame.K_d]:
            self.rotate(dt)
        if keys[pygame.K_w]:
            self.move(dt)
        if keys[pygame.K_s]:
            self.move(-dt)
        if keys[pygame.K_SPACE]:
            self.shoot(dt)