[iqiyi] PEP8

This commit is contained in:
Yen Chi Hsuan 2015-06-07 00:37:29 +08:00
parent 9bac8c57e3
commit f1da861018

View File

@ -16,19 +16,20 @@ import random
import zlib import zlib
import hashlib import hashlib
class IqiyiIE(InfoExtractor): class IqiyiIE(InfoExtractor):
IE_NAME = 'iqiyi' IE_NAME = 'iqiyi'
_VALID_URL = r'http://(?:www\.)iqiyi.com/.+?\.html' _VALID_URL = r'http://(?:www\.)iqiyi.com/.+?\.html'
_TEST = { _TEST = {
'url': 'http://www.iqiyi.com/v_19rrojlavg.html', 'url': 'http://www.iqiyi.com/v_19rrojlavg.html',
'md5': '2cb594dc2781e6c941a110d8f358118b', 'md5': '2cb594dc2781e6c941a110d8f358118b',
'info_dict': { 'info_dict': {
'id': '9c1fb1b99d192b21c559e5a1a2cb3c73', 'id': '9c1fb1b99d192b21c559e5a1a2cb3c73',
'title': '美国德州空中惊现奇异云团 酷似UFO', 'title': '美国德州空中惊现奇异云团 酷似UFO',
'ext': 'f4v', 'ext': 'f4v',
} }
} }
def construct_video_urls(self, data, video_id, _uuid, bid): def construct_video_urls(self, data, video_id, _uuid, bid):
@ -46,7 +47,7 @@ class IqiyiIE(InfoExtractor):
c = len(b) c = len(b)
s = '' s = ''
for i in range(c - 1, -1, -1): for i in range(c - 1, -1, -1):
a = do_xor(int(b[c-i-1], 16), i) a = do_xor(int(b[c - i - 1], 16), i)
s += chr(a) s += chr(a)
return s[::-1] return s[::-1]
@ -54,15 +55,14 @@ class IqiyiIE(InfoExtractor):
mg = ')(*&^flash@#$%a' mg = ')(*&^flash@#$%a'
tm = self._download_json( tm = self._download_json(
'http://data.video.qiyi.com/t?tn=' + str(random.random()), video_id)['t'] 'http://data.video.qiyi.com/t?tn=' + str(random.random()), video_id)['t']
t = str(int(math.floor(int(tm)/(600.0)))) t = str(int(math.floor(int(tm) / (600.0))))
return hashlib.md5( return hashlib.md5((t + mg + x).encode('utf8')).hexdigest()
(t+mg+x).encode('utf8')).hexdigest()
# get accept format # get accept format
# getting all format will spend minutes for a big video. # getting all format will spend minutes for a big video.
if bid == 'best': if bid == 'best':
bids = [int(i['bid']) for i in data['vp']['tkl'][0]['vs'] \ bids = [int(i['bid']) for i in data['vp']['tkl'][0]['vs']
if 0 < int(i['bid']) <= 10] if 0 < int(i['bid']) <= 10]
bid = str(max(bids)) bid = str(max(bids))
video_urls_dict = {} video_urls_dict = {}
@ -117,24 +117,24 @@ class IqiyiIE(InfoExtractor):
def get_format(self, bid): def get_format(self, bid):
_dict = { _dict = {
'1' : 'h6', '1': 'h6',
'2' : 'h5', '2': 'h5',
'3' : 'h4', '3': 'h4',
'4' : 'h3', '4': 'h3',
'5' : 'h2', '5': 'h2',
'10' : 'h1' '10': 'h1'
} }
return _dict.get(str(bid), None) return _dict.get(str(bid), None)
def get_bid(self, format_id): def get_bid(self, format_id):
_dict = { _dict = {
'h6' : '1', 'h6': '1',
'h5' : '2', 'h5': '2',
'h4' : '3', 'h4': '3',
'h3' : '4', 'h3': '4',
'h2' : '5', 'h2': '5',
'h1' : '10', 'h1': '10',
'best' : 'best' 'best': 'best'
} }
return _dict.get(format_id, None) return _dict.get(format_id, None)
@ -207,7 +207,7 @@ class IqiyiIE(InfoExtractor):
for format_id in video_urls_dict: for format_id in video_urls_dict:
video_urls = video_urls_dict[format_id] video_urls = video_urls_dict[format_id]
for i, video_url_info in enumerate(video_urls): for i, video_url_info in enumerate(video_urls):
if len(entries) < i+1: if len(entries) < i + 1:
entries.append({'formats': []}) entries.append({'formats': []})
entries[i]['formats'].append( entries[i]['formats'].append(
{ {
@ -222,7 +222,7 @@ class IqiyiIE(InfoExtractor):
self._sort_formats(entries[i]['formats']) self._sort_formats(entries[i]['formats'])
entries[i].update( entries[i].update(
{ {
'id': '_part%d' % (i+1), 'id': '_part%d' % (i + 1),
'title': title, 'title': title,
} }
) )