import pygame
import os
from pygame import locals

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\\玛酷机器人\\Desktop\\text"
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

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()
                    music1=pygame.mixer.Sound(path + "\\" + music_list[num])
                    music1_length=music1.get_length()
                    play_button = stop_img
                    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()
                    music1=pygame.mixer.Sound(path + "\\" + music_list[num])
                    music1_length=music1.get_length()
                    play_button = stop_img
                    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()
        music1=pygame.mixer.Sound(path + "\\" + music_list[num])
        music1_length=music1.get_length()


    play_time=pygame.mixer.music.get_pos()
    play_time=int(play_time/1000)
    play_m=play_time//60
    play_s=play_time%60
    if play_s<10:
        play_s="0"+str(play_s)

    time1 = str(play_m)+":"+str(play_s)
    # print(time1)

    music1_length=int(music1_length)
    music1_m=music1_length//60
    music1_s=music1_length%60
    if music1_s<10:
        music1_s="0"+str(music1_s)

    time2 = str(music1_m)+":"+str(music1_s)
    print(time2)

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