X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=youtube_dl%2Fextractor%2Fpbs.py;h=64cded70789249746a5e2b6604d86563a6ad499c;hb=7bb49d1057527776c3b4b4ccf2bd995a291c883d;hp=7444b7b5bae908cf0dc7740a1a5ca68f297cf7c0;hpb=acf1555d76f55c05516e23e45262dfd95df77314;p=youtube-dl diff --git a/youtube_dl/extractor/pbs.py b/youtube_dl/extractor/pbs.py index 7444b7b5b..64cded707 100644 --- a/youtube_dl/extractor/pbs.py +++ b/youtube_dl/extractor/pbs.py @@ -1,20 +1,22 @@ from __future__ import unicode_literals import re -import json from .common import InfoExtractor +from ..utils import ( + US_RATINGS, +) class PBSIE(InfoExtractor): _VALID_URL = r'''(?x)https?:// (?: # Direct video URL - video\.pbs\.org/video/(?P[0-9]+)/? | + video\.pbs\.org/(?:viralplayer|video)/(?P[0-9]+)/? | # Article with embedded player (?:www\.)?pbs\.org/(?:[^/]+/){2,5}(?P[^/]+)/?(?:$|[?\#]) | # Player - video\.pbs\.org/partnerplayer/(?P[^/]+)/ + video\.pbs\.org/(?:widget/)?partnerplayer/(?P[^/]+)/ ) ''' @@ -58,6 +60,11 @@ class PBSIE(InfoExtractor): info_url = 'http://video.pbs.org/videoInfo/%s?format=json' % video_id info = self._download_json(info_url, display_id) + rating_str = info.get('rating') + if rating_str is not None: + rating_str = rating_str.rpartition('-')[2] + age_limit = US_RATINGS.get(rating_str) + return { 'id': video_id, 'title': info['title'], @@ -66,4 +73,5 @@ class PBSIE(InfoExtractor): 'description': info['program'].get('description'), 'thumbnail': info.get('image_url'), 'duration': info.get('duration'), + 'age_limit': age_limit, }