2
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-12-17 00:23:14 +00:00
yt-dlp/yt_dlp/extractor/flickr.py

113 lines
4.5 KiB
Python
Raw Normal View History

import urllib.parse
2013-06-23 20:31:12 +00:00
from .common import InfoExtractor
from ..utils import (
2015-11-01 12:38:11 +00:00
ExtractorError,
format_field,
2015-11-01 01:08:19 +00:00
int_or_none,
qualities,
2013-06-23 20:31:12 +00:00
)
class FlickrIE(InfoExtractor):
2015-11-01 01:08:19 +00:00
_VALID_URL = r'https?://(?:www\.|secure\.)?flickr\.com/photos/[\w\-_@]+/(?P<id>\d+)'
2013-06-27 18:46:46 +00:00
_TEST = {
2014-01-17 02:07:01 +00:00
'url': 'http://www.flickr.com/photos/forestwander-nature-pictures/5645318632/in/photostream/',
2015-11-01 01:08:19 +00:00
'md5': '164fe3fa6c22e18d448d4d5af2330f31',
2014-01-17 02:07:01 +00:00
'info_dict': {
2014-09-24 08:46:33 +00:00
'id': '5645318632',
2015-11-01 01:08:19 +00:00
'ext': 'mpg',
'description': 'Waterfalls in the Springtime at Dark Hollow Waterfalls. These are located just off of Skyline Drive in Virginia. They are only about 6/10 of a mile hike but it is a pretty steep hill and a good climb back up.',
'title': 'Dark Hollow Waterfalls',
'duration': 19,
'timestamp': 1303528740,
'upload_date': '20110423',
'uploader_id': '10922353@N03',
'uploader': 'Forest Wander',
'uploader_url': 'https://www.flickr.com/photos/forestwander-nature-pictures/',
2015-11-01 01:08:19 +00:00
'comment_count': int,
2015-12-20 23:20:22 +00:00
'view_count': int,
'tags': list,
'license': 'Attribution-ShareAlike',
},
2013-06-27 18:46:46 +00:00
}
2015-11-01 01:08:19 +00:00
_API_BASE_URL = 'https://api.flickr.com/services/rest?'
# https://help.yahoo.com/kb/flickr/SLN25525.html
_LICENSES = {
'0': 'All Rights Reserved',
'1': 'Attribution-NonCommercial-ShareAlike',
'2': 'Attribution-NonCommercial',
'3': 'Attribution-NonCommercial-NoDerivs',
'4': 'Attribution',
'5': 'Attribution-ShareAlike',
'6': 'Attribution-NoDerivs',
'7': 'No known copyright restrictions',
'8': 'United States government work',
'9': 'Public Domain Dedication (CC0)',
'10': 'Public Domain Work',
}
2015-04-22 17:24:14 +00:00
def _call_api(self, method, video_id, api_key, note, secret=None):
2015-11-01 01:08:19 +00:00
query = {
'photo_id': video_id,
'method': f'flickr.{method}',
'api_key': api_key,
2015-11-01 01:08:19 +00:00
'format': 'json',
'nojsoncallback': 1,
}
if secret:
query['secret'] = secret
data = self._download_json(self._API_BASE_URL + urllib.parse.urlencode(query), video_id, note)
2015-11-01 12:38:11 +00:00
if data['stat'] != 'ok':
raise ExtractorError(data['message'])
return data
2013-06-23 20:31:12 +00:00
2015-11-01 01:08:19 +00:00
def _real_extract(self, url):
video_id = self._match_id(url)
2013-06-23 20:31:12 +00:00
2015-12-21 21:10:16 +00:00
api_key = self._download_json(
'https://www.flickr.com/hermes_error_beacon.gne', video_id,
'Downloading api key')['site_key']
2015-12-21 21:10:16 +00:00
video_info = self._call_api(
'photos.getInfo', video_id, api_key, 'Downloading video info')['photo']
2015-11-01 01:08:19 +00:00
if video_info['media'] == 'video':
2015-12-21 21:10:16 +00:00
streams = self._call_api(
'video.getStreamInfo', video_id, api_key,
'Downloading streams info', video_info['secret'])['streams']
2013-06-23 20:31:12 +00:00
2015-12-21 21:10:16 +00:00
preference = qualities(
['288p', 'iphone_wifi', '100', '300', '700', '360p', 'appletv', '720p', '1080p', 'orig'])
2013-06-23 20:31:12 +00:00
2015-11-01 01:08:19 +00:00
formats = []
for stream in streams['stream']:
stream_type = str(stream.get('type'))
2015-11-01 01:08:19 +00:00
formats.append({
'format_id': stream_type,
'url': stream['_content'],
'quality': preference(stream_type),
2015-11-01 01:08:19 +00:00
})
2013-06-23 20:31:12 +00:00
2015-11-01 01:08:19 +00:00
owner = video_info.get('owner', {})
uploader_id = owner.get('nsid')
uploader_path = owner.get('path_alias') or uploader_id
uploader_url = format_field(uploader_path, None, 'https://www.flickr.com/photos/%s/')
2013-06-23 20:31:12 +00:00
2015-11-01 01:08:19 +00:00
return {
'id': video_id,
'title': video_info['title']['_content'],
'description': video_info.get('description', {}).get('_content'),
'formats': formats,
'timestamp': int_or_none(video_info.get('dateuploaded')),
'duration': int_or_none(video_info.get('video', {}).get('duration')),
'uploader_id': uploader_id,
2015-11-01 01:08:19 +00:00
'uploader': owner.get('realname'),
'uploader_url': uploader_url,
2015-11-01 01:08:19 +00:00
'comment_count': int_or_none(video_info.get('comments', {}).get('_content')),
2015-12-20 23:20:22 +00:00
'view_count': int_or_none(video_info.get('views')),
'tags': [tag.get('_content') for tag in video_info.get('tags', {}).get('tag', [])],
'license': self._LICENSES.get(video_info.get('license')),
2015-11-01 01:08:19 +00:00
}
else:
raise ExtractorError('not a video', expected=True)