[funimation] Add new extractor
[youtube-dl] / youtube_dl / extractor / funimation.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3 from .common import InfoExtractor
4 from ..compat import compat_HTTPError
5 from ..utils import (
6     encode_dict,
7     sanitized_Request,
8     ExtractorError,
9     urlencode_postdata
10 )
11 import re
12
13 class FunimationIE(InfoExtractor):
14     _VALID_URL = r'https?://(?:www\.)?funimation\.com/shows/.+[^ ]/videos/official/(?P<id>[^?]+)'
15
16     _TEST = {
17         'url': 'http://www.funimation.com/shows/air/videos/official/breeze',
18         'info_dict': {
19             'id': 'AIRENG0001',
20             'title': 'Air - 1 - Breeze ',
21             'ext': 'mp4',
22             'thumbnail': 'http://www.funimation.com/admin/uploads/default/recap_thumbnails/7555590/home_spotlight/AIR0001.jpg',
23             'description': 'Travelling puppeteer Yukito arrives in a small town where he hopes to earn money through the magic of his puppets. When a young girl named Misuzu lures him to her home with the promise of food, his life changes forever. ',
24         }
25     }
26
27     def _login(self):
28         (username, password) = self._get_login_info()
29         if username is None:
30             return
31         login_url = 'http://www.funimation.com/login'
32         data = urlencode_postdata(encode_dict({
33             'loginForm2': 'loginform',
34             'email_field': username,
35             'password_field': password,
36         }))
37         login_request = sanitized_Request(login_url, data)
38         login_request.add_header('Content-Type', 'application/x-www-form-urlencoded')
39         try:
40             login = self._download_webpage(login_request, login_url)
41         except ExtractorError as e:
42             if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403:
43                 raise ExtractorError('Funimation is not available in your region.', expected=True)
44             raise
45         if re.search(r'<meta property="og:url" content="http://www.funimation.com/login"/>', login) is not None:
46                 raise ExtractorError('Unable to login, wrong username or password.', expected=True)
47
48     def _real_initialize(self):
49         self._login()
50
51     def _real_extract(self, url):
52         mobj = re.match(self._VALID_URL, url)
53         video_id = mobj.group('id')
54         try:
55             webpage = self._download_webpage(url, video_id)
56         except ExtractorError as e:
57             if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403:
58                 raise ExtractorError('Funimation is not available in your region.', expected=True)
59             raise
60         if re.search(r'"sdUrl":"http', webpage) is None:
61                 raise ExtractorError('You are not logged-in or the stream requires subscription.', expected=True)
62
63         m3u8 = self._search_regex(r'".+Url":"(.+?m3u8)"', webpage, 'm3u8') + self._search_regex(r'"authToken":"(.+?)"', webpage, 'm3u8')
64         formats = self._extract_m3u8_formats(m3u8.replace('\\', ''), video_id, ext='mp4', entry_protocol='m3u8_native')
65
66         video_show = self._search_regex(r'"artist":"(.+?)"', webpage, 'video_show')
67         video_track = self._search_regex(r'"videoNumber":"(\d+).0"', webpage, 'video_track')
68         video_title = self._search_regex(r'"title":"({0}.+?)"'.format(video_track), webpage, 'video_title')
69         video_id = self._search_regex(r'"FUNImationID":"(.+?)"', webpage, 'video_id')
70
71         return {
72             'id': video_id,
73             'title': video_show + ' - ' + video_title + ' ',
74             'formats': formats,
75             'thumbnail': self._og_search_thumbnail(webpage),
76             'description': self._og_search_description(webpage)
77         }