From ee10a32391ece2dcd85ae5d5e2cb8af68f71f953 Mon Sep 17 00:00:00 2001
From: BarsTiger <kit.barsic@gmail.com>
Date: Tue, 31 Oct 2023 12:54:02 +0200
Subject: [PATCH] Faster downloading

---
 bot/modules/youtube/downloader.py | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/bot/modules/youtube/downloader.py b/bot/modules/youtube/downloader.py
index 322e35f..76bf63f 100644
--- a/bot/modules/youtube/downloader.py
+++ b/bot/modules/youtube/downloader.py
@@ -27,21 +27,33 @@ class YouTubeBytestream:
             duration=int(duration),
         )
 
+    async def rerender(self):
+        segment = AudioSegment.from_file(
+            file=self.file
+        )
+
+        self.file = segment.export(BytesIO(), format='mp3', codec='libmp3lame')
+        return self
+
 
 @define
 class Downloader:
     audio_stream: Stream
     filename: str
+    duration: int
 
     @classmethod
     def from_id(cls, yt_id: str):
         video = YouTube.from_id(yt_id)
+
         audio_stream = video.streams.filter(
             only_audio=True,
         ).order_by('abr').desc().first()
+
         return cls(
             audio_stream=audio_stream,
             filename=f'{audio_stream.default_filename}.mp3',
+            duration=int(video.length),
         )
 
     def __to_bytestream(self):
@@ -49,14 +61,10 @@ class Downloader:
         self.audio_stream.stream_to_buffer(audio_io)
         audio_io.seek(0)
 
-        segment = AudioSegment.from_file(
-            file=audio_io
-        )
-
         return YouTubeBytestream.from_bytestream(
-            segment.export(BytesIO(), format='mp3', codec='libmp3lame'),
+            audio_io,
             self.filename,
-            segment.duration_seconds
+            self.duration,
         )
 
     async def to_bytestream(self):