player.py

from circleshape import *
from constants import *
from bullet import *

class Player(CircleShape):
    def __init__(self, x, y, shots_group):
        super().__init__(x, y, PLAYER_RADIUS)
        self.rotation = 0
        self.shots = shots_group
        self.timer = 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
        b = self.position - forward * self.radius - right
        c = self.position - forward * self.radius + right
        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 update(self, dt):
        self.timer -= dt
        forward = pygame.Vector2(0, 1).rotate(self.rotation)
        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.position += forward * PLAYER_SPEED * dt

        if keys[pygame.K_s]:
            self.position -= forward * PLAYER_SPEED * dt

        if keys[pygame.K_SPACE]:
            if self.timer <= 0:
                self.shoot()

    def shoot(self):
        self.timer = PLAYER_SHOOT_COOLDOWN 
        bullet = Shot(self.position[0], self.position[1])
        bullet.velocity = pygame.Vector2(0, 1)
        bullet.velocity = pygame.math.Vector2.rotate(bullet.velocity, self.rotation)
        bullet.velocity *= PLAYER_SHOOT_SPEED
        self.shots.add(bullet)