vidbit.py (2917B)
1 from __future__ import unicode_literals 2 3 from .common import InfoExtractor 4 from ..compat import compat_urlparse 5 from ..utils import ( 6 int_or_none, 7 js_to_json, 8 remove_end, 9 unified_strdate, 10 ) 11 12 13 class VidbitIE(InfoExtractor): 14 _VALID_URL = r'https?://(?:www\.)?vidbit\.co/(?:watch|embed)\?.*?\bv=(?P<id>[\da-zA-Z]+)' 15 _TESTS = [{ 16 'url': 'http://www.vidbit.co/watch?v=jkL2yDOEq2', 17 'md5': '1a34b7f14defe3b8fafca9796892924d', 18 'info_dict': { 19 'id': 'jkL2yDOEq2', 20 'ext': 'mp4', 21 'title': 'Intro to VidBit', 22 'description': 'md5:5e0d6142eec00b766cbf114bfd3d16b7', 23 'thumbnail': r're:https?://.*\.jpg$', 24 'upload_date': '20160618', 25 'view_count': int, 26 'comment_count': int, 27 } 28 }, { 29 'url': 'http://www.vidbit.co/embed?v=jkL2yDOEq2&auto=0&water=0', 30 'only_matching': True, 31 }] 32 33 def _real_extract(self, url): 34 video_id = self._match_id(url) 35 36 webpage = self._download_webpage( 37 compat_urlparse.urljoin(url, '/watch?v=%s' % video_id), video_id) 38 39 video_url, title = [None] * 2 40 41 config = self._parse_json(self._search_regex( 42 r'(?s)\.setup\(({.+?})\);', webpage, 'setup', default='{}'), 43 video_id, transform_source=js_to_json) 44 if config: 45 if config.get('file'): 46 video_url = compat_urlparse.urljoin(url, config['file']) 47 title = config.get('title') 48 49 if not video_url: 50 video_url = compat_urlparse.urljoin(url, self._search_regex( 51 r'file\s*:\s*(["\'])(?P<url>(?:(?!\1).)+)\1', 52 webpage, 'video URL', group='url')) 53 54 if not title: 55 title = remove_end( 56 self._html_search_regex( 57 (r'<h1>(.+?)</h1>', r'<title>(.+?)</title>'), 58 webpage, 'title', default=None) or self._og_search_title(webpage), 59 ' - VidBit') 60 61 description = self._html_search_meta( 62 ('description', 'og:description', 'twitter:description'), 63 webpage, 'description') 64 65 upload_date = unified_strdate(self._html_search_meta( 66 'datePublished', webpage, 'upload date')) 67 68 view_count = int_or_none(self._search_regex( 69 r'<strong>(\d+)</strong> views', 70 webpage, 'view count', fatal=False)) 71 comment_count = int_or_none(self._search_regex( 72 r'id=["\']cmt_num["\'][^>]*>\((\d+)\)', 73 webpage, 'comment count', fatal=False)) 74 75 return { 76 'id': video_id, 77 'url': video_url, 78 'title': title, 79 'description': description, 80 'thumbnail': self._og_search_thumbnail(webpage), 81 'upload_date': upload_date, 82 'view_count': view_count, 83 'comment_count': comment_count, 84 }