# coding: utf-8 from __future__ import unicode_literals import re from .common import InfoExtractor from ..compat import ( compat_urllib_parse_urlparse, compat_urlparse, ) from ..utils import ( ExtractorError, int_or_none, update_url_query, xpath_element, xpath_text, ) class AfreecaTVIE(InfoExtractor): IE_NAME = 'afreecatv' IE_DESC = 'afreecatv.com' _VALID_URL = r'''(?x) https?:// (?: (?:(?:live|afbbs|www)\.)?afreeca(?:tv)?\.com(?::\d+)? (?: /app/(?:index|read_ucc_bbs)\.cgi| /player/[Pp]layer\.(?:swf|html) )\?.*?\bnTitleNo=| vod\.afreecatv\.com/PLAYER/STATION/ ) (?P\d+) ''' _TESTS = [{ 'url': 'http://live.afreecatv.com:8079/app/index.cgi?szType=read_ucc_bbs&szBjId=dailyapril&nStationNo=16711924&nBbsNo=18605867&nTitleNo=36164052&szSkin=', 'md5': 'f72c89fe7ecc14c1b5ce506c4996046e', 'info_dict': { 'id': '36164052', 'ext': 'mp4', 'title': '데일리 에이프릴 요정들의 시상식!', 'thumbnail': 're:^https?://(?:video|st)img.afreecatv.com/.*$', 'uploader': 'dailyapril', 'uploader_id': 'dailyapril', 'upload_date': '20160503', } }, { 'url': 'http://afbbs.afreecatv.com:8080/app/read_ucc_bbs.cgi?nStationNo=16711924&nTitleNo=36153164&szBjId=dailyapril&nBbsNo=18605867', 'info_dict': { 'id': '36153164', 'title': "BJ유트루와 함께하는 '팅커벨 메이크업!'", 'thumbnail': 're:^https?://(?:video|st)img.afreecatv.com/.*$', 'uploader': 'dailyapril', 'uploader_id': 'dailyapril', }, 'playlist_count': 2, 'playlist': [{ 'md5': 'd8b7c174568da61d774ef0203159bf97', 'info_dict': { 'id': '36153164_1', 'ext': 'mp4', 'title': "BJ유트루와 함께하는 '팅커벨 메이크업!'", 'upload_date': '20160502', }, }, { 'md5': '58f2ce7f6044e34439ab2d50612ab02b', 'info_dict': { 'id': '36153164_2', 'ext': 'mp4', 'title': "BJ유트루와 함께하는 '팅커벨 메이크업!'", 'upload_date': '20160502', }, }], }, { 'url': 'http://www.afreecatv.com/player/Player.swf?szType=szBjId=djleegoon&nStationNo=11273158&nBbsNo=13161095&nTitleNo=36327652', 'only_matching': True, }, { 'url': 'http://vod.afreecatv.com/PLAYER/STATION/15055030', 'only_matching': True, }] @staticmethod def parse_video_key(key): video_key = {} m = re.match(r'^(?P\d{8})_\w+_(?P\d+)$', key) if m: video_key['upload_date'] = m.group('upload_date') video_key['part'] = m.group('part') return video_key def _real_extract(self, url): video_id = self._match_id(url) parsed_url = compat_urllib_parse_urlparse(url) info_url = compat_urlparse.urlunparse(parsed_url._replace( netloc='afbbs.afreecatv.com:8080', path='/api/video/get_video_info.php')) video_xml = self._download_xml( update_url_query(info_url, {'nTitleNo': video_id}), video_id) if xpath_element(video_xml, './track/video/file') is None: raise ExtractorError('Specified AfreecaTV video does not exist', expected=True) title = xpath_text(video_xml, './track/title', 'title') uploader = xpath_text(video_xml, './track/nickname', 'uploader') uploader_id = xpath_text(video_xml, './track/bj_id', 'uploader id') duration = int_or_none(xpath_text(video_xml, './track/duration', 'duration')) thumbnail = xpath_text(video_xml, './track/titleImage', 'thumbnail') entries = [] for i, video_file in enumerate(video_xml.findall('./track/video/file')): video_key = self.parse_video_key(video_file.get('key', '')) if not video_key: continue entries.append({ 'id': '%s_%s' % (video_id, video_key.get('part', i + 1)), 'title': title, 'upload_date': video_key.get('upload_date'), 'duration': int_or_none(video_file.get('duration')), 'url': video_file.text, }) info = { 'id': video_id, 'title': title, 'uploader': uploader, 'uploader_id': uploader_id, 'duration': duration, 'thumbnail': thumbnail, } if len(entries) > 1: info['_type'] = 'multi_video' info['entries'] = entries elif len(entries) == 1: info['url'] = entries[0]['url'] info['upload_date'] = entries[0].get('upload_date') else: raise ExtractorError( 'No files found for the specified AfreecaTV video, either' ' the URL is incorrect or the video has been made private.', expected=True) return info class AfreecaTVGlobalIE(AfreecaTVIE): IE_NAME = 'afreecatv:global' _VALID_URL = r'https?://(?:www\.)?afreeca\.tv/(?P\d+)(?:/v/(?P\d+))?' _TESTS = [{ 'url': 'http://afreeca.tv/36853014/v/58301', 'info_dict': { 'id': '58301', 'title': 'tryhard top100', 'uploader_id': '36853014', 'uploader': 'makgi Hearthstone Live!', }, 'playlist_count': 3, }] def _real_extract(self, url): channel_id, video_id = re.match(self._VALID_URL, url).groups() video_type = 'video' if video_id else 'live' query = { 'pt': 'view', 'bid': channel_id, } if video_id: query['vno'] = video_id video_data = self._download_json( 'http://api.afreeca.tv/%s/view_%s.php' % (video_type, video_type), video_id or channel_id, query=query)['channel'] if video_data.get('result') != 1: raise ExtractorError('%s said: %s' % (self.IE_NAME, video_data['remsg'])) title = video_data['title'] info = { 'thumbnail': video_data.get('thumb'), 'view_count': int_or_none(video_data.get('vcnt')), 'age_limit': int_or_none(video_data.get('grade')), 'uploader_id': channel_id, 'uploader': video_data.get('cname'), } if video_id: entries = [] for i, f in enumerate(video_data.get('flist', [])): video_key = self.parse_video_key(f.get('key', '')) f_url = f.get('file') if not video_key or not f_url: continue entries.append({ 'id': '%s_%s' % (video_id, video_key.get('part', i + 1)), 'title': title, 'upload_date': video_key.get('upload_date'), 'duration': int_or_none(f.get('length')), 'url': f_url, 'protocol': 'm3u8_native', 'ext': 'mp4', }) info.update({ 'id': video_id, 'title': title, 'duration': int_or_none(video_data.get('length')), }) if len(entries) > 1: info['_type'] = 'multi_video' info['entries'] = entries elif len(entries) == 1: i = entries[0].copy() i.update(info) info = i else: formats = [] for s in video_data.get('strm', []): s_url = s.get('purl') if not s_url: continue # TODO: extract rtmp formats if s.get('stype') == 'HLS': formats.extend(self._extract_m3u8_formats( s_url, channel_id, 'mp4', fatal=False)) self._sort_formats(formats) info.update({ 'id': channel_id, 'title': self._live_title(title), 'is_live': True, 'formats': formats, }) return info