From 70ed37737c1a7bb5d42b9fd3c4dd98b4459af6a7 Mon Sep 17 00:00:00 2001 From: ErrorNoInternet Date: Wed, 5 Feb 2025 21:52:45 -0500 Subject: [PATCH] refactor(youtubedl): CustomAudioSource -> TrackedAudioSource --- youtubedl.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/youtubedl.py b/youtubedl.py index 9233985..5e12b3f 100644 --- a/youtubedl.py +++ b/youtubedl.py @@ -12,7 +12,7 @@ from constants import BAR_LENGTH, EMBED_COLOR, YTDL_OPTIONS ytdl = yt_dlp.YoutubeDL(YTDL_OPTIONS) -class CustomAudioSource(disnake.AudioSource): +class TrackedAudioSource(disnake.AudioSource): def __init__(self, source): self._source = source self.read_count = 0 @@ -33,7 +33,7 @@ class CustomAudioSource(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(): raise disnake.ClientException("AudioSource must not be Opus encoded.") @@ -58,7 +58,7 @@ class PCMVolumeTransformer(disnake.AudioSource): class YTDLSource(PCMVolumeTransformer): 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) @@ -93,7 +93,7 @@ class YTDLSource(PCMVolumeTransformer): data = data["entries"][0] return cls( - CustomAudioSource( + TrackedAudioSource( disnake.FFmpegPCMAudio( data["url"] if stream else ytdl.prepare_filename(data), before_options="-vn -reconnect 1",