ryg/royalnet
ryg/royalnet
1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00
Code Activity

Work on suggestion mode

Browse source
This commit is contained in:
Steffo 2019-01-04 19:33:00 +01:00
parent 2dd5092b72
commit 30a951c69e
1 changed files with 61 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

82
discordbot.py
View file

@ -79,6 +79,7 @@ class Video:
self.name = None self.name = None
self.enqueuer = enqueuer self.enqueuer = enqueuer
self.audio_source = None self.audio_source = None
self.suggestion = None
def __str__(self): def __str__(self):
if self.name is None: if self.name is None:
@ -108,8 +109,9 @@ class Video:
"""Create an AudioSource to be played through Discord, and store and return it.""" """Create an AudioSource to be played through Discord, and store and return it."""
raise NotImplementedError() raise NotImplementedError()
def get_suggestion(self) -> typing.Optional["Video"]: def get_suggestion(self):
"""Get the next suggested video, to be used when the queue is in LoopMode.FOLLOW_SUGGESTION""" """Set the next suggested video in the self.suggestion variable, to be used when the queue is in
LoopMode.FOLLOW_SUGGESTION"""
raise NotImplementedError() raise NotImplementedError()
@ -186,14 +188,16 @@ class YoutubeDLVideo(Video):
self.audio_source = discord.PCMVolumeTransformer(discord.FFmpegPCMAudio(self.get_filename(), **ffmpeg_settings)) self.audio_source = discord.PCMVolumeTransformer(discord.FFmpegPCMAudio(self.get_filename(), **ffmpeg_settings))
return self.audio_source return self.audio_source
def get_suggestion(self) -> typing.Optional["YoutubeDLVideo"]: def get_suggestion(self):
if self.info["extractor"] != "youtube": if self.suggestion is not None:
# TODO: add more websites? return
return None
if self.youtube_api_key is None:
raise errors.MissingAPIKeyError()
# Ensure the video has info # Ensure the video has info
self.get_info() self.get_info()
if self.info["extractor"] != "youtube":
# TODO: add more websites?
self.suggestion = NotImplemented
if self.youtube_api_key is None:
raise errors.MissingAPIKeyError()
# Request search data (costs 100 API units) # Request search data (costs 100 API units)
r = requests.get("https://www.googleapis.com/youtube/v3/search?part=snippet", params={ r = requests.get("https://www.googleapis.com/youtube/v3/search?part=snippet", params={
"part": "snippet", "part": "snippet",
@ -206,11 +210,11 @@ class YoutubeDLVideo(Video):
j = r.json() j = r.json()
# Find the suggested video # Find the suggested video
if len(j["items"]) < 1: if len(j["items"]) < 1:
raise errors.NotFoundError() self.suggestion = ...
first_video_id = j["items"][0]["id"]["videoId"] first_video_id = j["items"][0]["id"]["videoId"]
return YoutubeDLVideo(f"https://www.youtube.com/watch?v={first_video_id}", self.suggestion = YoutubeDLVideo(f"https://www.youtube.com/watch?v={first_video_id}",
enqueuer=None, enqueuer=None,
youtube_api_key=self.youtube_api_key) youtube_api_key=self.youtube_api_key)
class LoopMode(enum.Enum): class LoopMode(enum.Enum):
@ -261,9 +265,12 @@ class VideoQueue:
pass pass
elif self.loop_mode == LoopMode.FOLLOW_SUGGESTIONS: elif self.loop_mode == LoopMode.FOLLOW_SUGGESTIONS:
if self.now_playing is None: if self.now_playing is None:
self.now_playing = None if len(self.list) > 0:
return self.now_playing = self.list.pop(0)
self.now_playing = self.now_playing.get_suggestion() else:
return
self.now_playing.get_suggestion()
self.now_playing = self.now_playing.suggestion
elif self.loop_mode == LoopMode.AUTO_SHUFFLE: elif self.loop_mode == LoopMode.AUTO_SHUFFLE:
self.shuffle() self.shuffle()
try: try:
@ -296,6 +303,25 @@ class VideoQueue:
def not_ready_videos(self, limit: typing.Optional[int] = None): def not_ready_videos(self, limit: typing.Optional[int] = None):
"""Return the non-ready videos in the first limit positions of the queue.""" """Return the non-ready videos in the first limit positions of the queue."""
video_list = [] video_list = []
# In single video repeat, the only video to be loaded is the one currently playing
if self.loop_mode == LoopMode.LOOP_SINGLE:
if self.now_playing is None or self.now_playing.is_ready:
return video_list
else:
video_list.append(self.now_playing)
return video_list
# In suggestion mode, preload the current video and the next suggested one
if self.loop_mode == LoopMode.FOLLOW_SUGGESTIONS:
video = self.now_playing
if video is None:
return video_list
video.get_suggestion()
if not video.is_ready:
video_list.append(video)
if not video.suggestion.is_ready:
video_list.append(video.suggestion)
return video_list
# In all other modes, the videos to be loaded are the current one plus the following ones
for video in (self.list[:limit] + ([self.now_playing] if self.now_playing else [])): for video in (self.list[:limit] + ([self.now_playing] if self.now_playing else [])):
if not video.is_ready: if not video.is_ready:
video_list.append(video) video_list.append(video)
@ -558,6 +584,12 @@ class RoyalDiscordBot(discord.Client):
except (KeyError, ValueError): except (KeyError, ValueError):
logger.warning("Command prefix not set, defaulting to '!'.") logger.warning("Command prefix not set, defaulting to '!'.")
self.command_prefix = "!" self.command_prefix = "!"
# Youtube API Key
try:
self.youtube_api_key = config["YouTube"]["youtube_data_api_key"]
except (KeyError, ValueError):
logger.warning("Youtube API key not set, disabling suggestion mode.")
self.youtube_api_key = None
# noinspection PyAsyncCall # noinspection PyAsyncCall
async def on_ready(self): async def on_ready(self):
@ -890,10 +922,14 @@ class RoyalDiscordBot(discord.Client):
if "entries" in info: if "entries" in info:
logger.debug(f"Playlist detected at {url}.") logger.debug(f"Playlist detected at {url}.")
for entry in info["entries"]: for entry in info["entries"]:
self.video_queue.add(YoutubeDLVideo(entry["webpage_url"], enqueuer=enqueuer), index) self.video_queue.add(YoutubeDLVideo(entry["webpage_url"],
enqueuer=enqueuer,
youtube_api_key=self.youtube_api_key), index)
return return
logger.debug(f"Single video detected at {url}.") logger.debug(f"Single video detected at {url}.")
self.video_queue.add(YoutubeDLVideo(url, enqueuer=enqueuer), index) self.video_queue.add(YoutubeDLVideo(url,
enqueuer=enqueuer,
youtube_api_key=self.youtube_api_key), index)
# noinspection PyUnusedLocal # noinspection PyUnusedLocal
@command @command
@ -1072,14 +1108,14 @@ class RoyalDiscordBot(discord.Client):
elif self.video_queue.loop_mode == LoopMode.LOOP_SINGLE: elif self.video_queue.loop_mode == LoopMode.LOOP_SINGLE:
msg += "Modalità attuale: :repeat_one: **Ripeti canzone singola**\n" msg += "Modalità attuale: :repeat_one: **Ripeti canzone singola**\n"
elif self.video_queue.loop_mode == LoopMode.FOLLOW_SUGGESTIONS: elif self.video_queue.loop_mode == LoopMode.FOLLOW_SUGGESTIONS:
msg += "Modalità attuale: :arrows_clockwise: **Continua con video suggeriti**\n" msg += "Modalità attuale: :sparkles: **Continua con video suggeriti**\n"
elif self.video_queue.loop_mode == LoopMode.AUTO_SHUFFLE: elif self.video_queue.loop_mode == LoopMode.AUTO_SHUFFLE:
msg += "Modalità attuale: :twisted_rightwards_arrows: **Video casuale dalla coda**\n" msg += "Modalità attuale: :twisted_rightwards_arrows: **Video casuale dalla coda**\n"
elif self.video_queue.loop_mode == LoopMode.LOOPING_SHUFFLE: elif self.video_queue.loop_mode == LoopMode.LOOPING_SHUFFLE:
msg += "Modalità attuale: :arrows_counterclockwise: **Video casuali infiniti dalla coda**\n" msg += "Modalità attuale: :arrows_counterclockwise: **Video casuali infiniti dalla coda**\n"
msg += "**Video in coda:**\n" msg += "**Video in coda:**\n"
if self.video_queue.now_playing is None: if self.video_queue.now_playing is None:
msg += ":cloud: _nessuno_" msg += ":cloud: _nessuno_\n"
else: else:
msg += f":arrow_forward: {str(self.video_queue.now_playing)}\n" msg += f":arrow_forward: {str(self.video_queue.now_playing)}\n"
if self.video_queue.loop_mode == LoopMode.NORMAL: if self.video_queue.loop_mode == LoopMode.NORMAL:
@ -1106,7 +1142,7 @@ class RoyalDiscordBot(discord.Client):
msg += f":asterisk: {str(video)}\n" msg += f":asterisk: {str(video)}\n"
msg += "all'infinito!" msg += "all'infinito!"
elif self.video_queue.loop_mode == LoopMode.FOLLOW_SUGGESTIONS: elif self.video_queue.loop_mode == LoopMode.FOLLOW_SUGGESTIONS:
msg += ":rainbow:" msg += "e i prossimi video suggeriti!"
elif self.video_queue.loop_mode == LoopMode.AUTO_SHUFFLE: elif self.video_queue.loop_mode == LoopMode.AUTO_SHUFFLE:
for index, video in enumerate(self.video_queue.list[:10]): for index, video in enumerate(self.video_queue.list[:10]):
msg += f":hash: {str(video)}\n" msg += f":hash: {str(video)}\n"
@ -1211,8 +1247,12 @@ class RoyalDiscordBot(discord.Client):
self.video_queue.loop_mode = LoopMode.LOOP_QUEUE self.video_queue.loop_mode = LoopMode.LOOP_QUEUE
await channel.send("🔁 Modalità di coda impostata: **Ripeti intera coda**") await channel.send("🔁 Modalità di coda impostata: **Ripeti intera coda**")
elif params[1] == "suggest": elif params[1] == "suggest":
if self.youtube_api_key is None:
await channel.send("⚠️ La modalità **Continua con video suggeriti**"
" non è configurata correttamente ed è stata disattivata.")
return
self.video_queue.loop_mode = LoopMode.FOLLOW_SUGGESTIONS self.video_queue.loop_mode = LoopMode.FOLLOW_SUGGESTIONS
await channel.send("🌈️ Modalità di coda impostata: **Continua con video suggeriti**") await channel.send("✨ Modalità di coda impostata: **Continua con video suggeriti**")
elif params[1] == "random": elif params[1] == "random":
self.video_queue.loop_mode = LoopMode.AUTO_SHUFFLE self.video_queue.loop_mode = LoopMode.AUTO_SHUFFLE
await channel.send("🔀 Modalità di coda impostata: **Video casuale dalla coda**") await channel.send("🔀 Modalità di coda impostata: **Video casuale dalla coda**")