177 lines
4.8 KiB
Python
177 lines
4.8 KiB
Python
#coding: utf8
|
|
import downloader
|
|
from io import BytesIO
|
|
from utils import Downloader, LazyUrl, get_print, try_n, lock, clean_title, get_max_range
|
|
import ffmpeg
|
|
import ytdl
|
|
from m3u8_tools import M3u8_stream
|
|
from ratelimit import limits, sleep_and_retry
|
|
CLIENT_ID = None
|
|
|
|
|
|
@lock
|
|
def get_cid(force=False):
|
|
global CLIENT_ID
|
|
if CLIENT_ID is None or force:
|
|
print('update cid...')
|
|
d = ytdl.YoutubeDL()
|
|
e = ytdl.extractor.soundcloud.SoundcloudIE(d)
|
|
e._update_client_id()
|
|
CLIENT_ID = e._CLIENT_ID
|
|
return CLIENT_ID
|
|
|
|
|
|
class Audio:
|
|
_url = None
|
|
|
|
def __init__(self, url, album_art, cw=None):
|
|
self.album_art = album_art
|
|
self.cw = cw
|
|
self.url = LazyUrl(url, self.get, self, pp=self.pp)
|
|
|
|
@try_n(2)
|
|
@sleep_and_retry
|
|
@limits(1, 1)
|
|
def get(self, url):
|
|
print_ = get_print(self.cw)
|
|
if self._url:
|
|
return self._url
|
|
|
|
ydl = ytdl.YoutubeDL()
|
|
self.info = info = ydl.extract_info(url)
|
|
|
|
formats = info['formats']
|
|
print(formats)
|
|
formats = sorted(formats, key=lambda x: int(x.get('abr', 0)), reverse=True)
|
|
url_audio = None
|
|
|
|
for format in formats:
|
|
protocol = format['protocol']
|
|
print_('【{}】 format【{}】 abr【{}】'.format(protocol, format['format'], format.get('abr', 0)))
|
|
if not url_audio and protocol in ['http', 'https']:
|
|
url_audio = format['url']
|
|
|
|
if not url_audio:
|
|
url_audio = M3u8_stream(formats[0]['url'])
|
|
self.album_art = False#
|
|
|
|
self.username = info['uploader']
|
|
self.title = '{} - {}'.format(self.username, info['title'])
|
|
self.filename = '{}{}'.format(clean_title(self.title, allow_dot=True, n=-4), '.mp3')
|
|
|
|
thumb = None
|
|
for t in info['thumbnails'][::-1]:
|
|
width = t.get('width', 1080)
|
|
if not 100 <= width <= 500:
|
|
continue
|
|
url_thumb = t['url']
|
|
thumb = BytesIO()
|
|
try:
|
|
downloader.download(url_thumb, buffer=thumb)
|
|
break
|
|
except Exception as e:
|
|
print(e)
|
|
thumb = None
|
|
self.thumb = thumb
|
|
|
|
self._url = url_audio
|
|
return self._url
|
|
|
|
def pp(self, filename):
|
|
if self.thumb and self.album_art:
|
|
self.thumb.seek(0)#
|
|
ffmpeg.add_cover(filename, self.thumb, {'artist':self.username, 'title':self.info['title']}, cw=self.cw)
|
|
|
|
|
|
|
|
class Downloader_soundcloud(Downloader):
|
|
type = 'soundcloud'
|
|
single = True
|
|
URLS = ['soundcloud.com']
|
|
#lock = True
|
|
audio = None
|
|
display_name = 'SoundCloud'
|
|
|
|
def init(self):
|
|
if 'soundcloud.com' in self.url.lower():
|
|
self.url = self.url.replace('http://', 'https://')
|
|
else:
|
|
self.url = 'https://soundcloud.com/{}'.format(self.url)
|
|
|
|
@classmethod
|
|
def fix_url(cls, url):
|
|
return url.split('?')[0]
|
|
|
|
def read(self):
|
|
album_art = self.ui_setting.albumArt.isChecked()
|
|
info = get_audios(self.url, self.cw, album_art)
|
|
audios = info['audios']
|
|
|
|
if not audios:
|
|
raise Exception('no audios')
|
|
|
|
# first audio must be valid
|
|
while audios:
|
|
audio = audios[0]
|
|
try:
|
|
audio.url()
|
|
break
|
|
except Exception as e:
|
|
e_ = e
|
|
print(e)
|
|
audios.remove(audio)
|
|
else:
|
|
raise e_
|
|
|
|
if len(audios) > 1:
|
|
audio = self.process_playlist(info['title'], audios)
|
|
else:
|
|
self.urls.append(audio.url)
|
|
self.title = audio.title
|
|
|
|
self.artist = audio.username
|
|
self.setIcon(audio.thumb)
|
|
|
|
|
|
@try_n(2)
|
|
def get_audios(url, cw, album_art):
|
|
print_ = get_print(cw)
|
|
url = url.rstrip('/')
|
|
if url.count('/') == 3:
|
|
url += '/tracks'
|
|
|
|
options = {
|
|
'extract_flat': True,
|
|
'playlistend': get_max_range(cw),
|
|
}
|
|
|
|
ydl = ytdl.YoutubeDL(options, cw=cw)
|
|
info = ydl.extract_info(url)
|
|
if 'entries' in info:
|
|
entries = info['entries']
|
|
title = info['title']
|
|
for _type in ['All', 'Tracks', 'Albums', 'Sets', 'Reposts', 'Likes', 'Spotlight']:
|
|
x = '({})'.format(_type)
|
|
if x in title:
|
|
title = title.replace(x, '')
|
|
kind = _type
|
|
break
|
|
else:
|
|
kind = 'Playlist'
|
|
print_('kind: {}'.format(kind))
|
|
info['title'] = '[{}] {}'.format(kind.capitalize(), title)
|
|
else:
|
|
entries = [info]
|
|
|
|
audios = []
|
|
for e in entries:
|
|
url = e.get('webpage_url') or e['url']
|
|
if '/sets/' in url:
|
|
continue
|
|
audio = Audio(url, album_art, cw=cw)
|
|
audios.append(audio)
|
|
|
|
info['audios'] = audios
|
|
|
|
return info
|