[YouTube] Show premium state in `availability` (#209)

Authored by colethedj

diff --git a/yt_dlp/extractor/youtube.py b/yt_dlp/extractor/youtube.py
index 794f51bb6cd3618919dff2a3c707dd436166a034..3e19d18a009cd15076654fd388e6f11f73ec6a4f 100644 (file)
--- a/yt_dlp/extractor/youtube.py
+++ b/yt_dlp/extractor/youtube.py
@@ -2296,8 +2296,10 @@ def chapter_time(mmlir):
         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 @@ def chapter_time(mmlir):
                     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)