|  | @@ -0,0 +1,79 @@
 | 
	
		
			
				|  |  | +# coding: utf-8
 | 
	
		
			
				|  |  | +from __future__ import unicode_literals
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +from .common import InfoExtractor
 | 
	
		
			
				|  |  | +from ..utils import (
 | 
	
		
			
				|  |  | +    determine_ext,
 | 
	
		
			
				|  |  | +    int_or_none,
 | 
	
		
			
				|  |  | +    merge_dicts,
 | 
	
		
			
				|  |  | +    parse_iso8601,
 | 
	
		
			
				|  |  | +    T,
 | 
	
		
			
				|  |  | +    traverse_obj,
 | 
	
		
			
				|  |  | +    txt_or_none,
 | 
	
		
			
				|  |  | +    urljoin,
 | 
	
		
			
				|  |  | +)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +class CaffeineTVIE(InfoExtractor):
 | 
	
		
			
				|  |  | +    _VALID_URL = r'https?://(?:www\.)?caffeine\.tv/[^/]+/video/(?P<id>[0-9a-f-]+)'
 | 
	
		
			
				|  |  | +    _TESTS = [{
 | 
	
		
			
				|  |  | +        'url': 'https://www.caffeine.tv/TsuSurf/video/cffc0a00-e73f-11ec-8080-80017d29f26e',
 | 
	
		
			
				|  |  | +        'info_dict': {
 | 
	
		
			
				|  |  | +            'id': 'cffc0a00-e73f-11ec-8080-80017d29f26e',
 | 
	
		
			
				|  |  | +            'ext': 'mp4',
 | 
	
		
			
				|  |  | +            'title': 'GOOOOD MORNINNNNN #highlights',
 | 
	
		
			
				|  |  | +            'timestamp': 1654702180,
 | 
	
		
			
				|  |  | +            'upload_date': '20220608',
 | 
	
		
			
				|  |  | +            'uploader': 'TsuSurf',
 | 
	
		
			
				|  |  | +            'duration': 3145,
 | 
	
		
			
				|  |  | +            'age_limit': 17,
 | 
	
		
			
				|  |  | +        },
 | 
	
		
			
				|  |  | +        'params': {
 | 
	
		
			
				|  |  | +            'format': 'bestvideo',
 | 
	
		
			
				|  |  | +        },
 | 
	
		
			
				|  |  | +    }]
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +    def _real_extract(self, url):
 | 
	
		
			
				|  |  | +        video_id = self._match_id(url)
 | 
	
		
			
				|  |  | +        json_data = self._download_json(
 | 
	
		
			
				|  |  | +            'https://api.caffeine.tv/social/public/activity/' + video_id,
 | 
	
		
			
				|  |  | +            video_id)
 | 
	
		
			
				|  |  | +        broadcast_info = traverse_obj(json_data, ('broadcast_info', T(dict))) or {}
 | 
	
		
			
				|  |  | +        title = broadcast_info['broadcast_title']
 | 
	
		
			
				|  |  | +        video_url = broadcast_info['video_url']
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +        ext = determine_ext(video_url)
 | 
	
		
			
				|  |  | +        if ext == 'm3u8':
 | 
	
		
			
				|  |  | +            formats = self._extract_m3u8_formats(
 | 
	
		
			
				|  |  | +                video_url, video_id, 'mp4', entry_protocol='m3u8',
 | 
	
		
			
				|  |  | +                fatal=False)
 | 
	
		
			
				|  |  | +        else:
 | 
	
		
			
				|  |  | +            formats = [{'url': video_url}]
 | 
	
		
			
				|  |  | +        self._sort_formats(formats)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +        return merge_dicts({
 | 
	
		
			
				|  |  | +            'id': video_id,
 | 
	
		
			
				|  |  | +            'title': title,
 | 
	
		
			
				|  |  | +            'formats': formats,
 | 
	
		
			
				|  |  | +        }, traverse_obj(json_data, {
 | 
	
		
			
				|  |  | +            'uploader': ((None, 'user'), 'username'),
 | 
	
		
			
				|  |  | +        }, get_all=False), traverse_obj(json_data, {
 | 
	
		
			
				|  |  | +            'like_count': ('like_count', T(int_or_none)),
 | 
	
		
			
				|  |  | +            'view_count': ('view_count', T(int_or_none)),
 | 
	
		
			
				|  |  | +            'comment_count': ('comment_count', T(int_or_none)),
 | 
	
		
			
				|  |  | +            'tags': ('tags', Ellipsis, T(txt_or_none)),
 | 
	
		
			
				|  |  | +            'is_live': 'is_live',
 | 
	
		
			
				|  |  | +            'uploader': ('user', 'name'),
 | 
	
		
			
				|  |  | +        }), traverse_obj(broadcast_info, {
 | 
	
		
			
				|  |  | +            'duration': ('content_duration', T(int_or_none)),
 | 
	
		
			
				|  |  | +            'timestamp': ('broadcast_start_time', T(parse_iso8601)),
 | 
	
		
			
				|  |  | +            'thumbnail': ('preview_image_path', T(lambda u: urljoin(url, u))),
 | 
	
		
			
				|  |  | +            'age_limit': ('content_rating', T(lambda r: r and {
 | 
	
		
			
				|  |  | +                # assume Apple Store ratings [1]
 | 
	
		
			
				|  |  | +                # 1. https://en.wikipedia.org/wiki/Mobile_software_content_rating_system
 | 
	
		
			
				|  |  | +                'FOUR_PLUS': 0,
 | 
	
		
			
				|  |  | +                'NINE_PLUS': 9,
 | 
	
		
			
				|  |  | +                'TWELVE_PLUS': 12,
 | 
	
		
			
				|  |  | +                'SEVENTEEN_PLUS': 17,
 | 
	
		
			
				|  |  | +            }.get(r, 17))),
 | 
	
		
			
				|  |  | +        }))
 |