From 7c145d9a2b6ce5098de5a6db00ac48a40efb837d Mon Sep 17 00:00:00 2001 From: Zhang Ning Date: Sun, 13 Sep 2015 18:40:51 +0800 Subject: [PATCH] support size for hunantv just use url_size api --- src/you_get/extractors/hunan.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/src/you_get/extractors/hunan.py b/src/you_get/extractors/hunan.py index 07f4df2d..6c4a9fa7 100644 --- a/src/you_get/extractors/hunan.py +++ b/src/you_get/extractors/hunan.py @@ -47,6 +47,19 @@ class Hunantv(VideoExtractor): self.streams[lstream['name']] = {'container': 'fhv', 'video_profile': lstream['name'], 'size' : 0} def extract(self, **kwargs): + if 'info_only' in kwargs and kwargs['info_only']: + for lstream in self.lstreams: + meta = '' + while True: + rn = randint(0, 99999999) + meta = json.loads(get_html("{}&random={}".format((lstream['url']),rn))) + if meta['status'] == 'ok': + if meta['info'].startswith('http://pcfastvideo.imgo.tv/'): + break + size = url_size(meta['info']) + self.streams[lstream['name']]['src'] = [meta['info']] + self.streams[lstream['name']]['size'] = size + if 'stream_id' in kwargs and kwargs['stream_id']: # Extract the stream stream_id = kwargs['stream_id'] @@ -68,7 +81,9 @@ class Hunantv(VideoExtractor): if meta['status'] == 'ok': if meta['info'].startswith('http://pcfastvideo.imgo.tv/'): break + size = url_size(meta['info']) self.streams[stream_id]['src'] = [meta['info']] + self.streams[stream_id]['size'] = size site = Hunantv() download = site.download_by_url