main.py

import pygame
from constants import *
from player import Player
from asteroid import Asteroid
from asteroidfield import AsteroidField
from bullet import Bullet

def main():
    printStartInformations()
    
    pygame.init()    
    screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
    time_clock = pygame.time.Clock()
    delta_time_from_last_tick = 0
    
    bullets, updatable_objects, drawable_objects, asteroids = initaiteGameObjects()    

    player = Player(SCREEN_WIDTH / 2, SCREEN_HEIGHT / 2)
    AsteroidField()

    on = True    
    while(on):
        on = check_for_exit()
        
        screen.fill((0, 0, 0))
        updatable_objects.update(delta_time_from_last_tick)
        
        for asteroid in asteroids:
            if asteroid.check_for_collision(player):
                print("Game over!")
                on = False
            for bullet in bullets:
                if bullet.check_for_collision(asteroid):
                    asteroid.destroy()
                    bullet.destroy()
                
        for drawable_object in drawable_objects:
            drawable_object.draw(screen)

        pygame.display.flip() 
        delta_time_from_last_tick = time_clock.tick(MAX_FRAME_PER_SECOND) / 1000

def initaiteGameObjects():
    updatable_objects = pygame.sprite.Group()
    drawable_objects = pygame.sprite.Group()    
    asteroids = pygame.sprite.Group()    
    bullets = pygame.sprite.Group()
    
    Asteroid.containers = (asteroids, updatable_objects, drawable_objects)
    Player.containers = (updatable_objects, drawable_objects)
    AsteroidField.containers = (updatable_objects)
    Bullet.containers = (bullets, updatable_objects, drawable_objects)

    return bullets, updatable_objects, drawable_objects, asteroids

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

def check_for_exit():
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            return False
    return True                 

if __name__ == "__main__":
    main()