[comedycentral] Re-add shortnames

In cc99d4f826, the shortname feature got deleted by accident. Re-add it as a separate IE.
This commit is contained in:
Philipp Hagemeister 2016-08-02 14:02:31 +02:00
parent 45408eb075
commit 6a9b3b61ea
2 changed files with 21 additions and 0 deletions

View File

@ -1,6 +1,7 @@
from __future__ import unicode_literals
from .mtv import MTVServicesInfoExtractor
from .common import InfoExtractor
class ComedyCentralIE(MTVServicesInfoExtractor):
@ -96,3 +97,22 @@ def _real_extract(self, url):
webpage, 'mrss url', group='url')
return self._get_videos_info_from_url(mrss_url, video_id)
class ComedyCentralShortnameIE(InfoExtractor):
_VALID_URL = r'^:(?P<id>tds|thedailyshow)$'
_TESTS = [{
'url': ':tds',
'only_matching': True,
}, {
'url': ':thedailyshow',
'only_matching': True,
}]
def _real_extract(self, url):
video_id = self._match_id(url)
shortcut_map = {
'tds': 'http://www.cc.com/shows/the-daily-show-with-trevor-noah/full-episodes',
'thedailyshow': 'http://www.cc.com/shows/the-daily-show-with-trevor-noah/full-episodes',
}
return self.url_result(shortcut_map[video_id])

View File

@ -159,6 +159,7 @@
from .collegerama import CollegeRamaIE
from .comedycentral import (
ComedyCentralIE,
ComedyCentralShortnameIE,
ComedyCentralTVIE,
ToshIE,
)