]> jfr.im git - yt-dlp.git/blame - yt_dlp/extractor/lbry.py
[cleanup] Fix infodict returned fields (#8906)
[yt-dlp.git] / yt_dlp / extractor / lbry.py
CommitLineData
29f7c58a 1import functools
8bdd16b4 2import json
127a2246 3import re
44c0d664 4import urllib.parse
8bdd16b4 5
6from .common import InfoExtractor
3d2623a8 7from ..networking import HEADRequest
8bdd16b4 8from ..utils import (
8bdd16b4 9 ExtractorError,
8a40bffa
M
10 OnDemandPagedList,
11 UnsupportedError,
12 determine_ext,
8bdd16b4 13 int_or_none,
14 mimetype2ext,
4dfbf869 15 parse_qs,
44c0d664 16 traverse_obj,
8bdd16b4 17 try_get,
44c0d664 18 url_or_none,
19 urlhandle_detect_ext,
29f7c58a 20 urljoin,
8bdd16b4 21)
22
23
29f7c58a 24class LBRYBaseIE(InfoExtractor):
48cceec1 25 _BASE_URL_REGEX = r'(?x)(?:https?://(?:www\.)?(?:lbry\.tv|odysee\.com)/|lbry://)'
29f7c58a 26 _CLAIM_ID_REGEX = r'[0-9a-f]{1,40}'
48cceec1 27 _OPT_CLAIM_ID = '[^$@:/?#&]+(?:[:#]%s)?' % _CLAIM_ID_REGEX
29f7c58a 28 _SUPPORTED_STREAM_TYPES = ['video', 'audio']
48cceec1 29 _PAGE_SIZE = 50
29f7c58a 30
31 def _call_api_proxy(self, method, display_id, params, resource):
59a0c358
M
32 headers = {'Content-Type': 'application/json-rpc'}
33 token = try_get(self._get_cookies('https://odysee.com'), lambda x: x['auth_token'].value)
34 if token:
35 headers['x-lbry-auth-token'] = token
8100c772 36 response = self._download_json(
29f7c58a 37 'https://api.lbry.tv/api/v1/proxy',
38 display_id, 'Downloading %s JSON metadata' % resource,
59a0c358 39 headers=headers,
29f7c58a 40 data=json.dumps({
41 'method': method,
42 'params': params,
8100c772 43 }).encode())
44 err = response.get('error')
45 if err:
46 raise ExtractorError(
47 f'{self.IE_NAME} said: {err.get("code")} - {err.get("message")}', expected=True)
48 return response['result']
29f7c58a 49
50 def _resolve_url(self, url, display_id, resource):
51 return self._call_api_proxy(
52 'resolve', display_id, {'urls': url}, resource)[url]
53
54 def _permanent_url(self, url, claim_name, claim_id):
57db6a87 55 return urljoin(
56 url.replace('lbry://', 'https://lbry.tv/'),
57 '/%s:%s' % (claim_name, claim_id))
29f7c58a 58
59 def _parse_stream(self, stream, url):
44c0d664 60 stream_type = traverse_obj(stream, ('value', 'stream_type', {str}))
61
62 info = traverse_obj(stream, {
63 'title': ('value', 'title', {str}),
64 'thumbnail': ('value', 'thumbnail', 'url', {url_or_none}),
65 'description': ('value', 'description', {str}),
66 'license': ('value', 'license', {str}),
67 'timestamp': ('timestamp', {int_or_none}),
68 'release_timestamp': ('value', 'release_time', {int_or_none}),
69 'tags': ('value', 'tags', ..., {lambda x: x or None}),
70 'duration': ('value', stream_type, 'duration', {int_or_none}),
71 'channel': ('signing_channel', 'value', 'title', {str}),
72 'channel_id': ('signing_channel', 'claim_id', {str}),
0e722f2f 73 'uploader_id': ('signing_channel', 'name', {str}),
44c0d664 74 })
75
0e722f2f
RD
76 if info.get('uploader_id') and info.get('channel_id'):
77 info['channel_url'] = self._permanent_url(url, info['uploader_id'], info['channel_id'])
29f7c58a 78
29f7c58a 79 return info
80
48cceec1
RD
81 def _fetch_page(self, display_id, url, params, page):
82 page += 1
83 page_params = {
84 'no_totals': True,
85 'page': page,
86 'page_size': self._PAGE_SIZE,
87 **params,
88 }
89 result = self._call_api_proxy(
90 'claim_search', display_id, page_params, f'page {page}')
91 for item in traverse_obj(result, ('items', lambda _, v: v['name'] and v['claim_id'])):
92 yield {
93 **self._parse_stream(item, url),
94 '_type': 'url',
95 'id': item['claim_id'],
96 'url': self._permanent_url(url, item['name'], item['claim_id']),
97 }
98
99 def _playlist_entries(self, url, display_id, claim_param, metadata):
100 qs = parse_qs(url)
101 content = qs.get('content', [None])[0]
102 params = {
103 'fee_amount': qs.get('fee_amount', ['>=0'])[0],
104 'order_by': {
105 'new': ['release_time'],
106 'top': ['effective_amount'],
107 'trending': ['trending_group', 'trending_mixed'],
108 }[qs.get('order', ['new'])[0]],
109 'claim_type': 'stream',
110 'stream_types': [content] if content in ['audio', 'video'] else self._SUPPORTED_STREAM_TYPES,
111 **claim_param,
112 }
113 duration = qs.get('duration', [None])[0]
114 if duration:
115 params['duration'] = {
116 'long': '>=1200',
117 'short': '<=240',
118 }[duration]
119 language = qs.get('language', ['all'])[0]
120 if language != 'all':
121 languages = [language]
122 if language == 'en':
123 languages.append('none')
124 params['any_languages'] = languages
125
126 entries = OnDemandPagedList(
127 functools.partial(self._fetch_page, display_id, url, params),
128 self._PAGE_SIZE)
129
130 return self.playlist_result(
131 entries, display_id, **traverse_obj(metadata, ('value', {
132 'title': 'title',
133 'description': 'description',
134 })))
135
29f7c58a 136
137class LBRYIE(LBRYBaseIE):
138 IE_NAME = 'lbry'
48cceec1
RD
139 _VALID_URL = LBRYBaseIE._BASE_URL_REGEX + rf'''
140 (?:\$/(?:download|embed)/)?
141 (?P<id>
142 [^$@:/?#]+/{LBRYBaseIE._CLAIM_ID_REGEX}
143 |(?:@{LBRYBaseIE._OPT_CLAIM_ID}/)?{LBRYBaseIE._OPT_CLAIM_ID}
144 )'''
8bdd16b4 145 _TESTS = [{
146 # Video
147 'url': 'https://lbry.tv/@Mantega:1/First-day-LBRY:1',
127a2246 148 'md5': '65bd7ec1f6744ada55da8e4c48a2edf9',
8bdd16b4 149 'info_dict': {
150 'id': '17f983b61f53091fb8ea58a9c56804e4ff8cff4d',
151 'ext': 'mp4',
152 'title': 'First day in LBRY? Start HERE!',
153 'description': 'md5:f6cb5c704b332d37f5119313c2c98f51',
154 'timestamp': 1595694354,
155 'upload_date': '20200725',
10db0d2f 156 'release_timestamp': 1595340697,
157 'release_date': '20200721',
29f7c58a 158 'width': 1280,
159 'height': 720,
8a40bffa
M
160 'thumbnail': 'https://spee.ch/7/67f2d809c263288c.png',
161 'license': 'None',
0e722f2f 162 'uploader_id': '@Mantega',
8a40bffa
M
163 'duration': 346,
164 'channel': 'LBRY/Odysee rats united!!!',
165 'channel_id': '1c8ad6a2ab4e889a71146ae4deeb23bb92dab627',
166 'channel_url': 'https://lbry.tv/@Mantega:1c8ad6a2ab4e889a71146ae4deeb23bb92dab627',
167 'tags': [
168 'first day in lbry',
169 'lbc',
170 'lbry',
171 'start',
172 'tutorial'
173 ],
8bdd16b4 174 }
175 }, {
176 # Audio
177 'url': 'https://lbry.tv/@LBRYFoundation:0/Episode-1:e',
178 'md5': 'c94017d3eba9b49ce085a8fad6b98d00',
179 'info_dict': {
180 'id': 'e7d93d772bd87e2b62d5ab993c1c3ced86ebb396',
181 'ext': 'mp3',
182 'title': 'The LBRY Foundation Community Podcast Episode 1 - Introduction, Streaming on LBRY, Transcoding',
183 'description': 'md5:661ac4f1db09f31728931d7b88807a61',
184 'timestamp': 1591312601,
185 'upload_date': '20200604',
10db0d2f 186 'release_timestamp': 1591312421,
187 'release_date': '20200604',
29f7c58a 188 'tags': list,
189 'duration': 2570,
190 'channel': 'The LBRY Foundation',
191 'channel_id': '0ed629d2b9c601300cacf7eabe9da0be79010212',
192 'channel_url': 'https://lbry.tv/@LBRYFoundation:0ed629d2b9c601300cacf7eabe9da0be79010212',
193 'vcodec': 'none',
8a40bffa
M
194 'thumbnail': 'https://spee.ch/d/0bc63b0e6bf1492d.png',
195 'license': 'None',
0e722f2f 196 'uploader_id': '@LBRYFoundation',
8bdd16b4 197 }
201c1459 198 }, {
201c1459 199 'url': 'https://odysee.com/@gardeningincanada:b/plants-i-will-never-grow-again.-the:e',
127a2246 200 'md5': 'c35fac796f62a14274b4dc2addb5d0ba',
201c1459 201 'info_dict': {
202 'id': 'e51671357333fe22ae88aad320bde2f6f96b1410',
203 'ext': 'mp4',
204 'title': 'PLANTS I WILL NEVER GROW AGAIN. THE BLACK LIST PLANTS FOR A CANADIAN GARDEN | Gardening in Canada 🍁',
205 'description': 'md5:9c539c6a03fb843956de61a4d5288d5e',
206 'timestamp': 1618254123,
207 'upload_date': '20210412',
208 'release_timestamp': 1618254002,
209 'release_date': '20210412',
210 'tags': list,
211 'duration': 554,
212 'channel': 'Gardening In Canada',
213 'channel_id': 'b8be0e93b423dad221abe29545fbe8ec36e806bc',
214 'channel_url': 'https://odysee.com/@gardeningincanada:b8be0e93b423dad221abe29545fbe8ec36e806bc',
0e722f2f
RD
215 'uploader_id': '@gardeningincanada',
216 'formats': 'mincount:3',
8a40bffa
M
217 'thumbnail': 'https://thumbnails.lbry.com/AgHSc_HzrrE',
218 'license': 'Copyrighted (contact publisher)',
201c1459 219 }
59a0c358
M
220 }, {
221 # HLS live stream (might expire)
222 'url': 'https://odysee.com/@RT:fd/livestream_RT:d',
223 'info_dict': {
224 'id': 'fdd11cb3ab75f95efb7b3bc2d726aa13ac915b66',
225 'ext': 'mp4',
226 'live_status': 'is_live',
227 'title': 'startswith:RT News | Livestream 24/7',
228 'description': 'md5:fe68d0056dfe79c1a6b8ce8c34d5f6fa',
229 'timestamp': int,
230 'upload_date': str,
231 'release_timestamp': int,
232 'release_date': str,
233 'tags': list,
59a0c358
M
234 'channel': 'RT',
235 'channel_id': 'fdd11cb3ab75f95efb7b3bc2d726aa13ac915b66',
236 'channel_url': 'https://odysee.com/@RT:fdd11cb3ab75f95efb7b3bc2d726aa13ac915b66',
237 'formats': 'mincount:1',
238 'thumbnail': 'startswith:https://thumb',
239 'license': 'None',
0e722f2f 240 'uploader_id': '@RT',
59a0c358
M
241 },
242 'params': {'skip_download': True}
44c0d664 243 }, {
244 # original quality format w/higher resolution than HLS formats
245 'url': 'https://odysee.com/@wickedtruths:2/Biotechnological-Invasion-of-Skin-(April-2023):4',
246 'md5': '305b0b3b369bde1b984961f005b67193',
247 'info_dict': {
248 'id': '41fbfe805eb73c8d3012c0c49faa0f563274f634',
249 'ext': 'mp4',
250 'title': 'Biotechnological Invasion of Skin (April 2023)',
48cceec1 251 'description': 'md5:fe28689db2cb7ba3436d819ac3ffc378',
44c0d664 252 'channel': 'Wicked Truths',
253 'channel_id': '23d2bbf856b0ceed5b1d7c5960bcc72da5a20cb0',
254 'channel_url': 'https://odysee.com/@wickedtruths:23d2bbf856b0ceed5b1d7c5960bcc72da5a20cb0',
0e722f2f 255 'uploader_id': '@wickedtruths',
48cceec1
RD
256 'timestamp': 1695114347,
257 'upload_date': '20230919',
44c0d664 258 'release_timestamp': 1685617473,
259 'release_date': '20230601',
260 'duration': 1063,
261 'thumbnail': 'https://thumbs.odycdn.com/4e6d39da4df0cfdad45f64e253a15959.webp',
262 'tags': ['smart skin surveillance', 'biotechnology invasion of skin', 'morgellons'],
263 'license': 'None',
264 'protocol': 'https', # test for direct mp4 download
265 },
8bdd16b4 266 }, {
267 'url': 'https://odysee.com/@BrodieRobertson:5/apple-is-tracking-everything-you-do-on:e',
268 'only_matching': True,
70c5802b 269 }, {
8a40bffa 270 'url': 'https://odysee.com/@ScammerRevolts:b0/I-SYSKEY\'D-THE-SAME-SCAMMERS-3-TIMES!:b',
70c5802b 271 'only_matching': True,
29f7c58a 272 }, {
273 'url': 'https://lbry.tv/Episode-1:e7d93d772bd87e2b62d5ab993c1c3ced86ebb396',
274 'only_matching': True,
275 }, {
276 'url': 'https://lbry.tv/$/embed/Episode-1/e7d93d772bd87e2b62d5ab993c1c3ced86ebb396',
277 'only_matching': True,
278 }, {
279 'url': 'https://lbry.tv/Episode-1:e7',
280 'only_matching': True,
281 }, {
282 'url': 'https://lbry.tv/@LBRYFoundation/Episode-1',
283 'only_matching': True,
284 }, {
285 'url': 'https://lbry.tv/$/download/Episode-1/e7d93d772bd87e2b62d5ab993c1c3ced86ebb396',
286 'only_matching': True,
a820dc72
RA
287 }, {
288 'url': 'https://lbry.tv/@lacajadepandora:a/TRUMP-EST%C3%81-BIEN-PUESTO-con-Pilar-Baselga,-Carlos-Senra,-Luis-Palacios-(720p_30fps_H264-192kbit_AAC):1',
289 'only_matching': True,
57db6a87 290 }, {
291 'url': 'lbry://@lbry#3f/odysee#7',
292 'only_matching': True,
8bdd16b4 293 }]
294
8bdd16b4 295 def _real_extract(self, url):
29f7c58a 296 display_id = self._match_id(url)
48cceec1 297 if display_id.startswith('@'):
29f7c58a 298 display_id = display_id.replace(':', '#')
48cceec1
RD
299 else:
300 display_id = display_id.replace('/', ':')
44c0d664 301 display_id = urllib.parse.unquote(display_id)
8bdd16b4 302 uri = 'lbry://' + display_id
29f7c58a 303 result = self._resolve_url(uri, display_id, 'stream')
59a0c358 304 headers = {'Referer': 'https://odysee.com/'}
44c0d664 305
306 formats = []
307 stream_type = traverse_obj(result, ('value', 'stream_type', {str}))
308
309 if stream_type in self._SUPPORTED_STREAM_TYPES:
59a0c358 310 claim_id, is_live = result['claim_id'], False
c25de59c
AG
311 streaming_url = self._call_api_proxy(
312 'get', claim_id, {'uri': uri}, 'streaming url')['streaming_url']
44c0d664 313
127a2246 314 # GET request to v3 API returns original video/audio file if available
315 direct_url = re.sub(r'/api/v\d+/', '/api/v3/', streaming_url)
6148833f 316 urlh = self._request_webpage(
317 direct_url, display_id, 'Checking for original quality', headers=headers, fatal=False)
318 if urlh and urlhandle_detect_ext(urlh) != 'm3u8':
44c0d664 319 formats.append({
127a2246 320 'url': direct_url,
44c0d664 321 'format_id': 'original',
322 'quality': 1,
323 **traverse_obj(result, ('value', {
324 'ext': ('source', (('name', {determine_ext}), ('media_type', {mimetype2ext}))),
325 'filesize': ('source', 'size', {int_or_none}),
326 'width': ('video', 'width', {int_or_none}),
327 'height': ('video', 'height', {int_or_none}),
328 }), get_all=False),
329 'vcodec': 'none' if stream_type == 'audio' else None,
330 })
331
332 # HEAD request returns redirect response to m3u8 URL if available
c25de59c 333 final_url = self._request_webpage(
59a0c358 334 HEADRequest(streaming_url), display_id, headers=headers,
3d2623a8 335 note='Downloading streaming redirect url info').url
44c0d664 336
c25de59c
AG
337 elif result.get('value_type') == 'stream':
338 claim_id, is_live = result['signing_channel']['claim_id'], True
c25de59c 339 live_data = self._download_json(
9fde8a6b
M
340 'https://api.odysee.live/livestream/is_live', claim_id,
341 query={'channel_claim_id': claim_id},
c25de59c 342 note='Downloading livestream JSON metadata')['data']
44c0d664 343 final_url = live_data.get('VideoURL')
59a0c358
M
344 # Upcoming videos may still give VideoURL
345 if not live_data.get('Live'):
44c0d664 346 final_url = None
3a4bb9f7 347 self.raise_no_formats('This stream is not live', True, claim_id)
44c0d664 348
c25de59c 349 else:
3a4bb9f7 350 raise UnsupportedError(url)
c25de59c 351
c25de59c 352 if determine_ext(final_url) == 'm3u8':
44c0d664 353 formats.extend(self._extract_m3u8_formats(
354 final_url, display_id, 'mp4', m3u8_id='hls', live=is_live, headers=headers))
355
c25de59c 356 return {
44c0d664 357 **self._parse_stream(result, url),
29f7c58a 358 'id': claim_id,
44c0d664 359 'formats': formats,
c25de59c
AG
360 'is_live': is_live,
361 'http_headers': headers,
362 }
29f7c58a 363
364
365class LBRYChannelIE(LBRYBaseIE):
366 IE_NAME = 'lbry:channel'
48cceec1 367 _VALID_URL = LBRYBaseIE._BASE_URL_REGEX + rf'(?P<id>@{LBRYBaseIE._OPT_CLAIM_ID})/?(?:[?&]|$)'
29f7c58a 368 _TESTS = [{
369 'url': 'https://lbry.tv/@LBRYFoundation:0',
370 'info_dict': {
371 'id': '0ed629d2b9c601300cacf7eabe9da0be79010212',
372 'title': 'The LBRY Foundation',
373 'description': 'Channel for the LBRY Foundation. Follow for updates and news.',
374 },
8a40bffa 375 'playlist_mincount': 29,
29f7c58a 376 }, {
377 'url': 'https://lbry.tv/@LBRYFoundation',
378 'only_matching': True,
57db6a87 379 }, {
380 'url': 'lbry://@lbry#3f',
381 'only_matching': True,
29f7c58a 382 }]
29f7c58a 383
384 def _real_extract(self, url):
385 display_id = self._match_id(url).replace(':', '#')
48cceec1 386 result = self._resolve_url(f'lbry://{display_id}', display_id, 'channel')
29f7c58a 387 claim_id = result['claim_id']
48cceec1
RD
388
389 return self._playlist_entries(url, claim_id, {'channel_ids': [claim_id]}, result)
390
391
392class LBRYPlaylistIE(LBRYBaseIE):
393 IE_NAME = 'lbry:playlist'
394 _VALID_URL = LBRYBaseIE._BASE_URL_REGEX + r'\$/(?:play)?list/(?P<id>[0-9a-f-]+)'
395 _TESTS = [{
396 'url': 'https://odysee.com/$/playlist/ffef782f27486f0ac138bde8777f72ebdd0548c2',
397 'info_dict': {
398 'id': 'ffef782f27486f0ac138bde8777f72ebdd0548c2',
399 'title': 'Théâtre Classique',
400 'description': 'Théâtre Classique',
401 },
402 'playlist_mincount': 4,
403 }, {
404 'url': 'https://odysee.com/$/list/9c6658b3dd21e4f2a0602d523a13150e2b48b770',
405 'info_dict': {
406 'id': '9c6658b3dd21e4f2a0602d523a13150e2b48b770',
407 'title': 'Social Media Exposed',
408 'description': 'md5:98af97317aacd5b85d595775ea37d80e',
409 },
410 'playlist_mincount': 34,
411 }, {
412 'url': 'https://odysee.com/$/playlist/938fb11d-215f-4d1c-ad64-723954df2184',
413 'info_dict': {
414 'id': '938fb11d-215f-4d1c-ad64-723954df2184',
415 },
416 'playlist_mincount': 1000,
417 }]
418
419 def _real_extract(self, url):
420 display_id = self._match_id(url)
421 result = traverse_obj(self._call_api_proxy('claim_search', display_id, {
422 'claim_ids': [display_id],
423 'no_totals': True,
424 'page': 1,
425 'page_size': self._PAGE_SIZE,
426 }, 'playlist'), ('items', 0))
427 claim_param = {'claim_ids': traverse_obj(result, ('value', 'claims', ..., {str}))}
428
429 return self._playlist_entries(url, display_id, claim_param, result)