summaryrefslogtreecommitdiff
path: root/youtube_dl/extractor/tubitv.py
blob: 36f6c16732c7217141e860456788d389eb6bdd94 (plain)
    1 # coding: utf-8
    2 from __future__ import unicode_literals
    3 
    4 import re
    5 
    6 from .common import InfoExtractor
    7 from ..utils import (
    8     ExtractorError,
    9     int_or_none,
   10     sanitized_Request,
   11     urlencode_postdata,
   12 )
   13 
   14 
   15 class TubiTvIE(InfoExtractor):
   16     _VALID_URL = r'https?://(?:www\.)?tubitv\.com/(?:video|movies|tv-shows)/(?P<id>[0-9]+)'
   17     _LOGIN_URL = 'http://tubitv.com/login'
   18     _NETRC_MACHINE = 'tubitv'
   19     _GEO_COUNTRIES = ['US']
   20     _TESTS = [{
   21         'url': 'http://tubitv.com/video/283829/the_comedian_at_the_friday',
   22         'md5': '43ac06be9326f41912dc64ccf7a80320',
   23         'info_dict': {
   24             'id': '283829',
   25             'ext': 'mp4',
   26             'title': 'The Comedian at The Friday',
   27             'description': 'A stand up comedian is forced to look at the decisions in his life while on a one week trip to the west coast.',
   28             'uploader_id': 'bc168bee0d18dd1cb3b86c68706ab434',
   29         },
   30     }, {
   31         'url': 'http://tubitv.com/tv-shows/321886/s01_e01_on_nom_stories',
   32         'only_matching': True,
   33     }, {
   34         'url': 'http://tubitv.com/movies/383676/tracker',
   35         'only_matching': True,
   36     }]
   37 
   38     def _login(self):
   39         (username, password) = self._get_login_info()
   40         if username is None:
   41             return
   42         self.report_login()
   43         form_data = {
   44             'username': username,
   45             'password': password,
   46         }
   47         payload = urlencode_postdata(form_data)
   48         request = sanitized_Request(self._LOGIN_URL, payload)
   49         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
   50         login_page = self._download_webpage(
   51             request, None, False, 'Wrong login info')
   52         if not re.search(r'id="tubi-logout"', login_page):
   53             raise ExtractorError(
   54                 'Login failed (invalid username/password)', expected=True)
   55 
   56     def _real_initialize(self):
   57         self._login()
   58 
   59     def _real_extract(self, url):
   60         video_id = self._match_id(url)
   61         video_data = self._download_json(
   62             'http://tubitv.com/oz/videos/%s/content' % video_id, video_id)
   63         title = video_data['title']
   64 
   65         formats = self._extract_m3u8_formats(
   66             self._proto_relative_url(video_data['url']),
   67             video_id, 'mp4', 'm3u8_native')
   68         self._sort_formats(formats)
   69 
   70         thumbnails = []
   71         for thumbnail_url in video_data.get('thumbnails', []):
   72             if not thumbnail_url:
   73                 continue
   74             thumbnails.append({
   75                 'url': self._proto_relative_url(thumbnail_url),
   76             })
   77 
   78         subtitles = {}
   79         for sub in video_data.get('subtitles', []):
   80             sub_url = sub.get('url')
   81             if not sub_url:
   82                 continue
   83             subtitles.setdefault(sub.get('lang', 'English'), []).append({
   84                 'url': self._proto_relative_url(sub_url),
   85             })
   86 
   87         return {
   88             'id': video_id,
   89             'title': title,
   90             'formats': formats,
   91             'subtitles': subtitles,
   92             'thumbnails': thumbnails,
   93             'description': video_data.get('description'),
   94             'duration': int_or_none(video_data.get('duration')),
   95             'uploader_id': video_data.get('publisher_id'),
   96         }

Generated by cgit