from __future__ import unicode_literals import re from .common import InfoExtractor from ..utils import ( ExtractorError, clean_html, determine_ext, int_or_none, js_to_json, parse_duration, ) class SnagFilmsEmbedIE(InfoExtractor): _VALID_URL = r'https?://(?:(?:www|embed)\.)?snagfilms\.com/embed/player\?.*\bfilmId=(?P[\da-f-]{36})' _TESTS = [{ 'url': 'http://embed.snagfilms.com/embed/player?filmId=74849a00-85a9-11e1-9660-123139220831&w=500', 'md5': '2924e9215c6eff7a55ed35b72276bd93', 'info_dict': { 'id': '74849a00-85a9-11e1-9660-123139220831', 'ext': 'mp4', 'title': '#whilewewatch', } }, { # invalid labels, 360p is better that 480p 'url': 'http://www.snagfilms.com/embed/player?filmId=17ca0950-a74a-11e0-a92a-0026bb61d036', 'md5': '882fca19b9eb27ef865efeeaed376a48', 'info_dict': { 'id': '17ca0950-a74a-11e0-a92a-0026bb61d036', 'ext': 'mp4', 'title': 'Life in Limbo', } }, { 'url': 'http://www.snagfilms.com/embed/player?filmId=0000014c-de2f-d5d6-abcf-ffef58af0017', 'only_matching': True, }] @staticmethod def _extract_url(webpage): mobj = re.search( r']+?src=(["\'])(?P(?:https?:)?//(?:embed\.)?snagfilms\.com/embed/player.+?)\1', webpage) if mobj: return mobj.group('url') def _real_extract(self, url): video_id = self._match_id(url) webpage = self._download_webpage(url, video_id) if '>This film is not playable in your area.<' in webpage: raise ExtractorError( 'Film %s is not playable in your area.' % video_id, expected=True) formats = [] for source in self._parse_json(js_to_json(self._search_regex( r'(?s)sources:\s*(\[.+?\]),', webpage, 'json')), video_id): file_ = source.get('file') if not file_: continue type_ = source.get('type') ext = determine_ext(file_) format_id = source.get('label') or ext if all(_ == 'm3u8' for _ in (type_, ext)): formats.extend(self._extract_m3u8_formats( file_, video_id, 'mp4', m3u8_id='hls')) else: bitrate = int_or_none(self._search_regex( [r'(\d+)kbps', r'_\d{1,2}x\d{1,2}_(\d{3,})\.%s' % ext], file_, 'bitrate', default=None)) height = int_or_none(self._search_regex( r'^(\d+)[pP]$', format_id, 'height', default=None)) formats.append({ 'url': file_, 'format_id': format_id, 'tbr': bitrate, 'height': height, }) self._sort_formats(formats) title = self._search_regex( [r"title\s*:\s*'([^']+)'", r'([^<]+)'], webpage, 'title') return { 'id': video_id, 'title': title, 'formats': formats, } class SnagFilmsIE(InfoExtractor): _VALID_URL = r'https?://(?:www\.)?snagfilms\.com/(?:films/title|show)/(?P[^?#]+)' _TESTS = [{ 'url': 'http://www.snagfilms.com/films/title/lost_for_life', 'md5': '19844f897b35af219773fd63bdec2942', 'info_dict': { 'id': '0000014c-de2f-d5d6-abcf-ffef58af0017', 'display_id': 'lost_for_life', 'ext': 'mp4', 'title': 'Lost for Life', 'description': 'md5:fbdacc8bb6b455e464aaf98bc02e1c82', 'thumbnail': 're:^https?://.*\.jpg', 'duration': 4489, 'categories': ['Documentary', 'Crime', 'Award Winning', 'Festivals'] } }, { 'url': 'http://www.snagfilms.com/show/the_world_cut_project/india', 'md5': 'e6292e5b837642bbda82d7f8bf3fbdfd', 'info_dict': { 'id': '00000145-d75c-d96e-a9c7-ff5c67b20000', 'display_id': 'the_world_cut_project/india', 'ext': 'mp4', 'title': 'India', 'description': 'md5:5c168c5a8f4719c146aad2e0dfac6f5f', 'thumbnail': 're:^https?://.*\.jpg', 'duration': 979, 'categories': ['Documentary', 'Sports', 'Politics'] } }, { # Film is not playable in your area. 'url': 'http://www.snagfilms.com/films/title/inside_mecca', 'only_matching': True, }, { # Film is not available. 'url': 'http://www.snagfilms.com/show/augie_alone/flirting', 'only_matching': True, }] def _real_extract(self, url): display_id = self._match_id(url) webpage = self._download_webpage(url, display_id) if ">Sorry, the Film you're looking for is not available.<" in webpage: raise ExtractorError( 'Film %s is not available.' % display_id, expected=True) film_id = self._search_regex(r'filmId=([\da-f-]{36})"', webpage, 'film id') snag = self._parse_json( self._search_regex( 'Snag\.page\.data\s*=\s*(\[.+?\]);', webpage, 'snag'), display_id) for item in snag: if item.get('data', {}).get('film', {}).get('id') == film_id: data = item['data']['film'] title = data['title'] description = clean_html(data.get('synopsis')) thumbnail = data.get('image') duration = int_or_none(data.get('duration') or data.get('runtime')) categories = [ category['title'] for category in data.get('categories', []) if category.get('title')] break else: title = self._search_regex( r'itemprop="title">([^<]+)<', webpage, 'title') description = self._html_search_regex( r'(?s)
(.+?)
', webpage, 'description', default=None) or self._og_search_description(webpage) thumbnail = self._og_search_thumbnail(webpage) duration = parse_duration(self._search_regex( r'([^<]+)<', webpage, 'duration', fatal=False)) categories = re.findall(r'([^<]+)', webpage) return { '_type': 'url_transparent', 'url': 'http://embed.snagfilms.com/embed/player?filmId=%s' % film_id, 'id': film_id, 'display_id': display_id, 'title': title, 'description': description, 'thumbnail': thumbnail, 'duration': duration, 'categories': categories, }