mirror of
				https://github.com/yt-dlp/yt-dlp.git
				synced 2025-11-04 00:25:15 +00:00 
			
		
		
		
	Closes #3400 Authored by: ischmidt20, bashonly Co-authored-by: bashonly <88596187+bashonly@users.noreply.github.com>
		
			
				
	
	
		
			162 lines
		
	
	
		
			7.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			162 lines
		
	
	
		
			7.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
import re
 | 
						|
import urllib.parse
 | 
						|
 | 
						|
from .turner import TurnerBaseIE
 | 
						|
from ..utils import (
 | 
						|
    float_or_none,
 | 
						|
    int_or_none,
 | 
						|
    make_archive_id,
 | 
						|
    strip_or_none,
 | 
						|
)
 | 
						|
from ..utils.traversal import traverse_obj
 | 
						|
 | 
						|
 | 
						|
class TBSIE(TurnerBaseIE):
 | 
						|
    _SITE_INFO = {
 | 
						|
        'tbs': ('TBS', 'eyJhbGciOiJSUzI1NiJ9.eyJzdWIiOiJkZTA0NTYxZS1iMTFhLTRlYTgtYTg5NC01NjI3MGM1NmM2MWIiLCJuYmYiOjE1MzcxODkzOTAsImlzcyI6ImF1dGguYWRvYmUuY29tIiwiaWF0IjoxNTM3MTg5MzkwfQ.Z7ny66kaqNDdCHf9Y9KsV12LrBxrLkGGxlYe2XGm6qsw2T-k1OCKC1TMzeqiZP735292MMRAQkcJDKrMIzNbAuf9nCdIcv4kE1E2nqUnjPMBduC1bHffZp8zlllyrN2ElDwM8Vhwv_5nElLRwWGEt0Kaq6KJAMZA__WDxKWC18T-wVtsOZWXQpDqO7nByhfj2t-Z8c3TUNVsA_wHgNXlkzJCZ16F2b7yGLT5ZhLPupOScd3MXC5iPh19HSVIok22h8_F_noTmGzmMnIRQi6bWYWK2zC7TQ_MsYHfv7V6EaG5m1RKZTV6JAwwoJQF_9ByzarLV1DGwZxD9-eQdqswvg'),
 | 
						|
        'tntdrama': ('TNT', 'eyJhbGciOiJSUzI1NiJ9.eyJzdWIiOiIwOTMxYTU4OS1jZjEzLTRmNjMtYTJmYy03MzhjMjE1NWU5NjEiLCJuYmYiOjE1MzcxOTA4MjcsImlzcyI6ImF1dGguYWRvYmUuY29tIiwiaWF0IjoxNTM3MTkwODI3fQ.AucKvtws7oekTXi80_zX4-BlgJD9GLvlOI9FlBCjdlx7Pa3eJ0AqbogynKMiatMbnLOTMHGjd7tTiq422unmZjBz70dhePAe9BbW0dIo7oQ57vZ-VBYw_tWYRPmON61MwAbLVlqROD3n_zURs85S8TlkQx9aNx9x_riGGELjd8l05CVa_pOluNhYvuIFn6wmrASOKI1hNEblBDWh468UWP571-fe4zzi0rlYeeHd-cjvtWvOB3bQsWrUVbK4pRmqvzEH59j0vNF-ihJF9HncmUicYONe47Mib3elfMok23v4dB1_UAlQY_oawfNcynmEnJQCcqFmbHdEwTW6gMiYsA'),
 | 
						|
        'trutv': ('truTV', 'eyJhbGciOiJSUzI1NiJ9.eyJzdWIiOiJhYzQyOTkwMi0xMDYzLTQyNTQtYWJlYS1iZTY2ODM4MTVmZGIiLCJuYmYiOjE1MzcxOTA4NjgsImlzcyI6ImF1dGguYWRvYmUuY29tIiwiaWF0IjoxNTM3MTkwODY4fQ.ewXl5LDMDvvx3nDXV4jCdSwUq_sOluKoOVsIjznAo6Zo4zrGe9rjlZ9DOmQKW66g6VRMexJsJ5vM1EkY8TC5-YcQw_BclK1FPGO1rH3Wf7tX_l0b1BVbSJQKIj9UgqDp_QbGcBXz24kN4So3U22mhs6di9PYyyfG68ccKL2iRprcVKWCslIHwUF-T7FaEqb0K57auilxeW1PONG2m-lIAcZ62DUwqXDWvw0CRoWI08aVVqkkhnXaSsQfLs5Ph1Pfh9Oq3g_epUm9Ss45mq6XM7gbOb5omTcKLADRKK-PJVB_JXnZnlsXbG0ttKE1cTKJ738qu7j4aipYTf-W0nKF5Q'),
 | 
						|
    }
 | 
						|
    _VALID_URL = fr'''(?x)
 | 
						|
        https?://(?:www\.)?(?P<site>{"|".join(map(re.escape, _SITE_INFO))})\.com
 | 
						|
        (?P<path>/(?:
 | 
						|
            (?P<watch>watch(?:tnt|tbs|trutv))|
 | 
						|
            movies|shows/[^/?#]+/(?:clips|season-\d+/episode-\d+)
 | 
						|
        )/(?P<id>[^/?#]+))
 | 
						|
    '''
 | 
						|
    _TESTS = [{
 | 
						|
        'url': 'https://www.tbs.com/shows/american-dad/season-6/episode-12/you-debt-your-life',
 | 
						|
        'info_dict': {
 | 
						|
            'id': '984bdcd8db0cc00dc699927f2a411c8c6e0e48f3',
 | 
						|
            'ext': 'mp4',
 | 
						|
            'title': 'You Debt Your Life',
 | 
						|
            'description': 'md5:f211cfeb9187fd3cdb53eb0e8930d499',
 | 
						|
            'duration': 1231.0,
 | 
						|
            'thumbnail': r're:https://images\.tbs\.com/tbs/.+\.(?:jpe?g|png)',
 | 
						|
            'chapters': 'count:4',
 | 
						|
            'season': 'Season 6',
 | 
						|
            'season_number': 6,
 | 
						|
            'episode': 'Episode 12',
 | 
						|
            'episode_number': 12,
 | 
						|
            'timestamp': 1478276239,
 | 
						|
            'upload_date': '20161104',
 | 
						|
        },
 | 
						|
        'params': {'skip_download': 'm3u8'},
 | 
						|
    }, {
 | 
						|
        'url': 'https://www.tntdrama.com/shows/the-librarians-the-next-chapter/season-1/episode-10/and-going-medieval',
 | 
						|
        'info_dict': {
 | 
						|
            'id': 'e487b31b663a8001864f62fd20907782f7b8ccb8',
 | 
						|
            'ext': 'mp4',
 | 
						|
            'title': 'And Going Medieval',
 | 
						|
            'description': 'md5:5aed0ae23a6cf148a02fe3c1be8359fa',
 | 
						|
            'duration': 2528.0,
 | 
						|
            'thumbnail': r're:https://images\.tntdrama\.com/tnt/.+\.(?:jpe?g|png)',
 | 
						|
            'chapters': 'count:7',
 | 
						|
            'season': 'Season 1',
 | 
						|
            'season_number': 1,
 | 
						|
            'episode': 'Episode 10',
 | 
						|
            'episode_number': 10,
 | 
						|
            'timestamp': 1743107520,
 | 
						|
            'upload_date': '20250327',
 | 
						|
        },
 | 
						|
        'params': {'skip_download': 'm3u8'},
 | 
						|
    }, {
 | 
						|
        'url': 'https://www.trutv.com/shows/the-carbonaro-effect/season-1/episode-1/got-the-bug-out',
 | 
						|
        'info_dict': {
 | 
						|
            'id': 'b457dd7458fd9e64b596355950b13a1ca799dc39',
 | 
						|
            'ext': 'mp4',
 | 
						|
            'title': 'Got the Bug Out',
 | 
						|
            'description': 'md5:9eeddf6248f73517b0e5969b8a43c025',
 | 
						|
            'duration': 1283.0,
 | 
						|
            'thumbnail': r're:https://images\.trutv\.com/tru/.+\.(?:jpe?g|png)',
 | 
						|
            'chapters': 'count:4',
 | 
						|
            'season': 'Season 1',
 | 
						|
            'season_number': 1,
 | 
						|
            'episode': 'Episode 1',
 | 
						|
            'episode_number': 1,
 | 
						|
            'timestamp': 1570040829,
 | 
						|
            'upload_date': '20191002',
 | 
						|
            '_old_archive_ids': ['trutv b457dd7458fd9e64b596355950b13a1ca799dc39'],
 | 
						|
        },
 | 
						|
        'params': {'skip_download': 'm3u8'},
 | 
						|
    }, {
 | 
						|
        'url': 'http://www.tntdrama.com/shows/the-alienist/clips/monster',
 | 
						|
        'only_matching': True,
 | 
						|
    }, {
 | 
						|
        'url': 'http://www.tbs.com/shows/search-party/season-1/episode-1/explicit-the-mysterious-disappearance-of-the-girl-no-one-knew',
 | 
						|
        'only_matching': True,
 | 
						|
    }, {
 | 
						|
        'url': 'http://www.tntdrama.com/movies/star-wars-a-new-hope',
 | 
						|
        'only_matching': True,
 | 
						|
    }, {
 | 
						|
        'url': 'https://www.trutv.com/shows/impractical-jokers/season-9/episode-1/you-dirty-dog',
 | 
						|
        'only_matching': True,
 | 
						|
    }, {
 | 
						|
        'url': 'https://www.trutv.com/watchtrutv/east',
 | 
						|
        'only_matching': True,
 | 
						|
    }, {
 | 
						|
        'url': 'https://www.tbs.com/watchtbs/east',
 | 
						|
        'only_matching': True,
 | 
						|
    }, {
 | 
						|
        'url': 'https://www.tntdrama.com/watchtnt/east',
 | 
						|
        'only_matching': True,
 | 
						|
    }]
 | 
						|
 | 
						|
    def _real_extract(self, url):
 | 
						|
        site, path, display_id, watch = self._match_valid_url(url).group('site', 'path', 'id', 'watch')
 | 
						|
        is_live = bool(watch)
 | 
						|
        webpage = self._download_webpage(url, display_id)
 | 
						|
        drupal_settings = self._search_json(
 | 
						|
            r'<script\b[^>]+\bdata-drupal-selector="drupal-settings-json"[^>]*>',
 | 
						|
            webpage, 'drupal settings', display_id)
 | 
						|
        video_data = next(v for v in drupal_settings['turner_playlist'] if is_live or v.get('url') == path)
 | 
						|
 | 
						|
        media_id = video_data['mediaID']
 | 
						|
        title = video_data['title']
 | 
						|
        tokenizer_query = urllib.parse.parse_qs(urllib.parse.urlparse(
 | 
						|
            drupal_settings['ngtv_token_url']).query)
 | 
						|
 | 
						|
        auth_info = traverse_obj(drupal_settings, ('top2', {dict})) or {}
 | 
						|
        site_name = auth_info.get('siteName') or self._SITE_INFO[site][0]
 | 
						|
        software_statement = auth_info.get('softwareStatement') or self._SITE_INFO[site][1]
 | 
						|
 | 
						|
        info = self._extract_ngtv_info(
 | 
						|
            media_id, tokenizer_query, software_statement, {
 | 
						|
                'url': url,
 | 
						|
                'site_name': site_name,
 | 
						|
                'auth_required': video_data.get('authRequired') == '1' or is_live,
 | 
						|
                'is_live': is_live,
 | 
						|
            })
 | 
						|
 | 
						|
        thumbnails = []
 | 
						|
        for image_id, image in video_data.get('images', {}).items():
 | 
						|
            image_url = image.get('url')
 | 
						|
            if not image_url or image.get('type') != 'video':
 | 
						|
                continue
 | 
						|
            i = {
 | 
						|
                'id': image_id,
 | 
						|
                'url': image_url,
 | 
						|
            }
 | 
						|
            mobj = re.search(r'(\d+)x(\d+)', image_url)
 | 
						|
            if mobj:
 | 
						|
                i.update({
 | 
						|
                    'width': int(mobj.group(1)),
 | 
						|
                    'height': int(mobj.group(2)),
 | 
						|
                })
 | 
						|
            thumbnails.append(i)
 | 
						|
 | 
						|
        info.update({
 | 
						|
            'id': media_id,
 | 
						|
            'title': title,
 | 
						|
            'description': strip_or_none(video_data.get('descriptionNoTags') or video_data.get('shortDescriptionNoTags')),
 | 
						|
            'duration': float_or_none(video_data.get('duration')) or info.get('duration'),
 | 
						|
            'timestamp': int_or_none(video_data.get('created')),
 | 
						|
            'season_number': int_or_none(video_data.get('season')),
 | 
						|
            'episode_number': int_or_none(video_data.get('episode')),
 | 
						|
            'thumbnails': thumbnails,
 | 
						|
            'is_live': is_live,
 | 
						|
        })
 | 
						|
        if site == 'trutv':
 | 
						|
            info['_old_archive_ids'] = [make_archive_id(site, media_id)]
 | 
						|
        return info
 |