Cleanup

yt_dlp/extractor/_extractors.py

@@ -1534,7 +1534,6 @@ from .pinterest import (
 )
 from .piramidetv import (
     PiramideTVChannelIE,
-    PiramideTVChannelURLIE,
     PiramideTVIE,
 )
 from .pixivsketch import (

yt_dlp/extractor/piramidetv.py

@@ -1,9 +1,6 @@
-from .common import InfoExtractor, SearchInfoExtractor
-from ..utils import (
-    traverse_obj,
-    unified_timestamp,
-    url_or_none,
-)
+from .common import InfoExtractor
+from ..utils import parse_iso8601, url_or_none
+from ..utils.traversal import traverse_obj
 
 
 class PiramideTVIE(InfoExtractor):
@@ -25,89 +22,55 @@ class PiramideTVIE(InfoExtractor):
         'url': 'https://piramide.tv/video/wcYn6li79NgN',
         'info_dict': {
             'id': 'wcYn6li79NgN',
+            'ext': 'mp4',
             'title': 'ACEPTO TENER UN BEBE CON MI NOVIA\u2026? | Parte 1',
             'description': '',
             'channel': 'ARTA GAME',
             'channel_id': 'arta_game',
+            'thumbnail': 'https://cdn.jwplayer.com/v2/media/cnEdGp5X/thumbnails/rHAaWfP7.jpg',
+            'timestamp': 1703434976,
+            'upload_date': '20231224',
         },
-        'playlist_count': 4,
     }]
 
     def _real_extract(self, url):
-        def extract_video(video_id, fatal=False):
-            video_data = self._download_json(f'https://hermes.piramide.tv/video/data/{video_id}',
-                                             video_id, fatal=fatal)
-            formats, subtitles = self._extract_m3u8_formats_and_subtitles(
-                f'https://cdn.piramide.tv/video/{video_id}/manifest.m3u8', video_id, fatal=fatal)
-            video_info = {**traverse_obj(video_data, {
-                'id': ('video', 'id', {str}),
-                'title': ('video', 'title', {str}),
-                'description': ('video', 'description', {str}),
-                'thumbnail': ('video', 'media', 'thumbnail', {url_or_none}),
-                'channel': ('video', 'channel', 'name', {str}),
-                'channel_id': ('video', 'channel', 'id', {str}),
-                'timestamp': ('video', 'date', {unified_timestamp}),
-            }),
-                'formats': formats,
-                'subtitles': subtitles,
-                'webpage_url': f'https://piramide.tv/video/{video_id}',
-                'webpage_url_basename': video_id,
-            }
-            next_video_id = traverse_obj(video_data, ('video', 'next_video', 'id', {str}))
-            return video_info, next_video_id
-
         video_id = self._match_id(url)
-        entries = []
-        while video_id is not None:
-            video, next_video = extract_video(video_id, (not entries))
-            if video.get('formats'):
-                entries.append(video)
-            video_id = next_video if next_video != video_id else None
-
-        if len(entries) == 1:
-            return entries[0]
-        elif len(entries) > 1:
-            return self.playlist_result(entries, **traverse_obj(entries[0], {
-                'id': ('id'),
-                'title': ('title'),
-                'description': ('description'),
-                'channel': ('channel'),
-                'channel_id': ('channel_id'),
-            }))
-        else:
-            return {'id': video_id}
+        video_data = self._download_json(
+            f'https://hermes.piramide.tv/video/data/{video_id}', video_id, fatal=False)
+        formats, subtitles = self._extract_m3u8_formats_and_subtitles(
+            f'https://cdn.piramide.tv/video/{video_id}/manifest.m3u8', video_id, fatal=False)
+        return {
+            'id': video_id,
+            'formats': formats,
+            'subtitles': subtitles,
+            **traverse_obj(video_data, ('video', {
+                'id': ('id', {str}),
+                'title': ('title', {str}),
+                'description': ('description', {str}),
+                'thumbnail': ('media', 'thumbnail', {url_or_none}),
+                'channel': ('channel', 'name', {str}),
+                'channel_id': ('channel', 'id', {str}),
+                'timestamp': ('date', {parse_iso8601}),
+            })),
+        }
 
 
-class PiramideTVChannelURLIE(InfoExtractor):
+class PiramideTVChannelIE(InfoExtractor):
     _VALID_URL = r'https?://piramide\.tv/channel/(?P<id>[\w-]+)'
     _TESTS = [{
         'url': 'https://piramide.tv/channel/thekalo',
-        'playlist_count': 10,
+        'playlist_mincount': 10,
         'info_dict': {
             'id': 'thekalo',
-            'title': 'thekalo',
         },
     }]
 
-    def _real_extract(self, url):
-        if query := self._match_id(url):
-            return self.url_result(url=f'piramidetvall:{query}', url_transparent=True)
-
+    def _entries(self, channel_name):
+        videos = self._download_json(
+            f'https://hermes.piramide.tv/channel/list/{channel_name}/date/100000', channel_name)
+        for video_id in traverse_obj(videos, ('videos', ..., 'id', {str})):
+            yield self.url_result(f'https://piramide.tv/video/{video_id}')
 
-class PiramideTVChannelIE(SearchInfoExtractor):
-    IE_NAME = 'PiramideTV:channel'
-    _SEARCH_KEY = 'piramidetv'
-    _TESTS = [{
-        'url': 'piramidetv5:bobicraft',
-        'playlist_count': 5,
-        'info_dict': {
-            'id': 'bobicraft',
-            'title': 'bobicraft',
-        },
-    }]
-
-    def _search_results(self, query):
-        videos = self._download_json(f'https://hermes.piramide.tv/channel/list/{query}/date/100000', query)
-        for video in videos.get('videos'):
-            if video_id := video.get('id'):
-                yield self.url_result(f'https://piramide.tv/video/{video_id}')
+    def _real_extract(self, url):
+        channel_name = self._match_id(url)
+        return self.playlist_result(self._entries(channel_name), channel_name)