[hotstar] checked code with flake8 and modified

This commit is contained in:
Alpesh Valia 2017-03-18 22:21:08 +05:30
parent 0ac58ee4ef
commit f4c968ac5b

View File

@ -101,6 +101,7 @@ class HotStarIE(InfoExtractor):
'series': video_data.get('contentTitle'), 'series': video_data.get('contentTitle'),
} }
class HotStarPlaylistIE(InfoExtractor): class HotStarPlaylistIE(InfoExtractor):
IE_NAME = 'hotstar:playlist' IE_NAME = 'hotstar:playlist'
_VALID_URL = r'https?://(?:www\.)?hotstar\.com/tv/(?P<playlist_title>.+)/(?P<series_id>\d+)/episodes/(?P<playlist_id>\d{1,})' _VALID_URL = r'https?://(?:www\.)?hotstar\.com/tv/(?P<playlist_title>.+)/(?P<series_id>\d+)/episodes/(?P<playlist_id>\d{1,})'
@ -117,12 +118,12 @@ class HotStarPlaylistIE(InfoExtractor):
'only_matching': True, 'only_matching': True,
}] }]
def _extract_episode_info(self, series_id, playlist_title, video ): def _extract_episode_info(self, series_id, playlist_title, video):
picture_url = video.get('urlPictures'); picture_url = video.get('urlPictures')
thumbnail = '' thumbnail = ''
if picture_url: if picture_url:
thumbnail = 'http://media0-starag.startv.in/r1/thumbs/PCTV/%s/%s/PCTV-%s-hs.jpg' % ( picture_url[-2:], picture_url, picture_url ) thumbnail = 'http://media0-starag.startv.in/r1/thumbs/PCTV/%s/%s/PCTV-%s-hs.jpg' % (picture_url[-2:], picture_url, picture_url)
episode_title = video.get('episodeTitle', '') episode_title = video.get('episodeTitle', '')
episode_title = episode_title.lower().replace(' ', '-') episode_title = episode_title.lower().replace(' ', '-')
@ -132,16 +133,16 @@ class HotStarPlaylistIE(InfoExtractor):
'id': video.get('contentId'), 'id': video.get('contentId'),
'title': video.get('episodeTitle'), 'title': video.get('episodeTitle'),
'description': video.get('longDescription'), 'description': video.get('longDescription'),
'thumbnail' : thumbnail, 'thumbnail': thumbnail,
'url' : url, 'url': url,
'_type' : 'url', '_type': 'url',
} }
return info_dict return info_dict
def _real_extract(self, url): def _real_extract(self, url):
mobj = re.match(self._VALID_URL, url) mobj = re.match(self._VALID_URL, url)
series_id = mobj.group('series_id') series_id = mobj.group('series_id')
playlist_id = mobj.group('playlist_id') playlist_id = mobj.group('playlist_id')
playlist_title = mobj.group('playlist_title') playlist_title = mobj.group('playlist_title')
collection = self._download_json( collection = self._download_json(
@ -151,6 +152,6 @@ class HotStarPlaylistIE(InfoExtractor):
videos = collection.get('resultObj', {}).get('response', {}).get('docs', []) videos = collection.get('resultObj', {}).get('response', {}).get('docs', [])
entries = [ entries = [
self._extract_episode_info( series_id, playlist_title, video ) self._extract_episode_info(series_id, playlist_title, video)
for video in videos if video.get('contentId')] for video in videos if video.get('contentId')]
return self.playlist_result(entries, playlist_id, playlist_title) return self.playlist_result(entries, playlist_id, playlist_title)