main.py

# this allows us to use code from
# the open-source pygame library
# throughout this file
import sys
import pygame

# import constants
from constants import *

from player import Player
from asteroid import Asteroid
from asteroidfield import AsteroidField
from shot import Shot

def main():
    # initialize pygame
    pygame.init()

    # clock and timer
    clock = pygame.time.Clock()
    dt = 0

    # setup screen
    screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))

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

    # containers
    Player.containers = (updatable, drawable)
    Asteroid.containers = (asteroids, updatable, drawable)
    AsteroidField.containers = (updatable)
    Shot.containers = (shots, updatable, drawable)

    # objects
    player = Player(SCREEN_WIDTH / 2, SCREEN_HEIGHT / 2)
    asteroidfield = AsteroidField()

    # game loop
    while True:
      # catch and handle exit
      for event in pygame.event.get():
          if event.type == pygame.QUIT:
              return

      # sprite logic
      for sprite in updatable:
          sprite.update(dt)
      
      # asteroid logic
      for asteroid in asteroids:
          if (asteroid.colliding_with(player)):
              print("Game over!")
              sys.exit(0)
          
          for shot in shots:
              if (asteroid.colliding_with(shot)):
                  shot.kill()
                  asteroid.split()

      # draw
      screen.fill("black")
      
      # sprite drawing
      for sprite in drawable:
          sprite.draw(screen)

      # screen refresh
      pygame.display.flip()

      # tick
      dt = clock.tick(60) / 1000

    print("Starting Asteroids!")
    print(f"Screen width: {SCREEN_WIDTH}")
    print(f"Screen height: {SCREEN_HEIGHT}")

if __name__ == "__main__":
    main()