m3u8 toggle (#293)

tgbot-collection · Sep 17, 2023 · 1ea8390 · 1ea8390
1 parent 32fe1a8
commit 1ea8390
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -142,6 +142,7 @@ You can configure all the following environment variables:
 * ENABLE_FFMPEG = os.getenv("ENABLE_FFMPEG", False)
 * PROVIDER_TOKEN: stripe token on Telegram payment
 * PLAYLIST_SUPPORT: download playlist support
+* M3U8_SUPPORT: download m3u8 files support
 * ENABLE_ARIA2: enable aria2c download
 * FREE_DOWNLOAD: free download count per day
 * TOKEN_PRICE: token price per 1 USD

diff --git a/ytdlbot/config.py b/ytdlbot/config.py
@@ -56,6 +56,7 @@
 PROVIDER_TOKEN = os.getenv("PROVIDER_TOKEN") or "1234"
 
 PLAYLIST_SUPPORT = os.getenv("PLAYLIST_SUPPORT", False)
+M3U8_SUPPORT = os.getenv("M3U8_SUPPORT", False)
 ENABLE_ARIA2 = os.getenv("ENABLE_ARIA2", False)
 
 FREE_DOWNLOAD = os.getenv("FREE_DOWNLOAD", 20)

diff --git a/ytdlbot/ytdl_bot.py b/ytdlbot/ytdl_bot.py
@@ -38,6 +38,7 @@
     ENABLE_VIP,
     OWNER,
     PLAYLIST_SUPPORT,
+    M3U8_SUPPORT,
     PROVIDER_TOKEN,
     REQUIRED_MEMBERSHIP,
     TOKEN_PRICE,
@@ -356,7 +357,9 @@ def link_checker(url: str) -> str:
     ):
         return "Playlist or channel links are disabled."
 
-    if re.findall(r"m3u8|\.m3u8|\.m3u$", url.lower()):
+    if not M3U8_SUPPORT and (
+        re.findall(r"m3u8|\.m3u8|\.m3u$", url.lower())
+    ):
         return "m3u8 links are disabled."
 
     with contextlib.suppress(yt_dlp.utils.DownloadError):