import random
import time
import pygame
from pygame.constants import *

# 飞机类
class HeroPlane(object):
    def __init__(self, screen):
        # 2. 创建一个图片，玩家飞机
        self.player = pygame.image.load("./feiji/me1.png")
        # 定义飞机坐标
        self.x = 480 / 2 - 102 / 2
        self.y = 550
        # 飞机速度
        self.speed = 5

        self.screen = screen
        self.bullets = []

    def key_control(self):
        # 监听键盘事件
        key_pressed = pygame.key.get_pressed()

        if key_pressed[K_w] or key_pressed[K_UP]:
            print('上')
            self.y -= self.speed
        if key_pressed[K_s] or key_pressed[K_DOWN]:
            print('下')
            self.y += self.speed
        if key_pressed[K_a] or key_pressed[K_LEFT]:
            print('左')
            self.x -= self.speed
        if key_pressed[K_d] or key_pressed[K_RIGHT]:
            print('右')
            self.x += self.speed
        if key_pressed[K_SPACE]:
            # 按下空格键发射自子弹
            bullet = Bullet(self.screen, self.x, self.y)
            # 把子弹放到列表里
            self.bullets.append(bullet)

    def display(self):
        # 5.将背景图片贴到窗口中
        self.screen.blit(self.player, (self.x, self.y))
        # 遍历所有子弹
        for bullet in self.bullets:
            # 让子弹飞 修改子弹y坐标
            bullet.auto_move()
            # 子弹显示在窗口
            bullet.display()


# 敌方飞机类
class EnemyPlane(object):
    def __init__(self, screen):
        # 2. 创建一个图片，玩家飞机
        self.player = pygame.image.load("./feiji/enemy1.png") # 57 43
        # 定义飞机坐标
        self.x = 0
        self.y = 0
        # 飞机速度
        self.speed = 4

        self.screen = screen
        self.bullets = []

        # 敌机移动方向
        self.direction = 'right'

    def display(self):
        # 5.将背景图片贴到窗口中
        self.screen.blit(self.player, (self.x, self.y))
        # 敌机自动移动
        self.auto_move()
        # 遍历所有子弹并自动移动
        for enemyBullet in self.bullets:
            enemyBullet.auto_move()
            enemyBullet.display()

    def auto_move(self):
       if self.direction == 'right':
           self.x += self.speed
       if self.direction == 'left':
           self.x -= self.speed

       if self.x >= 480 - 57:
           self.direction = 'left'
       if self.x <= 0:
           self.direction = 'right'

    def auto_fire(self):
        # 敌机子弹加入列表
        random_num = random.randint(1,30)
        if random_num == 7:
            enemyBullet = EnemyBullet(self.screen, self.x, self.y)
            self.bullets.append(enemyBullet)


# 子弹类
# 属性
class Bullet(object):
    def __init__(self, screen, x, y):
        # 坐标
        self.x = x + 102 / 2 - 5 / 2 + 1
        self.y = y - 11
        # 图片
        self.image = pygame.image.load('./feiji/bullet2.png')
        # 窗口
        self.screen = screen
        # 速度
        self.speed = 5
    def display(self):
        """显示子弹到窗口"""
        self.screen.blit(self.image, (self.x, self.y))
    def auto_move(self):
        """让子弹飞 修改子弹坐标"""
        self.y -= self.speed
# 敌方子弹类
# 属性
class EnemyBullet(object):
    def __init__(self, screen, x, y):
        # 坐标
        self.x = x + 57 / 2 - 5 / 2
        self.y = 43
        # 图片
        self.image = pygame.image.load('./feiji/bullet1.png') # 5 * 11
        # 窗口
        self.screen = screen
        # 速度
        self.speed = 5
    def display(self):
        """显示子弹到窗口"""
        self.screen.blit(self.image, (self.x, self.y))
    def auto_move(self):
        """让子弹飞 修改子弹坐标"""
        self.y += self.speed

class GameSound(object):
    def __init__(self):
        pygame.mixer.init() # 音乐模块初始化
        pygame.mixer.music.load("./feiji/game_music.wav")
        pygame.mixer.music.set_volume(0.5) # 声音大小
    def playBackgroundMusic(self):
        pygame.mixer.music.play(-1) # 开始播放音乐

def main():
    """完成整个程序的控制"""
    sound = GameSound()
    sound.playBackgroundMusic()

    # 1.创建一个窗口
    screen = pygame.display.set_mode((480, 700), 0, 32)
    # 2.创建一个图片，当做背景
    background = pygame.image.load("./feiji/background.png")
    # 3.创建飞机对象
    player = HeroPlane(screen)
    # 创建一个地方飞机的对象，注意不要忘记传窗口
    enemyPlane = EnemyPlane(screen)

    while True:
        # 3.将背景图片贴到窗口中
        screen.blit(background, (0, 0))

        # 获取事件
        for event in pygame.event.get():
            # 判断事件类型
            if event.type == QUIT:
                # 执行pygame退出
                pygame.quit()
                # python程序退出
                exit()

        # 执行飞机的按键监听
        player.key_control()
        # 飞机的显示
        player.display()
        # 敌方飞机的显示
        enemyPlane.display()
        # 敌方飞机自动开火
        enemyPlane.auto_fire()

        # 4.更新需要显示的内容
        pygame.display.update()
        time.sleep(0.01)

if __name__ == '__main__':
    main()