ErrorNoInternet/ErrorNoCord
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(youtubedl): CustomAudioSource -> TrackedAudioSource

Browse Source
This commit is contained in:
Ryan 2025-02-05 21:52:45 -05:00
parent 3719fc69b5
commit 70ed37737c
Signed by: ErrorNoInternet
GPG Key ID: 2486BFB7B1E6A4A3
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
youtubedl.py
View File

@ -12,7 +12,7 @@ from constants import BAR_LENGTH, EMBED_COLOR, YTDL_OPTIONS
ytdl = yt_dlp.YoutubeDL(YTDL_OPTIONS) ytdl = yt_dlp.YoutubeDL(YTDL_OPTIONS)
class CustomAudioSource(disnake.AudioSource): class TrackedAudioSource(disnake.AudioSource):
def __init__(self, source): def __init__(self, source):
self._source = source self._source = source
self.read_count = 0 self.read_count = 0
@ -33,7 +33,7 @@ class CustomAudioSource(disnake.AudioSource):
class PCMVolumeTransformer(disnake.AudioSource): class PCMVolumeTransformer(disnake.AudioSource):
def __init__(self, original: CustomAudioSource, volume: float = 1.0) -> None: def __init__(self, original: TrackedAudioSource, volume: float = 1.0) -> None:
if original.is_opus(): if original.is_opus():
raise disnake.ClientException("AudioSource must not be Opus encoded.") raise disnake.ClientException("AudioSource must not be Opus encoded.")
@ -58,7 +58,7 @@ class PCMVolumeTransformer(disnake.AudioSource):
class YTDLSource(PCMVolumeTransformer): class YTDLSource(PCMVolumeTransformer):
def __init__( def __init__(
self, source: CustomAudioSource, *, data: dict[str, Any], volume: float = 0.5 self, source: TrackedAudioSource, *, data: dict[str, Any], volume: float = 0.5
): ):
super().__init__(source, volume) super().__init__(source, volume)
@ -93,7 +93,7 @@ class YTDLSource(PCMVolumeTransformer):
data = data["entries"][0] data = data["entries"][0]
return cls( return cls(
CustomAudioSource( TrackedAudioSource(
disnake.FFmpegPCMAudio( disnake.FFmpegPCMAudio(
data["url"] if stream else ytdl.prepare_filename(data), data["url"] if stream else ytdl.prepare_filename(data),
before_options="-vn -reconnect 1", before_options="-vn -reconnect 1",