Env toggle to enable / disable Playlist download (tgbot-collection#200)

* Env toggle to enable / disable Playlist download

* i think this else is obsolete

README.md

@@ -137,6 +137,7 @@ you can configure all the following environment variables:
 * AUDIO_FORMAT: audio format, default is m4a. You can set to any known and supported format for ffmpeg. For
   example,`mp3`, `flac`, etc. ⚠️ m4a is the fastest. Other formats may affect performance.
 * ARCHIVE_ID: group or channel id/username. All downloads will send to this group first and then forward to end user.
+* PLAYLIST_SUPPORT: `True` or `False`, Ability to enable or disable downloads of playlist / channels by bot. Default: `False`.
   **Inline button will be lost during the forwarding.**
 
 ## 3.2 Set up init data

ytdlbot/config.py

@@ -62,3 +62,4 @@
 RATE = float(os.getenv("RATE", 0.01))
 BURST = int(os.getenv("BURST", 3))
 PROVIDER_TOKEN = os.getenv("PROVIDER_TOKEN") or "1234"
+PLAYLIST_SUPPORT = os.getenv("PLAYLIST_SUPPORT", False)

ytdlbot/ytdl_bot.py

@@ -29,7 +29,7 @@
 from client_init import create_app
 from config import (AUTHORIZED_USER, BURST, ENABLE_CELERY, ENABLE_FFMPEG,
                     ENABLE_VIP, MULTIPLY, OWNER, PROVIDER_TOKEN, QUOTA, RATE,
-                    REQUIRED_MEMBERSHIP)
+                    REQUIRED_MEMBERSHIP, PLAYLIST_SUPPORT)
 from constant import BotText
 from db import InfluxDB, MySQL, Redis
 from limit import VIP, verify_payment
@@ -331,10 +331,11 @@ def download_handler(client: "Client", message: "types.Message"):
         message.reply_text("I think you should send me a link.", quote=True)
         return
 
-    if re.findall(r"^https://www\.youtube\.com/channel/", VIP.extract_canonical_link(url)) or "list" in url:
-        message.reply_text("Channel/list download is disabled now. Please send me individual video link.", quote=True)
-        red.update_metrics("reject_channel")
-        return
+    if not PLAYLIST_SUPPORT:
+        if re.findall(r"^https://www\.youtube\.com/channel/", VIP.extract_canonical_link(url)) or "list" in url:
+            message.reply_text("Channel/list download is disabled now. Please send me individual video link.", quote=True)
+            red.update_metrics("reject_channel")
+            return
     # non vip user, consume too many token
     if (not VIP().check_vip(chat_id)) and (not lim.consume(str(chat_id).encode(), 1)):
         red.update_metrics("rate_limit")