From: shirt <redacted>
Date: Sun, 27 Mar 2022 05:28:26 +0000 (-0400)
Subject: [youtube] Detect DRM better
X-Git-Tag: 2022.04.08~68
X-Git-Url: https://jfr.im/git/yt-dlp.git/commitdiff_plain/727029c50889d4c36dcfcffa2be6cdec309edf56?hp=5c3895fff150871fde273a10c55691403931b4dc

[youtube] Detect DRM better

Authored by: shirt-dev
---

diff --git a/yt_dlp/extractor/youtube.py b/yt_dlp/extractor/youtube.py
index 0726e27b4..bd3a7d36b 100644
--- a/yt_dlp/extractor/youtube.py
+++ b/yt_dlp/extractor/youtube.py
@@ -3023,7 +3023,7 @@ def _extract_formats(self, streaming_data, video_id, player_url, is_live, durati
         streaming_formats = traverse_obj(streaming_data, (..., ('formats', 'adaptiveFormats'), ...), default=[])
 
         for fmt in streaming_formats:
-            if fmt.get('targetDurationSec') or fmt.get('drmFamilies'):
+            if fmt.get('targetDurationSec'):
                 continue
 
             itag = str_or_none(fmt.get('itag'))
@@ -3105,6 +3105,7 @@ def _extract_formats(self, streaming_data, video_id, player_url, is_live, durati
                 'fps': int_or_none(fmt.get('fps')) or None,
                 'height': height,
                 'quality': q(quality),
+                'has_drm': bool(fmt.get('drmFamilies')),
                 'tbr': tbr,
                 'url': fmt_url,
                 'width': int_or_none(fmt.get('width')),