134 lines
4.7 KiB
Python
134 lines
4.7 KiB
Python
import logging
|
|
import yt_dlp as youtube_dl
|
|
from constants import tr_cli as tr
|
|
import variables as var
|
|
from media.item import item_builders, item_loaders, item_id_generators
|
|
from media.url import URLItem, url_item_id_generator
|
|
|
|
|
|
log = logging.getLogger("bot")
|
|
|
|
|
|
def get_playlist_info(url, start_index=0, user=""):
|
|
ydl_opts = {
|
|
'extract_flat': 'in_playlist',
|
|
'verbose': var.config.getboolean('debug', 'youtube_dl')
|
|
}
|
|
|
|
cookie = var.config.get('youtube_dl', 'cookie_file')
|
|
if cookie:
|
|
ydl_opts['cookiefile'] = var.config.get('youtube_dl', 'cookie_file')
|
|
|
|
user_agent = var.config.get('youtube_dl', 'user_agent')
|
|
if user_agent:
|
|
youtube_dl.utils.std_headers['User-Agent'] = var.config.get('youtube_dl', 'user_agent')
|
|
|
|
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
|
|
attempts = var.config.getint('bot', 'download_attempts')
|
|
for i in range(attempts):
|
|
items = []
|
|
try:
|
|
info = ydl.extract_info(url, download=False)
|
|
# # if url is not a playlist but a video
|
|
# if 'entries' not in info and 'webpage_url' in info:
|
|
# music = {'type': 'url',
|
|
# 'title': info['title'],
|
|
# 'url': info['webpage_url'],
|
|
# 'user': user,
|
|
# 'ready': 'validation'}
|
|
# items.append(music)
|
|
# return items
|
|
|
|
playlist_title = info['title']
|
|
for j in range(start_index, min(len(info['entries']),
|
|
start_index + var.config.getint('bot', 'max_track_playlist'))):
|
|
# Unknow String if No title into the json
|
|
title = info['entries'][j]['title'] if 'title' in info['entries'][j] else "Unknown Title"
|
|
# Add youtube url if the url in the json isn't a full url
|
|
item_url = info['entries'][j]['url'] if info['entries'][j]['url'][0:4] == 'http' \
|
|
else "https://www.youtube.com/watch?v=" + info['entries'][j]['url']
|
|
print(info['entries'][j])
|
|
|
|
music = {
|
|
"type": "url_from_playlist",
|
|
"url": item_url,
|
|
"title": title,
|
|
"playlist_url": url,
|
|
"playlist_title": playlist_title,
|
|
"user": user
|
|
}
|
|
|
|
items.append(music)
|
|
|
|
except Exception as ex:
|
|
log.exception(ex, exc_info=True)
|
|
continue
|
|
|
|
return items
|
|
|
|
|
|
def playlist_url_item_builder(**kwargs):
|
|
return PlaylistURLItem(kwargs['url'],
|
|
kwargs['title'],
|
|
kwargs['playlist_url'],
|
|
kwargs['playlist_title'])
|
|
|
|
|
|
def playlist_url_item_loader(_dict):
|
|
return PlaylistURLItem("", "", "", "", _dict)
|
|
|
|
|
|
item_builders['url_from_playlist'] = playlist_url_item_builder
|
|
item_loaders['url_from_playlist'] = playlist_url_item_loader
|
|
item_id_generators['url_from_playlist'] = url_item_id_generator
|
|
|
|
|
|
class PlaylistURLItem(URLItem):
|
|
def __init__(self, url, title, playlist_url, playlist_title, from_dict=None):
|
|
if from_dict is None:
|
|
super().__init__(url)
|
|
self.title = title
|
|
self.playlist_url = playlist_url
|
|
self.playlist_title = playlist_title
|
|
else:
|
|
super().__init__("", from_dict)
|
|
self.playlist_title = from_dict['playlist_title']
|
|
self.playlist_url = from_dict['playlist_url']
|
|
|
|
self.type = "url_from_playlist"
|
|
|
|
def to_dict(self):
|
|
tmp_dict = super().to_dict()
|
|
tmp_dict['playlist_url'] = self.playlist_url
|
|
tmp_dict['playlist_title'] = self.playlist_title
|
|
|
|
return tmp_dict
|
|
|
|
def format_debug_string(self):
|
|
return "[url] {title} ({url}) from playlist {playlist}".format(
|
|
title=self.title,
|
|
url=self.url,
|
|
playlist=self.playlist_title
|
|
)
|
|
|
|
def format_song_string(self, user):
|
|
return tr("url_from_playlist_item",
|
|
title=self.title,
|
|
url=self.url,
|
|
playlist_url=self.playlist_url,
|
|
playlist=self.playlist_title,
|
|
user=user)
|
|
|
|
def format_current_playing(self, user):
|
|
display = tr("now_playing", item=self.format_song_string(user))
|
|
|
|
if self.thumbnail:
|
|
thumbnail_html = '<img width="80" src="data:image/jpge;base64,' + \
|
|
self.thumbnail + '"/>'
|
|
display += "<br />" + thumbnail_html
|
|
|
|
return display
|
|
|
|
def display_type(self):
|
|
return tr("url_from_playlist")
|