[mangomolo] fix domain regex
[youtube-dl] / youtube_dl / extractor / mangomolo.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import base64
5
6 from .common import InfoExtractor
7 from ..compat import compat_urllib_parse_unquote
8 from ..utils import (
9     int_or_none,
10 )
11
12
13 class MangomoloBaseIE(InfoExtractor):
14     def _get_real_id(self, page_id):
15         return page_id
16
17     def _real_extract(self, url):
18         page_id = self._get_real_id(self._match_id(url))
19         webpage = self._download_webpage(url, page_id)
20         hidden_inputs = self._hidden_inputs(webpage)
21         m3u8_entry_protocol = 'm3u8' if self._IS_LIVE else 'm3u8_native'
22
23         format_url = self._html_search_regex(
24             [
25                 r'file\s*:\s*"(https?://[^"]+?/playlist.m3u8)',
26                 r'<a[^>]+href="(rtsp://[^"]+)"'
27             ], webpage, 'format url')
28         formats = self._extract_wowza_formats(
29             format_url, page_id, m3u8_entry_protocol, ['smil'])
30         self._sort_formats(formats)
31
32         return {
33             'id': page_id,
34             'title': self._live_title(page_id) if self._IS_LIVE else page_id,
35             'uploader_id': hidden_inputs.get('userid'),
36             'duration': int_or_none(hidden_inputs.get('duration')),
37             'is_live': self._IS_LIVE,
38             'formats': formats,
39         }
40
41
42 class MangomoloVideoIE(MangomoloBaseIE):
43     IE_NAME = 'mangomolo:video'
44     _VALID_URL = r'https?://admin\.mangomolo\.com/analytics/index\.php/customers/embed/video\?.*?\bid=(?P<id>\d+)'
45     _IS_LIVE = False
46
47
48 class MangomoloLiveIE(MangomoloBaseIE):
49     IE_NAME = 'mangomolo:live'
50     _VALID_URL = r'https?://admin\.mangomolo\.com/analytics/index\.php/customers/embed/index\?.*?\bchannelid=(?P<id>(?:[A-Za-z0-9+/=]|%2B|%2F|%3D)+)'
51     _IS_LIVE = True
52
53     def _get_real_id(self, page_id):
54         return base64.b64decode(compat_urllib_parse_unquote(page_id).encode()).decode()