asteroid.py

import random
import pygame

from constants import ASTEROID_MIN_RADIUS
from circleshape import CircleShape

class Asteroid(CircleShape):
    def __init__(self, x, y, radius):
        super().__init__(x, y, radius)
    
    def draw(self, screen):
        pygame.draw.circle(
            screen,
            "white",
            self.position,
            self.radius,
            2
        )
    
    def split(self):
        self.kill()

        if self.radius < ASTEROID_MIN_RADIUS:
            return
        
        new_radius = self.radius - ASTEROID_MIN_RADIUS
        angle = random.uniform(20, 50)

        spawn_1 = Asteroid(self.position.x, self.position.y, new_radius)
        spawn_1.velocity = self.velocity.rotate(angle) * 1.2

        spawn_2 = Asteroid(self.position.x, self.position.y, new_radius)
        spawn_2.velocity = self.velocity.rotate(-angle) * 1.2

    def update(self, dt):
        self.position += self.velocity * dt