]>
jfr.im git - yt-dlp.git/blob - yt_dlp/extractor/reddit.py
2 from urllib
.parse
import urlparse
4 from .common
import InfoExtractor
16 class RedditIE(InfoExtractor
):
17 _VALID_URL
= r
'https?://(?P<subdomain>[^/]+\.)?reddit(?:media)?\.com/r/(?P<slug>[^/]+/comments/(?P<id>[^/?#&]+))'
19 'url': 'https://www.reddit.com/r/videos/comments/6rrwyj/that_small_heart_attack/',
23 'display_id': '6rrwyj',
24 'title': 'That small heart attack.',
25 'thumbnail': r
're:^https?://.*\.(?:jpg|png)',
26 'thumbnails': 'count:4',
27 'timestamp': 1501941939,
28 'upload_date': '20170805',
37 'skip_download': True,
40 'url': 'https://www.reddit.com/r/videos/comments/6rrwyj',
41 'only_matching': True,
44 'url': 'https://www.reddit.com/r/MadeMeSmile/comments/6t7wi5/wait_for_it/',
45 'only_matching': True,
48 'url': 'https://old.reddit.com/r/MadeMeSmile/comments/6t7wi5/wait_for_it/',
49 'only_matching': True,
52 'url': 'https://www.reddit.com/r/videos/comments/6t7sg9/comedians_hilarious_joke_about_the_guam_flag/',
53 'only_matching': True,
56 'url': 'https://www.reddit.com/r/videos/comments/6t75wq/southern_man_tries_to_speak_without_an_accent/',
57 'only_matching': True,
59 # reddit video @ nm reddit
60 'url': 'https://nm.reddit.com/r/Cricket/comments/8idvby/lousy_cameraman_finds_himself_in_cairns_line_of/',
61 'only_matching': True,
63 'url': 'https://www.redditmedia.com/r/serbia/comments/pu9wbx/ako_vu%C4%8Di%C4%87_izgubi_izbore_ja_%C4%87u_da_crknem/',
64 'only_matching': True,
68 def _gen_session_id():
70 rand_max
= 1 << (id_length
* 4)
71 return '%0.*x' % (id_length
, random
.randrange(rand_max
))
73 def _real_extract(self
, url
):
74 subdomain
, slug
, video_id
= self
._match
_valid
_url
(url
).group('subdomain', 'slug', 'id')
76 self
._set
_cookie
('.reddit.com', 'reddit_session', self
._gen
_session
_id
())
77 self
._set
_cookie
('.reddit.com', '_options', '%7B%22pref_quarantine_optin%22%3A%20true%7D')
78 data
= self
._download
_json
(f
'https://{subdomain}reddit.com/r/{slug}/.json', video_id
, fatal
=False)
80 # Fall back to old.reddit.com in case the requested subdomain fails
81 data
= self
._download
_json
(f
'https://old.reddit.com/r/{slug}/.json', video_id
)
82 data
= data
[0]['data']['children'][0]['data']
83 video_url
= data
['url']
85 # Avoid recursing into the same reddit URL
86 if 'reddit.com/' in video_url
and '/%s/' % video_id
in video_url
:
87 raise ExtractorError('No media found', expected
=True)
89 over_18
= data
.get('over_18')
92 elif over_18
is False:
99 def add_thumbnail(src
):
100 if not isinstance(src
, dict):
102 thumbnail_url
= url_or_none(src
.get('url'))
103 if not thumbnail_url
:
106 'url': unescapeHTML(thumbnail_url
),
107 'width': int_or_none(src
.get('width')),
108 'height': int_or_none(src
.get('height')),
111 for image
in try_get(data
, lambda x
: x
['preview']['images']) or []:
112 if not isinstance(image
, dict):
114 add_thumbnail(image
.get('source'))
115 resolutions
= image
.get('resolutions')
116 if isinstance(resolutions
, list):
117 for resolution
in resolutions
:
118 add_thumbnail(resolution
)
121 'title': data
.get('title'),
122 'thumbnails': thumbnails
,
123 'timestamp': float_or_none(data
.get('created_utc')),
124 'uploader': data
.get('author'),
125 'like_count': int_or_none(data
.get('ups')),
126 'dislike_count': int_or_none(data
.get('downs')),
127 'comment_count': int_or_none(data
.get('num_comments')),
128 'age_limit': age_limit
,
131 # Check if media is hosted on reddit:
132 reddit_video
= traverse_obj(data
, (('media', 'secure_media'), 'reddit_video'), get_all
=False)
135 try_get(reddit_video
, lambda x
: unescapeHTML(x
[y
]))
136 for y
in ('dash_url', 'hls_url')
140 display_id
= video_id
141 video_id
= self
._search
_regex
(
142 r
'https?://v\.redd\.it/(?P<id>[^/?#&]+)', reddit_video
['fallback_url'],
143 'video_id', default
=display_id
)
145 dash_playlist_url
= playlist_urls
[0] or f
'https://v.redd.it/{video_id}/DASHPlaylist.mpd'
146 hls_playlist_url
= playlist_urls
[1] or f
'https://v.redd.it/{video_id}/HLSPlaylist.m3u8'
148 formats
= self
._extract
_m
3u8_formats
(
149 hls_playlist_url
, display_id
, 'mp4',
150 entry_protocol
='m3u8_native', m3u8_id
='hls', fatal
=False)
151 formats
.extend(self
._extract
_mpd
_formats
(
152 dash_playlist_url
, display_id
, mpd_id
='dash', fatal
=False))
153 self
._sort
_formats
(formats
)
158 'display_id': display_id
,
160 'duration': int_or_none(reddit_video
.get('duration')),
163 parsed_url
= urlparse(video_url
)
164 if parsed_url
.netloc
== 'v.redd.it':
165 self
.raise_no_formats('This video is processing', expected
=True, video_id
=video_id
)
168 'id': parsed_url
.path
.split('/')[1],
169 'display_id': video_id
,
172 # Not hosted on reddit, must continue extraction
175 'display_id': video_id
,
176 '_type': 'url_transparent',