Update zingmp3.py

This commit is contained in:
Phạm Ngọc Quang Nam 2017-10-07 01:19:59 +07:00 committed by GitHub
parent e952847541
commit 9e854d7bb9

View File

@ -23,11 +23,9 @@ class ZingMp3BaseInfoExtractor(InfoExtractor):
expected=True) expected=True)
formats = [] formats = []
for quality, source_url in zip(item.get('qualities') or item.get('quality', []), item.get('source_list') or item.get('source', [])): for quality, source_url in item.get('source', []).items():
if not source_url or source_url == 'require vip': if not source_url or source_url == 'require vip':
continue continue
if not re.match(r'https?://', source_url):
source_url = '//' + source_url
source_url = self._proto_relative_url(source_url, 'http:') source_url = self._proto_relative_url(source_url, 'http:')
quality_num = int_or_none(quality) quality_num = int_or_none(quality)
f = { f = {
@ -58,12 +56,12 @@ class ZingMp3BaseInfoExtractor(InfoExtractor):
def _extract_player_json(self, player_json_url, id, page_type, playlist_title=None): def _extract_player_json(self, player_json_url, id, page_type, playlist_title=None):
player_json = self._download_json(player_json_url, id, 'Downloading Player JSON') player_json = self._download_json(player_json_url, id, 'Downloading Player JSON')
items = player_json['data'] items = player_json['data']
if 'item' in items: if 'items' in items:
items = items['item'] items = items['items']
if len(items) == 1: if type(items) == dict:
# one single song # one single song
data = self._extract_item(items[0], page_type) data = self._extract_item(items, page_type)
data['id'] = id data['id'] = id
return data return data
@ -132,12 +130,10 @@ class ZingMp3IE(ZingMp3BaseInfoExtractor):
], webpage, 'player xml url') ], webpage, 'player xml url')
playlist_title = None playlist_title = None
page_type = self._search_regex(r'/(?:html5)?xml/([^/-]+)', player_json_url, 'page type') page_type = self._search_regex(r'type=(\w+)', player_json_url, 'page type')
if page_type == 'video': if page_type == 'video':
player_json_url = update_url_query(player_json_url, {'format': 'json'}) player_json_url = update_url_query(player_json_url, {'format': 'json'})
else: elif page_type == 'album':
player_json_url = player_json_url.replace('/xml/', '/html5xml/')
if page_type == 'album':
playlist_title = self._og_search_title(webpage) playlist_title = self._og_search_title(webpage)
player_json_url = 'https://mp3.zing.vn/xhr' + player_json_url
return self._extract_player_json(player_json_url, page_id, page_type, playlist_title) return self._extract_player_json(player_json_url, page_id, page_type, playlist_title)