follow up of UsergeTeam#390 (UsergeTeam#398)

* allow different ytdl version also, in voice call plugins Co-authored-by: rking32 <[email protected]> * fix typo Co-authored-by: rking32 <[email protected]>
yasirarism · Nov 3, 2021 · 68f6636 · 68f6636
1 parent eb5529b
commit 68f6636
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/userge/plugins/utils/voice_call.py b/userge/plugins/utils/voice_call.py
@@ -12,14 +12,14 @@
 
 import asyncio
 import glob
+import importlib
 import os
 import re
 import shutil
 from traceback import format_exc
 from typing import List, Tuple
 
 import ffmpeg
-import youtube_dl as ytdl
 from pyrogram.raw import functions
 from pyrogram.types import (
     InlineKeyboardMarkup, InlineKeyboardButton, CallbackQuery, Message as RawMessage
@@ -34,6 +34,14 @@
 from userge.utils.exceptions import StopConversation
 
 CHANNEL = userge.getCLogger(__name__)
+LOGGER = userge.getLogger(__name__)
+
+reqd_module = os.environ.get("YOUTUBE_DL_PATH", "youtube_dl")
+try:
+    ytdl = importlib.import_module(reqd_module)
+except ModuleNotFoundError:
+    LOGGER.info("please fix your requirements.txt file")
+    raise
 
 VC_DB = get_collection("VC_CMDS_TOGGLE")
 CMDS_FOR_ALL = False