neoarz/Syntrel
1
0
Fork 0
mirror of https://github.com/neoarz/Syntrel.git synced 2025-12-25 03:40:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix(download): support for yt-dlp cookies in media download

Browse Source 
Introduces environment variable support for specifying yt-dlp cookies via file path or direct content for the media download command. Updates error handling to provide clearer messages when cookies are required and ensures temporary cookie files are cleaned up. Also updates .env.example and .gitignore to reflect these changes.
This commit is contained in:
neoarz
2025-10-03 12:49:50 -04:00
parent 85a43e66b0
commit 2c0b0afb39
3 changed files with 58 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
cogs/media/download.py
View File

@@ -89,6 +89,7 @@ def download_command():
processing_msg = await context.send(embed=processing_embed)
temp_dir = tempfile.mkdtemp()
temp_cookie_file = None
ydl_opts = {
'format': 'bestvideo[filesize<200M]+bestaudio[filesize<200M]/best[filesize<200M]/bestvideo+bestaudio/best',
@@ -102,6 +103,45 @@ def download_command():
'merge_output_format': 'mp4',
}
cookie_file_env = os.getenv('YTDLP_COOKIE_FILE') or os.getenv('YT_DLP_COOKIE_FILE')
cookies_text_env = os.getenv('YTDLP_COOKIES') or os.getenv('YT_DLP_COOKIES')
resolved_cookie_path = None
if cookie_file_env:
resolved_cookie_path = cookie_file_env
elif cookies_text_env:
try:
fd, temp_cookie_file = tempfile.mkstemp(prefix='yt_cookies_', text=True)
with os.fdopen(fd, 'w') as tmpf:
tmpf.write(cookies_text_env)
resolved_cookie_path = temp_cookie_file
except Exception:
temp_cookie_file = None
else:
default_local_cookie = os.path.join(os.path.dirname(__file__), 'files', 'cookies.txt')
resolved_cookie_path = default_local_cookie
if not (resolved_cookie_path and os.path.exists(resolved_cookie_path)):
embed = discord.Embed(
title="Error",
description=(
"Cookies file not found. Provide one via `YTDLP_COOKIE_FILE` or place a file at "
"`cogs/media/files/cookies.txt`."
),
color=0xE02B2B,
)
embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
if interaction is not None:
if not interaction.response.is_done():
await interaction.response.send_message(embed=embed, ephemeral=True)
else:
await interaction.followup.send(embed=embed, ephemeral=True)
else:
await context.send(embed=embed, ephemeral=True)
return
ydl_opts['cookiefile'] = resolved_cookie_path
try:
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info = await asyncio.get_event_loop().run_in_executor(
@@ -232,9 +272,15 @@ def download_command():
await context.channel.send(file=file)
except Exception as e:
err_text = str(e)
needs_cookies = ('Sign in to confirm' in err_text) or ('cookies' in err_text.lower()) or ('consent' in err_text.lower())
if needs_cookies:
extra = "\n\nThis source may require authentication. Provide cookies via env: `YTDLP_COOKIE_FILE` (path) or `YTDLP_COOKIES` (contents). See: https://github.com/yt-dlp/yt-dlp/wiki/FAQ#how-do-i-pass-cookies-to-yt-dlp"
else:
extra = ""
embed = discord.Embed(
title="Error",
description=f"Failed to download video: {str(e)}",
description=f"Failed to download video: {err_text}{extra}",
color=0xE02B2B,
)
embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
@@ -262,5 +308,10 @@ def download_command():
os.rmdir(temp_dir)
except:
pass
if temp_cookie_file:
try:
os.remove(temp_cookie_file)
except:
pass
return download