mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-27 03:03:01 +00:00
66 lines
2.4 KiB
Python
66 lines
2.4 KiB
Python
import datetime
|
|
import json
|
|
import re
|
|
|
|
from .common import InfoExtractor
|
|
|
|
from ..utils import (
|
|
ExtractorError,
|
|
)
|
|
|
|
class PhotobucketIE(InfoExtractor):
|
|
"""Information extractor for photobucket.com."""
|
|
|
|
# TODO: the original _VALID_URL was:
|
|
# r'(?:http://)?(?:[a-z0-9]+\.)?photobucket\.com/.*[\?\&]current=(.*\.flv)'
|
|
# Check if it's necessary to keep the old extracion process
|
|
_VALID_URL = r'(?:http://)?(?:[a-z0-9]+\.)?photobucket\.com/.*(([\?\&]current=)|_)(?P<id>.*)\.(?P<ext>(flv)|(mp4))'
|
|
IE_NAME = u'photobucket'
|
|
|
|
def _real_extract(self, url):
|
|
# Extract id from URL
|
|
mobj = re.match(self._VALID_URL, url)
|
|
if mobj is None:
|
|
raise ExtractorError(u'Invalid URL: %s' % url)
|
|
|
|
video_id = mobj.group('id')
|
|
|
|
video_extension = mobj.group('ext')
|
|
|
|
# Retrieve video webpage to extract further information
|
|
webpage = self._download_webpage(url, video_id)
|
|
|
|
# Extract URL, uploader, and title from webpage
|
|
self.report_extraction(video_id)
|
|
# We try first by looking the javascript code:
|
|
mobj = re.search(r'Pb\.Data\.Shared\.put\(Pb\.Data\.Shared\.MEDIA, (?P<json>.*?)\);', webpage)
|
|
if mobj is not None:
|
|
info = json.loads(mobj.group('json'))
|
|
return [{
|
|
'id': video_id,
|
|
'url': info[u'downloadUrl'],
|
|
'uploader': info[u'username'],
|
|
'upload_date': datetime.date.fromtimestamp(info[u'creationDate']).strftime('%Y%m%d'),
|
|
'title': info[u'title'],
|
|
'ext': video_extension,
|
|
'thumbnail': info[u'thumbUrl'],
|
|
}]
|
|
|
|
# We try looking in other parts of the webpage
|
|
video_url = self._search_regex(r'<link rel="video_src" href=".*\?file=([^"]+)" />',
|
|
webpage, u'video URL')
|
|
|
|
mobj = re.search(r'<title>(.*) video by (.*) - Photobucket</title>', webpage)
|
|
if mobj is None:
|
|
raise ExtractorError(u'Unable to extract title')
|
|
video_title = mobj.group(1).decode('utf-8')
|
|
video_uploader = mobj.group(2).decode('utf-8')
|
|
|
|
return [{
|
|
'id': video_id.decode('utf-8'),
|
|
'url': video_url.decode('utf-8'),
|
|
'uploader': video_uploader,
|
|
'upload_date': None,
|
|
'title': video_title,
|
|
'ext': video_extension.decode('utf-8'),
|
|
}]
|