mirror of
				https://gitlab.com/ytdl-org/youtube-dl.git
				synced 2025-11-03 07:57:08 -05:00 
			
		
		
		
	
		
			
				
	
	
		
			147 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			147 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
from __future__ import unicode_literals
 | 
						|
 | 
						|
import re
 | 
						|
import json
 | 
						|
 | 
						|
from .common import InfoExtractor
 | 
						|
from ..compat import compat_urlparse
 | 
						|
from ..utils import (
 | 
						|
    int_or_none,
 | 
						|
)
 | 
						|
 | 
						|
 | 
						|
class AppleTrailersIE(InfoExtractor):
 | 
						|
    _VALID_URL = r'https?://(?:www\.)?trailers\.apple\.com/(?:trailers|ca)/(?P<company>[^/]+)/(?P<movie>[^/]+)'
 | 
						|
    _TESTS = [{
 | 
						|
        'url': 'http://trailers.apple.com/trailers/wb/manofsteel/',
 | 
						|
        'info_dict': {
 | 
						|
            'id': 'manofsteel',
 | 
						|
        },
 | 
						|
        'playlist': [
 | 
						|
            {
 | 
						|
                'md5': 'd97a8e575432dbcb81b7c3acb741f8a8',
 | 
						|
                'info_dict': {
 | 
						|
                    'id': 'manofsteel-trailer4',
 | 
						|
                    'ext': 'mov',
 | 
						|
                    'duration': 111,
 | 
						|
                    'title': 'Trailer 4',
 | 
						|
                    'upload_date': '20130523',
 | 
						|
                    'uploader_id': 'wb',
 | 
						|
                },
 | 
						|
            },
 | 
						|
            {
 | 
						|
                'md5': 'b8017b7131b721fb4e8d6f49e1df908c',
 | 
						|
                'info_dict': {
 | 
						|
                    'id': 'manofsteel-trailer3',
 | 
						|
                    'ext': 'mov',
 | 
						|
                    'duration': 182,
 | 
						|
                    'title': 'Trailer 3',
 | 
						|
                    'upload_date': '20130417',
 | 
						|
                    'uploader_id': 'wb',
 | 
						|
                },
 | 
						|
            },
 | 
						|
            {
 | 
						|
                'md5': 'd0f1e1150989b9924679b441f3404d48',
 | 
						|
                'info_dict': {
 | 
						|
                    'id': 'manofsteel-trailer',
 | 
						|
                    'ext': 'mov',
 | 
						|
                    'duration': 148,
 | 
						|
                    'title': 'Trailer',
 | 
						|
                    'upload_date': '20121212',
 | 
						|
                    'uploader_id': 'wb',
 | 
						|
                },
 | 
						|
            },
 | 
						|
            {
 | 
						|
                'md5': '5fe08795b943eb2e757fa95cb6def1cb',
 | 
						|
                'info_dict': {
 | 
						|
                    'id': 'manofsteel-teaser',
 | 
						|
                    'ext': 'mov',
 | 
						|
                    'duration': 93,
 | 
						|
                    'title': 'Teaser',
 | 
						|
                    'upload_date': '20120721',
 | 
						|
                    'uploader_id': 'wb',
 | 
						|
                },
 | 
						|
            },
 | 
						|
        ]
 | 
						|
    }, {
 | 
						|
        'url': 'http://trailers.apple.com/ca/metropole/autrui/',
 | 
						|
        'only_matching': True,
 | 
						|
    }]
 | 
						|
 | 
						|
    _JSON_RE = r'iTunes.playURL\((.*?)\);'
 | 
						|
 | 
						|
    def _real_extract(self, url):
 | 
						|
        mobj = re.match(self._VALID_URL, url)
 | 
						|
        movie = mobj.group('movie')
 | 
						|
        uploader_id = mobj.group('company')
 | 
						|
 | 
						|
        playlist_url = compat_urlparse.urljoin(url, 'includes/playlists/itunes.inc')
 | 
						|
 | 
						|
        def fix_html(s):
 | 
						|
            s = re.sub(r'(?s)<script[^<]*?>.*?</script>', '', s)
 | 
						|
            s = re.sub(r'<img ([^<]*?)>', r'<img \1/>', s)
 | 
						|
            # The ' in the onClick attributes are not escaped, it couldn't be parsed
 | 
						|
            # like: http://trailers.apple.com/trailers/wb/gravity/
 | 
						|
 | 
						|
            def _clean_json(m):
 | 
						|
                return 'iTunes.playURL(%s);' % m.group(1).replace('\'', ''')
 | 
						|
            s = re.sub(self._JSON_RE, _clean_json, s)
 | 
						|
            s = '<html>%s</html>' % s
 | 
						|
            return s
 | 
						|
        doc = self._download_xml(playlist_url, movie, transform_source=fix_html)
 | 
						|
 | 
						|
        playlist = []
 | 
						|
        for li in doc.findall('./div/ul/li'):
 | 
						|
            on_click = li.find('.//a').attrib['onClick']
 | 
						|
            trailer_info_json = self._search_regex(self._JSON_RE,
 | 
						|
                                                   on_click, 'trailer info')
 | 
						|
            trailer_info = json.loads(trailer_info_json)
 | 
						|
            title = trailer_info['title']
 | 
						|
            video_id = movie + '-' + re.sub(r'[^a-zA-Z0-9]', '', title).lower()
 | 
						|
            thumbnail = li.find('.//img').attrib['src']
 | 
						|
            upload_date = trailer_info['posted'].replace('-', '')
 | 
						|
 | 
						|
            runtime = trailer_info['runtime']
 | 
						|
            m = re.search(r'(?P<minutes>[0-9]+):(?P<seconds>[0-9]{1,2})', runtime)
 | 
						|
            duration = None
 | 
						|
            if m:
 | 
						|
                duration = 60 * int(m.group('minutes')) + int(m.group('seconds'))
 | 
						|
 | 
						|
            first_url = trailer_info['url']
 | 
						|
            trailer_id = first_url.split('/')[-1].rpartition('_')[0].lower()
 | 
						|
            settings_json_url = compat_urlparse.urljoin(url, 'includes/settings/%s.json' % trailer_id)
 | 
						|
            settings = self._download_json(settings_json_url, trailer_id, 'Downloading settings json')
 | 
						|
 | 
						|
            formats = []
 | 
						|
            for format in settings['metadata']['sizes']:
 | 
						|
                # The src is a file pointing to the real video file
 | 
						|
                format_url = re.sub(r'_(\d*p.mov)', r'_h\1', format['src'])
 | 
						|
                formats.append({
 | 
						|
                    'url': format_url,
 | 
						|
                    'format': format['type'],
 | 
						|
                    'width': int_or_none(format['width']),
 | 
						|
                    'height': int_or_none(format['height']),
 | 
						|
                })
 | 
						|
 | 
						|
            self._sort_formats(formats)
 | 
						|
 | 
						|
            playlist.append({
 | 
						|
                '_type': 'video',
 | 
						|
                'id': video_id,
 | 
						|
                'formats': formats,
 | 
						|
                'title': title,
 | 
						|
                'duration': duration,
 | 
						|
                'thumbnail': thumbnail,
 | 
						|
                'upload_date': upload_date,
 | 
						|
                'uploader_id': uploader_id,
 | 
						|
                'http_headers': {
 | 
						|
                    'User-Agent': 'QuickTime compatible (youtube-dl)',
 | 
						|
                },
 | 
						|
            })
 | 
						|
 | 
						|
        return {
 | 
						|
            '_type': 'playlist',
 | 
						|
            'id': movie,
 | 
						|
            'entries': playlist,
 | 
						|
        }
 |