]> jfr.im git - yt-dlp.git/commitdiff
[extractor/mediaset] Fix embed extraction
authorpukkandan <redacted>
Wed, 31 Aug 2022 16:54:31 +0000 (22:24 +0530)
committerpukkandan <redacted>
Wed, 31 Aug 2022 16:54:41 +0000 (22:24 +0530)
Closes #4804

yt_dlp/extractor/mediaset.py

index 0671c29a66e34c16583218b8c09e7276172b6751..ebe894f74033814a249e20d8b757161d6f51a11f 100644 (file)
@@ -172,31 +172,27 @@ class MediasetIE(ThePlatformBaseIE):
     }]
 
     def _extract_from_webpage(self, url, webpage):
-        def _qs(url):
-            return parse_qs(url)
-
         def _program_guid(qs):
             return qs.get('programGuid', [None])[0]
 
-        entries = []
         for mobj in re.finditer(
                 r'<iframe\b[^>]+\bsrc=(["\'])(?P<url>(?:https?:)?//(?:www\.)?video\.mediaset\.it/player/playerIFrame(?:Twitter)?\.shtml.*?)\1',
                 webpage):
             embed_url = mobj.group('url')
-            embed_qs = _qs(embed_url)
+            embed_qs = parse_qs(embed_url)
             program_guid = _program_guid(embed_qs)
             if program_guid:
-                entries.append(embed_url)
+                yield self.url_result(embed_url)
                 continue
+
             video_id = embed_qs.get('id', [None])[0]
             if not video_id:
                 continue
             urlh = self._request_webpage(embed_url, video_id, note='Following embed URL redirect')
             embed_url = urlh.geturl()
-            program_guid = _program_guid(_qs(embed_url))
+            program_guid = _program_guid(parse_qs(embed_url))
             if program_guid:
-                entries.append(embed_url)
-        return entries
+                yield self.url_result(embed_url)
 
     def _parse_smil_formats(self, smil, smil_url, video_id, namespace=None, f4m_params=None, transform_rtmp_url=None):
         for video in smil.findall(self._xpath_ns('.//video', namespace)):