python 游戏- pygame space invader

# pygame install: windows: cmd: pip install pygame
import pygame
from os import path
import random
import math

# init the pygame
pygame.init()

# create the screen
screen = pygame.display.set_mode((800, 600))

# title and icon
pygame.display.set_caption("Space Invader")
game_folder = path.dirname('C:\\Users\\ZZJ\\img')
img_dir = path.join(game_folder, 'img')
icon = pygame.image.load(path.join(img_dir, 'playerShip1_orange.png'))
pygame.display.set_icon(icon)

# background
background = pygame.image.load(path.join(img_dir, 'blue.png')).convert()
background = pygame.transform.scale(background, (800, 600)) # change the size
background_rect = background.get_rect()  

# player
playerImg = pygame.image.load(path.join(img_dir, 'playerShip1_orange.png'))
playerX = 370
playerY = 480
playerX_change = 0

# enemy
enemyImg = []
enemyX = []
enemyY = []
enemyX_change = []
enemyY_change = []
num_of_enemies = 6

for i in range(num_of_enemies):
    enemyImg.append(pygame.image.load(path.join(img_dir, 'meteorBrown_med1.png')))
    enemyX.append(random.randint(0, 735))
    enemyY.append(random.randint(50, 300))
    enemyX_change.append(4)
    enemyY_change.append(40)

# bullet
# ready: you cannot see bullet
# fire: you can see bullet
bulletImg = pygame.image.load(path.join(img_dir, 'laserRed16.png'))
bulletX = 0
bulletY = 480
bulletX_change = 0
bulletY_change = 5
bullet_state = 'ready'

score_value = 0
font_name = pygame.font.match_font('arial')
def draw_text(surf, text, size, x, y):
    text = "score" + ': ' + text
    font = pygame.font.Font(font_name, size)
    text_surface = font.render(text, True, (255, 255, 255))
    text_rect = text_surface.get_rect()
    text_rect.midtop = (x, y)
    surf.blit(text_surface, text_rect)

# player function
def player(x, y):
    screen.blit(playerImg, (x, y))

# bullet function
def fire_bullet(x, y):
    global bullet_state
    bullet_state = 'fire'
    screen.blit(bulletImg, (x + 36, y + 10))

    # enemy function
def enemy(x, y, i):
    screen.blit(enemyImg[i], (x, y))
    
def isCollision(enemyX, enemyY, bulletX, bulletY):
    distance = math.sqrt(math.pow(enemyX - bulletX, 2) + math.pow(enemyY - bulletY, 2))
    if distance < 27:
        return True
    else:
        return False
    
# game loop
running = True
while running:
    # screen
    screen.fill((0, 0, 0))
    # background
    screen.blit(background, (0, 0))
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
    # if key press
    if event.type == pygame.KEYDOWN:
        if event.key == pygame.K_LEFT:
            playerX_change = -0.8
        if event.key == pygame.K_RIGHT:
            playerX_change = 0.8
        if event.key == pygame.K_SPACE:
            if bullet_state is 'ready':
                # get the current x axis
                bulletX = playerX
                fire_bullet(bulletX, bulletY)
    if event.type == pygame.KEYUP:
        if event.key == pygame.K_RIGHT or event.key == pygame.K_LEFT:
            playerX_change = 0
    # RGB = red, green, blue
    playerX += playerX_change
    
    # player boundary
    if playerX <= 0:
        playerX = 0
    elif playerX >= 736:
        playerX = 736
        
    # enemy boundary
    enemyX += enemyX_change
    for i in range(num_of_enemies):
        enemyX[i] += enemyX_change[i]
        if enemyX[i] <= 0:
            enemyX_change[i] = 4
            enemyY[i] += enemyY_change[i]
        elif enemyX[i] >= 736:
            enemyX_change[i] = -4
            enemyY[i] += enemyY_change[i]
        # collision function
        collision = isCollision(enemyX[i], enemyY[i], bulletX, bulletY)
        if collision:
            bulletY = 480
            bullet_state = 'ready'
            score_value += 1
            enemyX[i] = random.randint(0, 735)
            enemyY[i] = random.randint(50, 300)
            
        enemy(enemyX[i], enemyY[i], i)
    # bullet movement
    if bulletY <= 0:
        bulletY = 480
        bullet_state = 'ready'
    if bullet_state is 'fire':
        fire_bullet(bulletX, bulletY)
        bulletY -= bulletY_change
    
        
    player(playerX, playerY)
    draw_text(screen, str(score_value), 18, 800 / 2, 10)
    pygame.display.update() 
    
pygame.quit()