#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Created on 2023/02/22
@author: Irony
@site: https://pyqt.site https://github.com/PyQt5
@email: 892768447@qq.com
@file: HotPlaylist.py
@description:
"""

import os
import sys

from Lib.CoverItemWidget import CoverItemWidget
from lxml.etree import HTML  # @UnresolvedImport

try:
    from PyQt5.QtCore import Qt, QTimer, QUrl, pyqtSignal
    from PyQt5.QtNetwork import QNetworkAccessManager, QNetworkRequest
    from PyQt5.QtSvg import QSvgWidget
    from PyQt5.QtWidgets import (QAbstractSlider, QApplication, QGridLayout,
                                 QScrollArea, QWidget)
except ImportError:
    from PySide2.QtCore import Qt, QTimer, QUrl, pyqtSignal
    from PySide2.QtNetwork import QNetworkAccessManager, QNetworkRequest
    from PySide2.QtSvg import QSvgWidget
    from PySide2.QtWidgets import (QAbstractSlider, QApplication, QGridLayout,
                                   QScrollArea, QWidget)

# offset=0,35,70,105
Url = "https://music.163.com/discover/playlist/?order=hot&cat=%E5%85%A8%E9%83%A8&limit=35&offset={0}"

Agent = b"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/110.0.0.0 Safari/537.36 Edg/110.0.1587.50"

Referer = b"https://music.163.com"

# 作者
Actor = '''<a href="{href}" target="_blank" title="{title}" style="text-decoration: none;font-size: 12px;color: #999999;">{title}</a>&nbsp;'''


class GridWidget(QWidget):
    Page = 0
    loadStarted = pyqtSignal(bool)

    def __init__(self, *args, **kwargs):
        super(GridWidget, self).__init__(*args, **kwargs)
        self._layout = QGridLayout(self, spacing=20)
        self._layout.setContentsMargins(20, 20, 20, 20)
        # 异步网络下载管理器
        self._manager = QNetworkAccessManager(self)
        self._manager.finished.connect(self.onFinished)

    def load(self):
        if self.Page == -1 or self.Page > 10:
            return
        self.loadStarted.emit(True)
        # 延迟一秒后调用目的在于显示进度条
        QTimer.singleShot(1000, self._load)

    def _load(self):
        print("load url:", Url.format(self.Page * 35))
        url = QUrl(Url.format(self.Page * 35))
        req = QNetworkRequest(url)
        req.setRawHeader(b"User-Agent", Agent)
        self._manager.get(req)

    def onFinished(self, reply):
        # 请求完成后会调用该函数
        req = reply.request()  # 获取请求
        iwidget = req.attribute(QNetworkRequest.User + 1, None)
        path = req.attribute(QNetworkRequest.User + 2, None)
        html = reply.readAll().data()
        reply.deleteLater()
        del reply
        if iwidget and path and html:
            # 这里是图片下载完毕
            open(path, "wb").write(html)
            iwidget.setCover(path)
            return
        # 解析网页
        self._parseHtml(html)
        self.loadStarted.emit(False)

    def splist(self, src, length):
        # 等分列表
        return (src[i:i + length] for i in range(len(src)) if i % length == 0)

    def _parseHtml(self, html):
        # print(html)
        #         encoding = chardet.detect(html) or {}
        #         html = html.decode(encoding.get("encoding","utf-8"))
        html = HTML(html)
        # 查找所有的li list_item
        lis = html.xpath("//ul[@id='m-pl-container']/li")
        # print(lis)
        if not lis:
            self.Page = -1  # 后面没有页面了
            return
        lack_count = self._layout.count() % 35  # 获取布局中上次还缺几个5行*6列的标准
        row_count = int(self._layout.count() / 5)  # 行数
        print("lack_count:", lack_count)
        self.Page += 1  # 自增+1
        if lack_count != 0:  # 上一次没有满足一行5个,需要补齐
            lack_li = lis[:lack_count]
            lis = lis[lack_count:]
            self._makeItem(lack_li, row_count)  # 补齐
            if lack_li and lis:
                row_count += 1
                self._makeItem(lis, row_count)  # 完成剩下的
        else:
            self._makeItem(lis, row_count)

    def _makeItem(self, li_s, row_count):
        li_s = self.splist(li_s, 5)
        for row, lis in enumerate(li_s):
            for col, li in enumerate(lis):
                a = li.find('.//div/a')
                play_url = "https://music.163.com" + a.get("href")  # 歌单播放地址
                img = li.find(".//div/img")
                cover_url = img.get("src")  # 封面图片
                playlist_title = a.get("title")  # 歌单名
                # 歌手
                author_info = li.xpath(".//p[2]/a")[0]
                playlist_author = "<span style=\"font-size: 12px;\"作者：{}</span>".format(
                    Actor.format(href="https://music.163.com" +
                                 author_info.get("href"),
                                 title=author_info.get("title")))
                # 播放数
                play_count = (li.xpath(".//div/div/span[2]/text()") or [""])[0]
                path = "cache/{0}.jpg".format(
                    os.path.splitext(
                        os.path.basename(cover_url).split('?')[0])[0])
                cover_path = "Data/pic_v.png"
                if os.path.isfile(path):
                    cover_path = path

                # print(cover_path, playlist_title,
                #       playlist_author, play_count, play_url, cover_url, path)
                iwidget = CoverItemWidget(self, manager=self._manager)
                iwidget.init(cover_path, playlist_title, playlist_author,
                             play_count, play_url, cover_url, path)
                self._layout.addWidget(iwidget, row_count + row, col)


class Window(QScrollArea):

    def __init__(self, *args, **kwargs):
        super(Window, self).__init__(*args, **kwargs)
        self.resize(800, 600)
        self.setFrameShape(self.NoFrame)
        self.setWidgetResizable(True)
        self.setAlignment(Qt.AlignCenter)
        self._loadStart = False
        # 网格窗口
        self._widget = GridWidget(self)
        self._widget.loadStarted.connect(self.setLoadStarted)
        self.setWidget(self._widget)
        # 连接竖着的滚动条滚动事件
        self.verticalScrollBar().actionTriggered.connect(self.onActionTriggered)
        # 进度条
        self.loadWidget = QSvgWidget(self,
                                     minimumHeight=120,
                                     minimumWidth=120,
                                     visible=False)
        self.loadWidget.load('Data/Svg_icon_loading.svg')

    def setLoadStarted(self, started):
        self._loadStart = started
        self.loadWidget.setVisible(started)

    def onActionTriggered(self, action):
        # 这里要判断action=QAbstractSlider.SliderMove，可以避免窗口大小改变的问题
        # 同时防止多次加载同一个url
        if action != QAbstractSlider.SliderMove or self._loadStart:
            return
        # 使用sliderPosition获取值可以同时满足鼠标滑动和拖动判断
        if self.verticalScrollBar().sliderPosition() == self.verticalScrollBar(
        ).maximum():
            # 可以下一页了
            self._widget.load()

    def resizeEvent(self, event):
        super(Window, self).resizeEvent(event)
        self.loadWidget.setGeometry(
            int((self.width() - self.loadWidget.minimumWidth()) / 2),
            int((self.height() - self.loadWidget.minimumHeight()) / 2),
            self.loadWidget.minimumWidth(), self.loadWidget.minimumHeight())


if __name__ == "__main__":
    os.makedirs("cache", exist_ok=True)
    app = QApplication(sys.argv)
    w = Window()
    w.show()
    w._widget.load()
    sys.exit(app.exec_())