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

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
"""
@File    :  searchModel.py
@Date    :  2021/8/17
@Author  :  Yaronzz
@Version :  1.0
@Contact :  yaronhuang@foxmail.com
@Desc    :
"""
import _thread
import threading

import aigpy.stringHelper
from PyQt5.QtCore import pyqtSignal
from aigpy.modelHelper import ModelBase

import tidal_dl
from tidal_dl import Type
from tidal_dl.util import API, getAudioQualityList, getCurAudioQuality, getCurVideoQuality, getVideoQualityList, setCurAudioQuality, setCurVideoQuality
from tidal_gui.view.searchView import SearchView
from tidal_gui.viewModel.viewModel import ViewModel


class SearchModel(ViewModel):
    SIGNAL_ADD_TASKITEM = pyqtSignal(ModelBase)

    def __init__(self):
        super(SearchModel, self).__init__()
        self._lock = threading.Lock()
        self._resultData = None

        self.view = SearchView()
        self.view.setPageIndex(1, 1)
        self.view.setTrackQualityItems(getAudioQualityList(), getCurAudioQuality())
        self.view.setVideoQualityItems(getVideoQualityList(), getCurVideoQuality())
        self.view.connectButton('search', self.__search__)
        self.view.connectButton('prePage', self.__searchPre__)
        self.view.connectButton('nextPage', self.__searchNext__)
        self.view.connectButton('download', self.__download__)
        self.view.connectQualityComboBox('track', self.__changeAudioQuality__)
        self.view.connectQualityComboBox('video', self.__changeVideoQuality__)
        self.view.connectTab(lambda: self.__search__(0))
        self.SIGNAL_REFRESH_VIEW.connect(self.__refresh__)

    def __getSumByResult__(self, stype: Type):
        if stype == Type.Album:
            return self._resultData.albums.totalNumberOfItems
        elif stype == Type.Artist:
            return self._resultData.artists.totalNumberOfItems
        elif stype == Type.Track:
            return self._resultData.tracks.totalNumberOfItems
        elif stype == Type.Video:
            return self._resultData.videos.totalNumberOfItems
        elif stype == Type.Playlist:
            return self._resultData.playlists.totalNumberOfItems
        return 0

    def __refresh__(self, stype: str, data):
        if stype == 'setPageIndex':
            self.view.setPageIndex(data[0], data[1])
        elif stype == 'setTableItems':
            self.view.setTableItems(data[0], data[1], data[2])
        elif stype == 'setSearchErrmsg':
            self.view.setSearchErrmsg(data)

    def __startThread__(self, index: int):
        def __thread_search__(model: SearchModel, index: int):
            typeIndex = model.view.getSelectedTabIndex()
            searchText = model.view.getSearchText()

            if aigpy.stringHelper.isNull(searchText):
                model._lock.release()
                return

            # search
            limit = 20
            offset = (index - 1) * limit
            stype = tidal_dl.Type(typeIndex)
            msg, model._resultData = API.search(searchText, stype, offset, limit)

            if not aigpy.stringHelper.isNull(msg):
                model.SIGNAL_REFRESH_VIEW.emit('setSearchErrmsg', msg)
                model._lock.release()
                return

            # get page index
            total = model.__getSumByResult__(stype)
            if total <= 0:
                model.SIGNAL_REFRESH_VIEW.emit('setSearchErrmsg', 'Search results are empty...')
                model._lock.release()
                return

            maxIdx = total // limit + (1 if total % limit > 0 else 0)
            if index > maxIdx:
                model._lock.release()
                return

            # set view
            model.SIGNAL_REFRESH_VIEW.emit('setPageIndex', (index, maxIdx))
            model.SIGNAL_REFRESH_VIEW.emit('setTableItems', (stype, offset, model._resultData))
            model._lock.release()

        _thread.start_new_thread(__thread_search__, (self, index))

    def __search__(self, num: int = 0):
        if not self._lock.acquire(False):
            return

        self.view.setSearchErrmsg('')

        if aigpy.string.isNull(self.view.getSearchText()):
            self._lock.release()
            return

        index = 1
        if num != 0:
            curIdx, curSum = self.view.getPageIndex()
            if curIdx + num > 1:
                index = curIdx + num

        self.__startThread__(index)

    def __searchNext__(self):
        self.__search__(1)

    def __searchPre__(self):
        self.__search__(-1)

    def __download__(self):
        if self._resultData is None:
            self.view.setSearchErrmsg('Please search first...')
            return

        typeIndex = self.view.getSelectedTabIndex()
        stype = tidal_dl.Type(typeIndex)

        items = []
        if stype == Type.Album:
            items = self._resultData.albums.items
        elif stype == Type.Track:
            items = self._resultData.tracks.items
        elif stype == Type.Video:
            items = self._resultData.videos.items
        elif stype == Type.Playlist:
            items = self._resultData.playlists.items

        index = self.view.getSelectedTableIndex(stype)
        if index < 0 or index >= len(items):
            self.view.setSearchErrmsg('Please select one item...')
            return

        self.SIGNAL_ADD_TASKITEM.emit(items[index])

    def __changeAudioQuality__(self):
        setCurAudioQuality(self.view.getTrackQualityText())
    
    def __changeVideoQuality__(self):
        setCurVideoQuality(self.view.getVideoQualityText())