From ffaf6e66e3830d0e4750aec2cfdaff1a6bd9c2ad Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sergey=20M=E2=80=A4?= Date: Sat, 19 Dec 2015 19:08:47 +0600 Subject: [PATCH] [toggle] Improve --- youtube_dl/extractor/togglesg.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/youtube_dl/extractor/togglesg.py b/youtube_dl/extractor/togglesg.py index d3903b58d..244c79e8d 100644 --- a/youtube_dl/extractor/togglesg.py +++ b/youtube_dl/extractor/togglesg.py @@ -84,12 +84,15 @@ class ToggleSgIE(InfoExtractor): def _real_extract(self, url): video_id = self._match_id(url) - webpage = self._download_webpage(url, video_id, note='Downloading video page') + webpage = self._download_webpage( + url, video_id, note='Downloading video page') api_user = self._search_regex( - r'apiUser:\s*"([^"]+)"', webpage, 'apiUser', default=self._API_USER) + r'apiUser\s*:\s*(["\'])(?P.+?)\1', webpage, 'apiUser', + default=self._API_USER, group='user') api_pass = self._search_regex( - r'apiPass:\s*"([^"]+)"', webpage, 'apiPass', default=self._API_PASS) + r'apiPass\s*:\s*(["\'])(?P.+?)\1', webpage, 'apiPass', + default=self._API_PASS, group='pass') params = { 'initObj': { @@ -131,11 +134,10 @@ class ToggleSgIE(InfoExtractor): video_file['URL'], video_id, ext='mp4', m3u8_id=vid_format, note='Downloading %s m3u8 information' % vid_format, errnote='Failed to download %s m3u8 information' % vid_format, - fatal=False - ) + fatal=False) if m3u8_formats: formats.extend(m3u8_formats) - if ext in ['mp4', 'wvm']: + elif ext in ('mp4', 'wvm'): # wvm are drm-protected files formats.append({ 'ext': ext,