nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix getting video from an url, and catch errors

Browse source
This commit is contained in:
Romain Bignon 2011-03-13 14:38:20 +01:00
commit ce0fec8d34
1 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

15
weboob/backends/youtube/backend.py
View file

@ -20,6 +20,7 @@ from __future__ import with_statement
import datetime import datetime
import gdata.youtube.service import gdata.youtube.service
import re
import urllib import urllib
from weboob.capabilities.video import ICapVideo from weboob.capabilities.video import ICapVideo
@ -87,9 +88,21 @@ class YoutubeBackend(BaseBackend, ICapVideo):
LICENSE = 'GPLv3' LICENSE = 'GPLv3'
BROWSER = YoutubeBrowser BROWSER = YoutubeBrowser
URL_RE = re.compile(r'https?://.*youtube.com/watch\?v=(.*)')
def get_video(self, _id): def get_video(self, _id):
m = self.URL_RE.match(_id)
if m:
_id = m.group(1)
yt_service = gdata.youtube.service.YouTubeService() yt_service = gdata.youtube.service.YouTubeService()
entry = yt_service.GetYouTubeVideoEntry(video_id=_id) try:
entry = yt_service.GetYouTubeVideoEntry(video_id=_id)
except gdata.service.Error, e:
if e.args[0]['status'] == 400:
return None
raise
video = get_video(entry) video = get_video(entry)
video.url = get_video_url(video) video.url = get_video_url(video)
return video return video