diff --git a/weboob/backends/youtube/backend.py b/weboob/backends/youtube/backend.py index 60214c07..1cad4a8e 100644 --- a/weboob/backends/youtube/backend.py +++ b/weboob/backends/youtube/backend.py @@ -38,6 +38,9 @@ class YoutubeBackend(Backend, ICapVideoProvider): def inner(self, *args, **kwargs): if not self.browser: self.browser = YoutubeBrowser() + url = args[0] + if u'youtube.com' not in url: + return None return func(self, *args, **kwargs) return inner diff --git a/weboob/backends/youtube/browser.py b/weboob/backends/youtube/browser.py index a56fd2f1..d701118d 100644 --- a/weboob/backends/youtube/browser.py +++ b/weboob/backends/youtube/browser.py @@ -26,7 +26,7 @@ from weboob.tools.parser import LxmlHtmlParser from .pages import VideoPage class YoutubeBrowser(Browser): - regex = re.compile(r'&t=([^ ,&]*)') + video_signature_regex = re.compile(r'&t=([^ ,&]*)') def __init__(self, *args, **kwargs): kwargs['parser'] = LxmlHtmlParser() @@ -38,13 +38,16 @@ class YoutubeBrowser(Browser): return self.page.title def get_video_url(self, page_url): - result = self.openurl(page_url).read() - for _signature in re.finditer(self.regex, result): - signature = _signature.group(1) - break + def find_video_signature(data): + for video_signature in re.finditer(self.video_signature_regex, data): + return video_signature.group(1) + return None + data = self.openurl(page_url).read() + video_signature = find_video_signature(data) + m = re.match(r'http://.*\.youtube\.com/watch\?v=(.+)', page_url) + if m: + video_id = m.group(1) + url = 'http://www.youtube.com/get_video?video_id=%s&t=%s&fmt=18' % (video_id, video_signature) + return url else: return None - m = re.match(r'http://.*\.youtube\.com/watch\?v=(.+)', page_url) - video_id = m.group(1) - url = 'http://www.youtube.com/get_video?video_id=%s&t=%s&fmt=18' % (video_id, signature) - return url