mirror of https://github.com/vitiko98/qobuz-dl
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
import time
|
|
import sys
|
|
from pick import pick
|
|
|
|
|
|
class Search:
|
|
def __init__(self, Qz, query, limit=10):
|
|
self.Total = []
|
|
self.IDs = []
|
|
self.Types = []
|
|
self.Tracks = Qz.search_tracks(query, limit)['tracks']['items']
|
|
self.Albums = Qz.search_albums(query, limit)['albums']['items']
|
|
|
|
def seconds(self, duration):
|
|
return time.strftime("%M:%S", time.gmtime(duration))
|
|
|
|
def isHRes(self, item):
|
|
if item:
|
|
return 'HI-RES'
|
|
else:
|
|
return 'Lossless'
|
|
|
|
def appendInfo(self, i, bool):
|
|
self.IDs.append(i['id'])
|
|
self.Types.append(bool)
|
|
|
|
def itResults(self, iterable):
|
|
for i in iterable:
|
|
try:
|
|
items = (i['artist']['name'], i['title'],
|
|
self.seconds(i['duration']), self.isHRes(i['hires']))
|
|
self.Total.append('[RELEASE] {} - {} - {} [{}]'.format(*items))
|
|
self.appendInfo(i, True)
|
|
except KeyError:
|
|
items = (i['performer']['name'], i['title'],
|
|
self.seconds(i['duration']), self.isHRes(i['hires']))
|
|
self.Total.append('[TRACK] {} - {} - {} [{}]'.format(*items))
|
|
self.appendInfo(i, False)
|
|
|
|
def getResults(self, tracks=False):
|
|
self.itResults(self.Albums)
|
|
if tracks:
|
|
self.itResults(self.Tracks)
|
|
|
|
def pickResults(self):
|
|
title = ('Select [space] the item(s) you want to download '
|
|
'(one or more)\nPress Ctrl + c to quit\n')
|
|
quality = ('Select [intro] the quality (the quality will be automat'
|
|
'ically\ndowngraded if the selected is not found)')
|
|
Qualitys = ['Lossless', 'Hi-res =< 96kHz', 'Hi-Res > 96 kHz']
|
|
try:
|
|
self.Selected = pick(self.Total, title,
|
|
multiselect=True, min_selection_count=1)
|
|
self.quality = pick(Qualitys, quality)
|
|
except KeyboardInterrupt:
|
|
sys.exit('Bye')
|