main.py

import pygame
import sys
from constants import *
from player import Player
from asteroid import Asteroid
from asteroidfield import AsteroidField
from shot import Shot

def main():
    pygame.init()
    print("Starting asteroids!")
    print("Screen width:", SCREEN_WIDTH)
    print("Screen height:", SCREEN_HEIGHT)

    # screen and clock setup
    screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
    clock = pygame.time.Clock()

    # groups
    updatable = pygame.sprite.Group()
    drawable = pygame.sprite.Group()
    asteroids = pygame.sprite.Group()
    shots = pygame.sprite.Group()

    # asteroid setup
    Asteroid.containers = (asteroids, updatable, drawable)
    AsteroidField.containers = updatable
    asteroid_field = AsteroidField()

    # player setup
    Player.containers = (updatable, drawable)
    player = Player(SCREEN_WIDTH / 2, SCREEN_HEIGHT / 2) # center of the screen, x and y coordinates

    # shots setup
    Shot.containers = (shots, updatable, drawable)

    dt = 0
    # gameloop
    while True:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                return
        screen.fill("black")
    
        # handle updates
        for thing in updatable:
            thing.update(dt)

        # handle collisions
        for asteroid in asteroids:
            if player.collision(asteroid):
                    print("Game over!")
                    sys.exit()
            
            # handle shots
            for shot in shots:
                if shot.collision(asteroid):
                    shot.kill()
                    asteroid.split()

        # handle drawing
        for thing in drawable:
            thing.draw(screen)

        


        pygame.display.flip()

        dt  = pygame.time.Clock().tick(60) / 1000

if __name__ == "__main__":
    main()