238 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			238 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| # 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<id>\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<upload_date>\d{8})_\w+_(?P<part>\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<channel_id>\d+)(?:/v/(?P<video_id>\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
 |