Define the audio formats tuple in python backend

1 files changed, 2 insertions, 2 deletions

diff --git a/app/ytdl.py b/app/ytdl.py
index b72f883..c001a16 100644
--- a/app/ytdl.py
+++ b/app/ytdl.py
@@ -7,7 +7,7 @@ import asyncio
 import multiprocessing

 import logging

 import re

-from dl_formats import get_format, get_opts

+from dl_formats import get_format, get_opts, AUDIO_FORMATS

 

 log = logging.getLogger('ytdl')

 

@@ -234,7 +234,7 @@ class DownloadQueue:
             if not self.queue.exists(entry['id']):

                 dl = DownloadInfo(entry['id'], entry['title'], entry.get('webpage_url') or entry['url'], quality, format, folder)

                 # Keep consistent with frontend

-                base_directory = self.config.DOWNLOAD_DIR if (quality != 'audio' and format  not in ("m4a", "mp3", "opus", "wav")) else self.config.AUDIO_DOWNLOAD_DIR

+                base_directory = self.config.DOWNLOAD_DIR if (quality != 'audio' and format not in AUDIO_FORMATS) else self.config.AUDIO_DOWNLOAD_DIR

                 if folder:

                     if not self.config.CUSTOM_DIRS:

                         return {'status': 'error', 'msg': f'A folder for the download was specified but CUSTOM_DIRS is not true in the configuration.'}