[funimation] Fix login message
[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
14 class FunimationIE(InfoExtractor):
15     _VALID_URL = r'https?://(?:www\.)?funimation\.com/shows/.+[^ ]/videos/official/(?P<id>[^?]+)'
16
17     _TEST = {
18         'url': 'http://www.funimation.com/shows/air/videos/official/breeze',
19         'info_dict': {
20             'id': 'AIRENG0001',
21             'title': 'Air - 1 - Breeze ',
22             'ext': 'mp4',
23             'thumbnail': 'http://www.funimation.com/admin/uploads/default/recap_thumbnails/7555590/home_spotlight/AIR0001.jpg',
24             '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. ',
25         }
26     }
27
28     def _login(self):
29         (username, password) = self._get_login_info()
30         if username is None:
31             return
32         login_url = 'http://www.funimation.com/login'
33         data = urlencode_postdata(encode_dict({
34             'loginForm2': 'loginform',
35             'email_field': username,
36             'password_field': password,
37         }))
38         login_request = sanitized_Request(login_url, data)
39         login_request.add_header('Content-Type', 'application/x-www-form-urlencoded')
40         try:
41             login = self._download_webpage(
42                 login_request, None, 'Logging in as %s' % username)
43         except ExtractorError as e:
44             if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403:
45                 raise ExtractorError('Funimation is not available in your region.', expected=True)
46             raise
47         if re.search(r'<meta property="og:url" content="http://www.funimation.com/login"/>', login) is not None:
48             raise ExtractorError('Unable to login, wrong username or password.', expected=True)
49
50     def _real_initialize(self):
51         self._login()
52
53     def _real_extract(self, url):
54         mobj = re.match(self._VALID_URL, url)
55         video_id = mobj.group('id')
56         try:
57             webpage = self._download_webpage(url, video_id)
58         except ExtractorError as e:
59             if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403:
60                 raise ExtractorError('Funimation is not available in your region.', expected=True)
61             raise
62         if re.search(r'"sdUrl":"http', webpage) is None:
63                 raise ExtractorError('You are not logged-in or the stream requires subscription.', expected=True)
64
65         m3u8 = self._search_regex(r'".+Url":"(.+?m3u8)"', webpage, 'm3u8') + self._search_regex(r'"authToken":"(.+?)"', webpage, 'm3u8')
66         formats = self._extract_m3u8_formats(m3u8.replace('\\', ''), video_id, ext='mp4', entry_protocol='m3u8_native')
67
68         video_show = self._search_regex(r'"artist":"(.+?)"', webpage, 'video_show')
69         video_track = self._search_regex(r'"videoNumber":"(\d+).0"', webpage, 'video_track')
70         video_title = self._search_regex(r'"title":"({0}.+?)"'.format(video_track), webpage, 'video_title')
71         video_id = self._search_regex(r'"FUNImationID":"(.+?)"', webpage, 'video_id')
72
73         return {
74             'id': video_id,
75             'title': video_show + ' - ' + video_title + ' ',
76             'formats': formats,
77             'thumbnail': self._og_search_thumbnail(webpage),
78             'description': self._og_search_description(webpage)
79         }