Akkariin/youtube-dl
1
0
Fork 0
mirror of https://github.com/l1ving/youtube-dl synced 2025-02-03 14:12:53 +08:00
Code Issues Releases Wiki Activity

Code quality

Browse Source
This commit is contained in:
Shrek 2017-04-11 16:29:10 +03:00
parent 46cd65550c
commit e46382b41b
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
youtube_dl/extractor/puhutv.py
View File

@ -2,7 +2,7 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import re import re
import logging
from .common import InfoExtractor from .common import InfoExtractor
@ -27,33 +27,33 @@ class PuhuTvIE(InfoExtractor):
mobj = re.match(self._VALID_URL, url) mobj = re.match(self._VALID_URL, url)
display_id = mobj.group('display_id') display_id = mobj.group('display_id')
video_data = self._download_json('https://puhutv.com/api/slug/' + str(display_id), display_id)['data'] video_data = self._download_json('https://puhutv.com/api/slug/' + str(display_id), display_id).get('data')
video_id = str(video_data['id']) video_id = str(video_data.get('id'))
video_title_name = video_data['title']['name'] video_title_name = video_data.get('title').get('name')
video_title_display_name = video_data['display_name'] video_title_display_name = video_data.get('display_name')
video_title = video_title_name + ' ' + video_title_display_name video_title = video_title_name + ' ' + video_title_display_name
dl_data = self._download_json('https://puhutv.com/api/assets/'+video_id+'/videos', video_id)['data'] dl_data = self._download_json('https://puhutv.com/api/assets/' + video_id + '/videos', video_id).get('data')
formats = [] formats = []
mp_fmts = [] mp_fmts = []
for dl in dl_data['videos']: for dl in dl_data.get('videos'):
if (dl['url'].__contains__('playlist.m3u8')): if (dl.get('url').__contains__('playlist.m3u8')):
formats.extend(self._extract_m3u8_formats(dl['url'], str(dl['id']), 'mp4')) formats.extend(self._extract_m3u8_formats(dl.get('url'), str(dl.get('id')), 'mp4'))
elif (dl['url'].__contains__('/mp4/')): elif (dl.get('url').__contains__('/mp4/')):
fmt = str(re.match(self._URL_FORMAT, dl['url']).group('vid_format')) fmt = str(re.match(self._URL_FORMAT, dl.get('url')).group('vid_format'))
mp_fmts.append({ mp_fmts.append({
'url': dl['url'], 'url': dl.get('url'),
'format_id': 'mp4-' + fmt, 'format_id': 'mp4-' + fmt,
'height': fmt 'height': fmt
}) })
mp_fmts.sort(key=lambda x: int(x['height'].replace('p',''))) mp_fmts.sort(key=lambda x: int(x.get('height').replace('p', '')))
formats.extend(mp_fmts) formats.extend(mp_fmts)
# extract video metadata # extract video metadata
thumbnail = 'https://' + video_data['content']['images']['wide']['main'] thumbnail = 'https://' + video_data.get('content').get('images').get('wide').get('main')
description = video_data['description'] description = video_data.get('description')
upload_date = video_data['created_at'].split('T')[0].replace('-','') upload_date = video_data.get('created_at').split('T')[0].replace('-', '')
return { return {
'id': display_id, 'id': display_id,