mirror of
https://github.com/Wojtek242/qobuz-dl.git
synced 2024-10-22 22:50:51 +02:00
102 lines
2.9 KiB
Python
102 lines
2.9 KiB
Python
|
from qo_utils.search import Search
|
||
|
from qo_utils import downloader
|
||
|
import argparse
|
||
|
import re
|
||
|
import os
|
||
|
import sys
|
||
|
import json
|
||
|
import qopy
|
||
|
|
||
|
|
||
|
def getArgs():
|
||
|
parser = argparse.ArgumentParser(prog='python3 main.py')
|
||
|
parser.add_argument("-a", action="store_true",
|
||
|
help="enable albums-only search")
|
||
|
parser.add_argument("-i", action="store_true",
|
||
|
help="run Qo-Dl-curses on URL input mode")
|
||
|
parser.add_argument("-q", metavar="int", default=6,
|
||
|
help="FLAC quality (6, 7, 27) (default: 6)")
|
||
|
parser.add_argument("-l", metavar="int", default=10,
|
||
|
help="limit of search results by type (default: 10)")
|
||
|
parser.add_argument("-d", metavar="PATH", default='Qobuz Downloads',
|
||
|
help="custom directory for downloads")
|
||
|
return parser.parse_args()
|
||
|
|
||
|
|
||
|
def getSession():
|
||
|
print('Logging...')
|
||
|
with open('config.json') as f:
|
||
|
config = json.load(f)
|
||
|
return qopy.Client(config['email'], config['password'])
|
||
|
|
||
|
|
||
|
def musicDir(dir):
|
||
|
fix = os.path.normpath(dir)
|
||
|
if not os.path.isdir(fix):
|
||
|
os.mkdir(fix)
|
||
|
return fix
|
||
|
|
||
|
|
||
|
def get_id(url):
|
||
|
return re.match(r'https?://(?:w{0,3}|play|open)\.qobuz\.com/(?:(?'
|
||
|
':album|track)/|[a-z]{2}-[a-z]{2}/album/-?\w+(?:-\w+)'
|
||
|
'*-?/|user/library/favorites/)(\w+)', url).group(1)
|
||
|
|
||
|
|
||
|
def searchSelected(Qz, path, start):
|
||
|
q = ['6', '7', '27']
|
||
|
quality = q[start.quality[1]]
|
||
|
for i in start.Selected:
|
||
|
if start.Types[i[1]]:
|
||
|
downloader.iterateIDs(Qz, start.IDs[i[1]], path, quality, True)
|
||
|
else:
|
||
|
downloader.iterateIDs(Qz, start.IDs[i[1]], path, quality, False)
|
||
|
|
||
|
|
||
|
def fromUrl(Qz, path, link, quality):
|
||
|
if '/track/' in link:
|
||
|
id = get_id(link)
|
||
|
downloader.iterateIDs(Qz, id, path, quality, False)
|
||
|
else:
|
||
|
id = get_id(link)
|
||
|
downloader.iterateIDs(Qz, id, path, quality, True)
|
||
|
|
||
|
|
||
|
def interactive(Qz, path, limit, tracks=True):
|
||
|
while True:
|
||
|
try:
|
||
|
query = input("\nEnter your search: [Ctrl + c to quit]\n- ")
|
||
|
print('Searching...')
|
||
|
start = Search(Qz, query, limit)
|
||
|
start.getResults(tracks)
|
||
|
start.pickResults()
|
||
|
searchSelected(Qz, path, start)
|
||
|
except KeyboardInterrupt:
|
||
|
sys.exit('\nBye')
|
||
|
|
||
|
|
||
|
def inputMode(Qz, path, quality):
|
||
|
while True:
|
||
|
try:
|
||
|
link = input("\nAlbum/track URL: [Ctrl + c to quit]\n- ")
|
||
|
fromUrl(Qz, path, link, quality)
|
||
|
except KeyboardInterrupt:
|
||
|
sys.exit('\nBye')
|
||
|
|
||
|
|
||
|
def main():
|
||
|
arguments = getArgs()
|
||
|
directory = musicDir(arguments.d) + '/'
|
||
|
Qz = getSession()
|
||
|
if not arguments.i:
|
||
|
if arguments.a:
|
||
|
interactive(Qz, directory, arguments.l, False)
|
||
|
else:
|
||
|
interactive(Qz, directory, arguments.l, True)
|
||
|
else:
|
||
|
inputMode(Qz, directory, arguments.q)
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
sys.exit(main())
|