From b28f8d244a3cb901eff5be438ad8f0508bd8e024 Mon Sep 17 00:00:00 2001 From: Matthew Date: Sat, 3 Apr 2021 21:50:58 +1300 Subject: [PATCH] [YouTube] Show premium state in `availability` (#209) Authored by colethedj --- yt_dlp/extractor/youtube.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/yt_dlp/extractor/youtube.py b/yt_dlp/extractor/youtube.py index 794f51bb6c..3e19d18a00 100644 --- a/yt_dlp/extractor/youtube.py +++ b/yt_dlp/extractor/youtube.py @@ -2296,8 +2296,10 @@ class YoutubeIE(YoutubeBaseInfoExtractor): is_private = bool_or_none(video_details.get('isPrivate')) is_unlisted = bool_or_none(microformat.get('isUnlisted')) is_membersonly = None + is_premium = None if initial_data and is_private is not None: is_membersonly = False + is_premium = False contents = try_get(initial_data, lambda x: x['contents']['twoColumnWatchNextResults']['results']['results']['contents'], list) for content in contents or []: badges = try_get(content, lambda x: x['videoPrimaryInfoRenderer']['badges'], list) @@ -2306,13 +2308,16 @@ class YoutubeIE(YoutubeBaseInfoExtractor): if label.lower() == 'members only': is_membersonly = True break - if is_membersonly: + elif label.lower() == 'premium': + is_premium = True + break + if is_membersonly or is_premium: break # TODO: Add this for playlists info['availability'] = self._availability( is_private=is_private, - needs_premium=False, # Youtube no longer have premium-only videos? + needs_premium=is_premium, needs_subscription=is_membersonly, needs_auth=info['age_limit'] >= 18, is_unlisted=None if is_private is None else is_unlisted)