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

Refactor out allow check

Browse Source
This commit is contained in:
MMaker 2025-02-27 12:52:37 -05:00
parent 96326f543f
commit 19befc9eb5
Signed by: mmaker
GPG Key ID: CCE79B8FEDA40FB2
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
app.py
View File

@ -191,11 +191,6 @@ def get_cdn_url(video_id):
"""Get the CDN URL for a video""" """Get the CDN URL for a video"""
return f"{CDN_BASE_URL}/niconico/{video_id}.mp4" return f"{CDN_BASE_URL}/niconico/{video_id}.mp4"
def allow_download(params):
if params['video']['duration'] > 60 * 15:
return False
return True
def get_video_resolution(params): def get_video_resolution(params):
if not params: if not params:
return None, None return None, None
@ -314,10 +309,13 @@ def proxy(video_id):
) if params else None ) if params else None
video_width, video_height = get_video_resolution(params) if params else (None, None) video_width, video_height = get_video_resolution(params) if params else (None, None)
download_allowed = allow_download(params) if params else False download_allowed = True
if params['video']['duration'] > 60 * 20: # 20 minutes
logger.info(f"{video_id}: Video download ignored due to duration ({params['video']['duration']} seconds)")
download_allowed = False
request_user_agent = request.headers.get('User-Agent', '').lower() request_user_agent = request.headers.get('User-Agent', '').lower()
if download_allowed and 'discordbot' not in request_user_agent: if download_allowed and 'discordbot' not in request_user_agent:
logger.info(f"{video_id}: Video download ignored for due to user agent ({request_user_agent})") logger.info(f"{video_id}: Video download ignored due to user agent ({request_user_agent})")
download_allowed = False download_allowed = False
video_quality = get_video_quality(params) if params else None video_quality = get_video_quality(params) if params else None
if download_allowed and video_quality is not None: if download_allowed and video_quality is not None:
@ -334,6 +332,7 @@ def proxy(video_id):
cdn_video_url = get_cdn_url(video_id) cdn_video_url = get_cdn_url(video_id)
og_tags = soup.find_all("meta", attrs={"property": True}) og_tags = soup.find_all("meta", attrs={"property": True})
for tag in og_tags: for tag in og_tags:
# Remove attribute(s) added by niconico
if 'data-server' in tag.attrs: if 'data-server' in tag.attrs:
del tag.attrs['data-server'] del tag.attrs['data-server']
# Fix thumbnail # Fix thumbnail