feat: add to_mp3 attribute in FFmpeg

krypton-byte · Feb 17, 2024 · dc45c74 · dc45c74
1 parent 8efb632
commit dc45c74
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/neonize/utils/ffmpeg.py b/neonize/utils/ffmpeg.py
@@ -182,6 +182,28 @@ def call(self, cmd: List[str]):
             )
         return out
 
+    def to_mp3(self) -> bytes:
+        temp = tempfile.gettempdir() + "/" + time.time().__str__() + ".mp3"
+        self.call(
+            [
+                "ffmpeg",
+                "-i",
+                self.filepath,
+                "-vn",
+                "-ar",
+                "44100",
+                "-ac",
+                "2",
+                "-b:a",
+                "192k",
+                temp,
+            ]
+        )
+        with open(temp, "rb") as file:
+            buf = file.read()
+        os.remove(temp)
+        return buf
+
     def extract_thumbnail(
         self,
         format: ImageFormat = ImageFormat.JPG,