main.py

from colordict import THECOLORS
import pygame
from time import time
from entities import *
from constants import *
import pygame.freetype


def doCollisions():
    if pygame.sprite.spritecollideany(bob, Foe.shots) or pygame.sprite.spritecollide(bob, Foe.foes, False):
        bob.getHit()
    for foe in Foe.foes:
        for shot in pygame.sprite.spritecollide(foe, Player.shots, False):
            foe.hp -= 1
        if foe.hp <= 0:
            if type(foe) is Boss:
                Boss.defeated = True
                Foe.shots = []
            bob.levelUp()
    Player.shots = [shot for shot in Player.shots if not pygame.sprite.spritecollideany(shot, Foe.foes)]
    Foe.foes = [foe for foe in Foe.foes if foe.hp > 0]


def animate():
    screen.fill([200, 200, 255])
    # pygame.draw.line(screen, THECOLORS["red"], (0,0), (300,300), 30)
    for i in range(bob.lives):
        screen.blit(bob.image, [1280 - (30 * i), 30])
    if len(Foe.foes) > 0:
        bob.showLaser(screen)
    for foe in Foe.foes:  # move foes, draw along w/ health bars
        foe.updatePos()
        foe.show(screen)
    for shot in Player.shots + Foe.shots:
        shot.move()
        screen.blit(shot.image, shot.rect)
    if Boss.defeated:
        showMsg('YOU WON!!!')
    screen.blit(bob.image, bob.rect)
    pygame.display.flip()


def removeDeadShots():
    """stop moving shots if they're off the screen or past duration"""
    Player.shots = [shot for shot in Player.shots if not shot.isOffScreen()]
    Foe.shots = [shot for shot in Foe.shots if not shot.isOffScreen() or
                 (type(shot) is FollowShot and time() - shot.timeFired > FollowShot.duration)]


def showMsg(msg):
    font = pygame.freetype.Font(None, 230)
    font.render_to(screen, [20, 100], msg, THECOLORS["white"])


def initGame():
    global screen, clock, bob

    pygame.init()
    width, height = 1300, 680  # sets window size to 1300, 680
    screen = pygame.display.set_mode((width, height))
    clock = pygame.time.Clock()
    screen.fill([170, 170, 250])
    bob = Player()


def start():
    pygame.time.set_timer(FOE_ENTER_EVENT, FOE_INTERVAL)  # sets time between foe entry
    pygame.time.set_timer(FOE_SHOOT_EVENT, 500)
    pygame.time.set_timer(BOB_SHOOT_EVENT, 400)
    Foe.addFoe(bob)
    bob.fire()


if __name__ == "__main__":
    initGame()
    running = True
    started = False

    while running:
        for event in pygame.event.get():
            if event.type == pygame.MOUSEBUTTONUP and not started:
                started = True
                start()
            elif event.type == pygame.QUIT:
                running = False
            elif event.type == pygame.MOUSEMOTION:  # move
                bob.rect.centerx, bob.rect.centery = event.pos
            if started:
                if event.type == INVINC_EVENT:  # bob invincible timer
                    bob.recolor()
                elif event.type == BOB_SHOOT_EVENT:
                    bob.fire()
                elif event.type == FOE_ENTER_EVENT:
                    Foe.addFoe(bob)
                elif event.type == FOE_SHOOT_EVENT:
                    for foe in Foe.foes:
                        foe.fire()

        animate()
        if started:
            doCollisions()
            removeDeadShots()
            if bob.lives <= 0:
                print("you lose")
                running = False
        clock.tick(FRAMERATE)
        # print ("fps = ", str(clock.get_fps()))
    pygame.quit()