Tidal-Media-Downloader/TIDALDL-PY/tidal_gui/viewModel/taskItemModel.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
"""
@File    :  taskItemModel.py
@Date    :  2021/9/14
@Author  :  Yaronzz
@Version :  1.0
@Contact :  yaronhuang@foxmail.com
@Desc    :
"""
import _thread
from asyncio import tasks
from enum import Enum
import os
import time

import aigpy.stringHelper

from tidal_dl import Type
from tidal_dl.model import Album, Track, Video, Playlist
from tidal_dl.util import API, getArtistsNames, getBasePath, getDurationString
from tidal_gui.view.taskItemView import TaskItemView
from tidal_gui.view.taskView import TaskStatus
from tidal_gui.viewModel.downloadItemModel import DownloadItemModel, DownloadStatus
from tidal_gui.viewModel.viewModel import ViewModel


class TaskItemModel(ViewModel):
    def __init__(self, data):
        super(TaskItemModel, self).__init__()
        self.view = TaskItemView()
        self.data = data
        self.downloadModelList = []
        self.path = ''

        if isinstance(data, Album):
            self.__initAlbum__(data)
        elif isinstance(data, Track):
            self.__initTrack__(data)
        elif isinstance(data, Video):
            self.__initVideo__(data)
        elif isinstance(data, Playlist):
            self.__initPlaylist__(data)

        self.view.connectButton('retry', self.__btnFuncRetry__)
        self.view.connectButton('cancel', self.__btnFuncCancel__)
        self.view.connectButton('delete', self.__btnFuncDelete__)
        self.view.connectButton('open', self.__btnFuncOpen__)

        self.SIGNAL_REFRESH_VIEW.connect(self.__refresh__)

    def getTaskStatus(self) -> TaskStatus:
        if len(self.downloadModelList) <= 0:
            return TaskStatus.Download
        
        errorNum = 0
        for item in self.downloadModelList:
            if item.status in [DownloadStatus.WAIT, DownloadStatus.RUNNING, DownloadStatus.CANCEL]:
                return TaskStatus.Download
            elif item.status == DownloadStatus.ERROR:
                errorNum += 1
        
        if errorNum > 0:
            return TaskStatus.Error
        return TaskStatus.Complete

    def __refresh__(self, stype: str, obj):
        if stype == "setPic":
            self.view.setPic(obj)
        elif stype == "addListItems":
            for index, item in enumerate(obj):
                downItem = DownloadItemModel(index + 1, item, self.path)
                self.view.addListItem(downItem.view)
                self.downloadModelList.append(downItem)

    def __btnFuncRetry__(self):
        for item in self.downloadModelList:
            item.retry()

    def __btnFuncCancel__(self):
        for item in self.downloadModelList:
            item.stopDownload()

    def __btnFuncDelete__(self):
        for item in self.downloadModelList:
            item.stopDownload()

    def __btnFuncOpen__(self):
        if self.path == '':
            return
        if os.path.exists(self.path):
            os.startfile(self.path)

    def __initAlbum__(self, data: Album):
        self.path = getBasePath(data)

        title = data.title
        desc = f"by {getArtistsNames(data.artists)} " \
               f"{getDurationString(data.duration)} " \
               f"Track-{data.numberOfTracks} " \
               f"Video-{data.numberOfVideos}"
        self.view.setLabel(title, desc)

        def __thread_func__(model: TaskItemModel, album: Album):
            cover = API.getCoverData(album.cover)
            model.SIGNAL_REFRESH_VIEW.emit('setPic', cover)

            msg, tracks, videos = API.getItems(album.id, Type.Album)
            if not aigpy.stringHelper.isNull(msg):
                model.view.setErrmsg(msg)
                return

            for item in tracks:
                item.album = album
            for item in videos:
                item.album = album

            model.SIGNAL_REFRESH_VIEW.emit('addListItems', tracks + videos)
            time.sleep(1)

        _thread.start_new_thread(__thread_func__, (self, data))

    def __initTrack__(self, data: Track):
        title = data.title
        desc = f"by {getArtistsNames(data.artists)} " \
               f"{getDurationString(data.duration)} "
        self.view.setLabel(title, desc)

        def __thread_func__(model: TaskItemModel, track: Track):
            mag, track.album = API.getAlbum(track.album.id)
            model.path = getBasePath(track)
            cover = API.getCoverData(track.album.cover)
            model.SIGNAL_REFRESH_VIEW.emit('setPic', cover)
            model.SIGNAL_REFRESH_VIEW.emit('addListItems', [track])
            time.sleep(1)

        _thread.start_new_thread(__thread_func__, (self, data))

    def __initVideo__(self, data: Video):
        self.path = getBasePath(data)

        title = data.title
        desc = f"by {getArtistsNames(data.artists)} " \
               f"{getDurationString(data.duration)} "
        self.view.setLabel(title, desc)

        def __thread_func__(model: TaskItemModel, video: Video):
            cover = API.getCoverData(video.imageID)
            model.SIGNAL_REFRESH_VIEW.emit('setPic', cover)
            model.SIGNAL_REFRESH_VIEW.emit('addListItems',  [video])
            time.sleep(1)

        _thread.start_new_thread(__thread_func__, (self, data))

    def __initPlaylist__(self, data: Playlist):
        self.path = getBasePath(data)

        title = data.title
        desc = f"{getDurationString(data.duration)} " \
               f"Track-{data.numberOfTracks} " \
               f"Video-{data.numberOfVideos}"
        self.view.setLabel(title, desc)

        def __thread_func__(model: TaskItemModel, playlist: Playlist):
            cover = API.getCoverData(playlist.squareImage)
            model.SIGNAL_REFRESH_VIEW.emit('setPic', cover)

            msg, tracks, videos = API.getItems(playlist.uuid, Type.Playlist)
            if not aigpy.stringHelper.isNull(msg):
                model.view.setErrmsg(msg)
                return

            for item in tracks:
                mag, album = API.getAlbum(item.album.id)
                item.playlist = playlist
                item.album = album
            for item in videos:
                item.playlist = playlist

            model.SIGNAL_REFRESH_VIEW.emit('addListItems', tracks + videos)
            time.sleep(1)

        _thread.start_new_thread(__thread_func__, (self, data))