artie/artemis
1
0
Fork 0
mirror of https://github.com/artiemis/artemis.git synced 2026-02-14 08:31:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add user-friendly error message for youtube ban

Browse Source
This commit is contained in:
artie 2024-10-03 19:30:13 +02:00
parent 4476e4df73
commit dcc8d34235
1 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
artemis/cogs/media.py
View File

@ -1,7 +1,6 @@
from __future__ import annotations, unicode_literals from __future__ import annotations, unicode_literals
import asyncio import asyncio
import html
import re import re
import shlex import shlex
import struct import struct
@ -19,6 +18,7 @@ from bs4 import BeautifulSoup
from discord.ext import commands from discord.ext import commands
from pycaption import SRTWriter, WebVTTReader from pycaption import SRTWriter, WebVTTReader
from yt_dlp.utils import parse_duration from yt_dlp.utils import parse_duration
from yt_dlp.extractor.youtube import YoutubeIE
from .. import utils from .. import utils
from ..utils.common import ArtemisError, compress_image from ..utils.common import ArtemisError, compress_image
@ -56,7 +56,25 @@ def format_ytdlp_error(error: str) -> str:
return ret return ret
YOUTUBE_BANNED_MESSAGE = """
# Artemis has been banned from YouTube.
## Here are some trusted alternatives:
**Cross-platform**
**[Cobalt](<https://cobalt.tools/>)** - user-friendly web application
**Android**
**[Seal](<https://github.com/JunkFood02/Seal>)** - pretty open-source app that wraps yt-dlp
**Desktop (Advanced)**
**[yt-dlp](<https://github.com/yt-dlp/yt-dlp>)** - the best command-line downloader
"""
async def run_ytdlp(query: str, opts: dict, download: bool = True) -> dict: async def run_ytdlp(query: str, opts: dict, download: bool = True) -> dict:
if YoutubeIE.suitable(query):
raise ArtemisError(YOUTUBE_BANNED_MESSAGE)
try: try:
with yt_dlp.YoutubeDL(opts) as ytdl: with yt_dlp.YoutubeDL(opts) as ytdl:
return await asyncio.to_thread(ytdl.extract_info, query, download=download) return await asyncio.to_thread(ytdl.extract_info, query, download=download)
@ -339,7 +357,7 @@ class Media(commands.Cog):
trim = flags.trim trim = flags.trim
ss, to = flags.ss, None ss, to = flags.ss, None
async def monitor_download(): async def _monitor_download():
nonlocal msg, state nonlocal msg, state
while not finished: while not finished:
content = "Processing..." content = "Processing..."
@ -424,7 +442,7 @@ class Media(commands.Cog):
ytdl_opts["format"] = format ytdl_opts["format"] = format
info_dict = None info_dict = None
asyncio.create_task(monitor_download()) # asyncio.create_task(monitor_download())
async with ctx.typing(): async with ctx.typing():
info_dict = await run_ytdlp(url, ytdl_opts) info_dict = await run_ytdlp(url, ytdl_opts)
state = "uploading" state = "uploading"