mmaker/niconico-gay
1
Fork 0
Code Issues Pull requests Activity

Only download video if request from Discord

Browse source
This commit is contained in:
MMaker 2025-02-27 10:29:36 -05:00
commit ac86c5f5ee
Signed by: mmaker
GPG key ID: CCE79B8FEDA40FB2
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
app.py
View file

@ -290,6 +290,10 @@ def proxy(video_id):
video_width, video_height = get_video_resolution(params) if params else (None, None)
download_allowed = allow_download(params) if params else False
request_user_agent = request.headers.get('User-Agent', '')
if download_allowed and 'discordbot' not in request_user_agent:
logger.info(f"Download ignored for video ID {video_id} due to user agent ({request_user_agent})")
download_allowed = False
video_quality = get_video_quality(params) if params else None
if download_allowed and video_quality is not None:
video_in_cdn = is_video_in_cdn(video_id)