[tutv] Simplify and use unicode_literals

This commit is contained in:
Jaime Marquínez Ferrándiz 2014-01-29 15:21:12 +01:00
parent 26a78d4bbf
commit d4a21e0b49
1 changed files with 13 additions and 14 deletions

View File

@ -1,3 +1,4 @@
from __future__ import unicode_literals
import base64 import base64
import re import re
@ -6,15 +7,16 @@ from ..utils import (
compat_parse_qs, compat_parse_qs,
) )
class TutvIE(InfoExtractor): class TutvIE(InfoExtractor):
_VALID_URL=r'https?://(?:www\.)?tu\.tv/videos/(?P<id>[^/?]+)' _VALID_URL = r'https?://(?:www\.)?tu\.tv/videos/(?P<id>[^/?]+)'
_TEST = { _TEST = {
u'url': u'http://tu.tv/videos/noah-en-pabellon-cuahutemoc', 'url': 'http://tu.tv/videos/noah-en-pabellon-cuahutemoc',
u'file': u'2742556.flv', 'file': '2742556.flv',
u'md5': u'5eb766671f69b82e528dc1e7769c5cb2', 'md5': '5eb766671f69b82e528dc1e7769c5cb2',
u'info_dict': { 'info_dict': {
u"title": u"Noah en pabellon cuahutemoc" 'title': 'Noah en pabellon cuahutemoc',
} },
} }
def _real_extract(self, url): def _real_extract(self, url):
@ -22,18 +24,15 @@ class TutvIE(InfoExtractor):
video_id = mobj.group('id') video_id = mobj.group('id')
webpage = self._download_webpage(url, video_id) webpage = self._download_webpage(url, video_id)
internal_id = self._search_regex(r'codVideo=([0-9]+)', webpage, u'internal video ID') internal_id = self._search_regex(r'codVideo=([0-9]+)', webpage, 'internal video ID')
data_url = u'http://tu.tv/flvurl.php?codVideo=' + str(internal_id) data_url = 'http://tu.tv/flvurl.php?codVideo=' + str(internal_id)
data_content = self._download_webpage(data_url, video_id, note=u'Downloading video info') data_content = self._download_webpage(data_url, video_id, note='Downloading video info')
data = compat_parse_qs(data_content) data = compat_parse_qs(data_content)
video_url = base64.b64decode(data['kpt'][0]).decode('utf-8') video_url = base64.b64decode(data['kpt'][0]).decode('utf-8')
ext = video_url.partition(u'?')[0].rpartition(u'.')[2]
info = { return {
'id': internal_id, 'id': internal_id,
'url': video_url, 'url': video_url,
'ext': ext,
'title': self._og_search_title(webpage), 'title': self._og_search_title(webpage),
} }
return [info]