rename AllstarBase to AllstarBaseIE

Co-authored-by: sepro <4618135+seproDev@users.noreply.github.com>

yt_dlp/extractor/allstar.py

@@ -59,53 +59,35 @@ _QUERIES = {
     }''' % (_FIELDS, _EXTRA_FIELDS),
 }
 
-_PAGESIZE = 10
 
+class AllstarBaseIE(InfoExtractor):
+    @staticmethod
+    def _parse_video_data(video_data):
+        def _media_url_or_none(path):
+            return urljoin('https://media.allstar.gg/', str_or_none(path))
 
-def _media_url_or_none(path):
-    return urljoin('https://media.allstar.gg/', str_or_none(path))
+        def _profile_url_or_none(path):
+            return urljoin('https:/allstar.gg/u/', str_or_none(path))
 
+        return traverse_obj(video_data, {
+            'id': ('_id', {str_or_none}),
+            'display_id': ('shareId', {str_or_none}),
+            'title': ('clipTitle', {str_or_none}),
+            'url': ('clipLink', {_media_url_or_none}),
+            'thumbnail': ('clipImageThumb', {_media_url_or_none}),
+            'duration': ('clipLength', {int_or_none}),
+            'filesize': ('clipSizeBytes', {int_or_none}),
+            'timestamp': ('createdDate', {int_or_none}),
+            'uploader': ('username', {str_or_none}),
+            'uploader_id': ('user', '_id', {str_or_none}),
+            'uploader_url': ('user', '_id', {_profile_url_or_none}),
+            'view_count': ('views', {int_or_none}),
+            'categories': ('game', {str_or_none}),
+        })
 
-def _profile_url_or_none(path):
-    return urljoin('https:/allstar.gg/u/', str_or_none(path))
-
-
-def _parse_video_data(video_data):
-    return traverse_obj(video_data, {
-        'id': ('_id', {str_or_none}),
-        'display_id': ('shareId', {str_or_none}),
-        'title': ('clipTitle', {str_or_none}),
-        'url': ('clipLink', {_media_url_or_none}),
-        'thumbnail': ('clipImageThumb', {_media_url_or_none}),
-        'duration': ('clipLength', {int_or_none}),
-        'filesize': ('clipSizeBytes', {int_or_none}),
-        'timestamp': ('createdDate', {int_or_none}),
-        'uploader': ('username', {str_or_none}),
-        'uploader_id': ('user', '_id', {str_or_none}),
-        'uploader_url': ('user', '_id', {_profile_url_or_none}),
-        'view_count': ('views', {int_or_none}),
-        'categories': ('game', {str_or_none}),
-    })
-
-
-def _set_webpage_url(info_dict):
-    video_id = info_dict.get('id')
-    video_url = info_dict.get('url')
-
-    if video_url is None or video_id is None:
-        return info_dict
-
-    base_name = 'clip' if '/clips/' in video_url else 'montage'
-    info_dict['webpage_url'] = f'https://allstar.gg/{base_name}?{base_name}={video_id}'
-    info_dict['webpage_url_basename'] = base_name
-
-    return info_dict
-
-
-class AllstarBase(InfoExtractor):
-    def _send_query(self, query, variables={}, path=(), video_id=None):
+    def _send_query(self, query, variables={}, path=(), video_id=None, note=None):
         response = self._download_json(
-            'https://a1.allstar.gg/graphql', video_id,
+            'https://a1.allstar.gg/graphql', video_id, note=note,
             headers={'content-type': 'application/json'},
             data=json.dumps({'variables': variables, 'query': query}).encode())
 
@@ -116,7 +98,7 @@ class AllstarBase(InfoExtractor):
         return traverse_obj(response, path)
 
 
-class AllstarIE(AllstarBase):
+class AllstarIE(AllstarBaseIE):
     _VALID_URL = r'https?://(?:www\.)?allstar\.gg/(?P<type>(?:clip|montage))\?(?P=type)=(?P<id>[^/?#&]+)'
 
     _TESTS = [{
@@ -187,13 +169,13 @@ class AllstarIE(AllstarBase):
 
         assert query_id in _QUERIES
 
-        return _parse_video_data(
+        return self._parse_video_data(
             self._send_query(
                 _QUERIES.get(query_id), {'id': video_id},
                 ('data', 'video'), video_id))
 
 
-class AllstarProfileIE(AllstarBase):
+class AllstarProfileIE(AllstarBaseIE):
     _VALID_URL = r'https?://(?:www\.)?allstar\.gg/(?:profile\?user=|u/)(?P<id>[^/?#&]+)'
 
     _TESTS = [{
@@ -226,6 +208,22 @@ class AllstarProfileIE(AllstarBase):
         'playlist_mincount': 1
     }]
 
+    _PAGE_SIZE = 10
+
+    @staticmethod
+    def _set_webpage_url(info_dict):
+        video_id = info_dict.get('id')
+        video_url = info_dict.get('url')
+
+        if video_url is None or video_id is None:
+            return info_dict
+
+        base_name = 'clip' if '/clips/' in video_url else 'montage'
+        info_dict['webpage_url'] = f'https://allstar.gg/{base_name}?{base_name}={video_id}'
+        info_dict['webpage_url_basename'] = base_name
+
+        return info_dict
+
     def _get_page(self, user_id, game, query_id, page_num):
         page_num += 1
 
@@ -234,8 +232,8 @@ class AllstarProfileIE(AllstarBase):
                     'user': user_id,
                     'page': page_num,
                     'game': int_or_none(game),
-                }, ('data', 'videos', 'data'), f'{user_id} page {page_num}'):
-            yield _set_webpage_url(_parse_video_data(video_data))
+                }, ('data', 'videos', 'data'), user_id, f'Downloading page {page_num}'):
+            yield self._set_webpage_url(self._parse_video_data(video_data))
 
     def _get_user_data(self, user_id, path=()):
         return traverse_obj(
@@ -253,12 +251,12 @@ class AllstarProfileIE(AllstarBase):
         query_id = view.lower()
 
         if query_id not in ('clips', 'montages', 'mobile clips'):
-            raise UnsupportedError(f'view={view}')
+            raise UnsupportedError(url)
 
         assert query_id in _QUERIES
 
         return self.playlist_result(
             OnDemandPagedList(
                 functools.partial(
-                    self._get_page, user_id, game, query_id), _PAGESIZE),
+                    self._get_page, user_id, game, query_id), self._PAGE_SIZE),
             user_id, f'{user_name} - {view}')