import pygame
from pygame import locals
import os
def fill_zero(a):
    if a<10:
        a="0"+str(a)
    return str(a)

pygame.init()  # 初始化
# 创建窗口
screen = pygame.display.set_mode((640, 480))
# 载入图片、资源
bg_img = pygame.image.load('background.png')  # 背景图
play_img = pygame.image.load('play.png')  # 播放按钮
stop_img = pygame.image.load('stop.png')  # 暂停按钮
last_img = pygame.image.load('last.png')  # 上一曲按钮
next_img = pygame.image.load('next.png')  # 下一曲按钮
logo_img = pygame.image.load('logo.png')  # 下一曲按钮

music_list = []
path = "C:\\Users\\Administrator\\Desktop\\test"
filelist = os.listdir(path)
num = -1
for i in filelist:
    if i[-4:] == ".wav" or i[-4:] == ".ogg":
        music_list.append(i)
volume = 0.2
pygame.mixer.music.set_volume(volume)  # 初始播放音量
click = 0
play_button = stop_img

angle=0

while True:
    for event in pygame.event.get():
        if event.type == locals.QUIT:
            exit()
        # 按键，控制声音大小
        if event.type == locals.KEYDOWN:
            if event.key == locals.K_w:
                volume += 0.1
                if volume > 1:
                    volume = 1
                pygame.mixer.music.set_volume(volume)
            if event.key == locals.K_s:
                volume -= 0.1
                if volume < 0:
                    volume = 0
                pygame.mixer.music.set_volume(volume)

        if event.type == locals.MOUSEBUTTONDOWN:
            if event.button == 1:  # 左击
                x, y = event.pos
                if x > 270 and x < 370 and y > 350 and y < 450: # 播放按钮
                    click += 1
                    if click % 2 == 0:
                        play_button = stop_img
                        pygame.mixer.music.unpause()
                    else:
                        play_button = play_img
                        pygame.mixer.music.pause()
                if x > 420 and x < 520 and y > 350 and y < 400:     # 下一曲
                    num += 1
                    if num > len(music_list) - 1:
                        num = 0
                    pygame.mixer.music.load(path + "\\" + music_list[num])
                    pygame.mixer.music.play()
                    play_button = stop_img
                    music=pygame.mixer.Sound(path + "\\" + music_list[num])
                    music_time=int(music.get_length())
                    if click % 2 == 0:
                        click += 2
                    else:
                        click += 1

                if x > 120 and x < 220 and y > 350 and y < 400:     # 上一曲
                    num -= 1
                    if num < 0:
                        num = len(music_list) - 1
                    pygame.mixer.music.load(path + "\\" + music_list[num])
                    pygame.mixer.music.play()
                    play_button = stop_img
                    music=pygame.mixer.Sound(path + "\\" + music_list[num])
                    music_time=int(music.get_length())
                    if click % 2 == 0:
                        click += 2
                    else:
                        click += 1

    if pygame.mixer.music.get_busy() == False:
        num += 1
        if num > len(music_list)-1:
            num = 0
        pygame.mixer.music.load(path + "\\" + music_list[num])
        pygame.mixer.music.play()
        music=pygame.mixer.Sound(path + "\\" + music_list[num])
        music_time=int(music.get_length())

    if click % 2 == 0:
        angle+=1
    new_logo=pygame.transform.rotate(logo_img,angle)
    rect=new_logo.get_rect(center=(320,200))
    new_rect=(rect[0],rect[1])

    play_time=pygame.mixer.music.get_pos()
    play_time=play_time//1000
    play_m=str(play_time//60)
    play_s=play_time%60
    play_s=fill_zero(play_s)
    music_m=str(music_time//60)
    music_s=music_time%60
    music_s=fill_zero(music_s)
    allpl_time="/"+music_m + ":" + music_s
    haspld_time=play_m + ":" + play_s
    time=pygame.font.Font("neuropol.ttf",25)
    has_played=time.render(haspld_time,True,(255,255,255))
    all_play=time.render(allpl_time,True,(0,0,0))

    # 绘制画面
    screen.blit(bg_img, (0, 0))      # 填充背景
    screen.blit(play_button, (270, 330))   # 暂停按钮
    screen.blit(new_logo, new_rect)    # 中间logo图
    screen.blit(last_img, (120, 350))   # 上一曲
    screen.blit(next_img, (420, 350))   # 下一曲
    screen.blit(has_played, (65,435))
    screen.blit(all_play,(165,435))
    # 刷新画面
    pygame.display.update()