This commit is contained in:
Jasmine Hou 2024-08-21 22:32:43 -04:00 committed by GitHub
commit 96274a354b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -5,48 +5,52 @@ import re
from .common import InfoExtractor
from ..utils import (
ExtractorError,
parse_duration,
unsmuggle_url,
url_or_none,
)
from ..compat import (
compat_parse_qs,
compat_urlparse,
)
from ..compat import compat_parse_qs
class SenateISVPIE(InfoExtractor):
# [committee, stream_number, stream_domain, stream_id, msl3]
_COMM_MAP = [
['ag', '76440', 'http://ag-f.akamaihd.net'],
['aging', '76442', 'http://aging-f.akamaihd.net'],
['approps', '76441', 'http://approps-f.akamaihd.net'],
['armed', '76445', 'http://armed-f.akamaihd.net'],
['banking', '76446', 'http://banking-f.akamaihd.net'],
['budget', '76447', 'http://budget-f.akamaihd.net'],
['cecc', '76486', 'http://srs-f.akamaihd.net'],
['commerce', '80177', 'http://commerce1-f.akamaihd.net'],
['csce', '75229', 'http://srs-f.akamaihd.net'],
['dpc', '76590', 'http://dpc-f.akamaihd.net'],
['energy', '76448', 'http://energy-f.akamaihd.net'],
['epw', '76478', 'http://epw-f.akamaihd.net'],
['ethics', '76449', 'http://ethics-f.akamaihd.net'],
['finance', '76450', 'http://finance-f.akamaihd.net'],
['foreign', '76451', 'http://foreign-f.akamaihd.net'],
['govtaff', '76453', 'http://govtaff-f.akamaihd.net'],
['help', '76452', 'http://help-f.akamaihd.net'],
['indian', '76455', 'http://indian-f.akamaihd.net'],
['intel', '76456', 'http://intel-f.akamaihd.net'],
['intlnarc', '76457', 'http://intlnarc-f.akamaihd.net'],
['jccic', '85180', 'http://jccic-f.akamaihd.net'],
['jec', '76458', 'http://jec-f.akamaihd.net'],
['judiciary', '76459', 'http://judiciary-f.akamaihd.net'],
['rpc', '76591', 'http://rpc-f.akamaihd.net'],
['rules', '76460', 'http://rules-f.akamaihd.net'],
['saa', '76489', 'http://srs-f.akamaihd.net'],
['smbiz', '76461', 'http://smbiz-f.akamaihd.net'],
['srs', '75229', 'http://srs-f.akamaihd.net'],
['uscc', '76487', 'http://srs-f.akamaihd.net'],
['vetaff', '76462', 'http://vetaff-f.akamaihd.net'],
['arch', '', 'http://ussenate-f.akamaihd.net/']
['ag', '76440', 'http://ag-f.akamaihd.net', '2036803', 'agriculture'],
['aging', '76442', 'http://aging-f.akamaihd.net', '2036801', 'aging'],
['approps', '76441', 'http://approps-f.akamaihd.net', '2036802', 'appropriations'],
['armed', '76445', 'http://armed-f.akamaihd.net', '2036800', 'armedservices'],
['banking', '76446', 'http://banking-f.akamaihd.net', '2036799', 'banking'],
['budget', '76447', 'http://budget-f.akamaihd.net', '2036798', 'budget'],
['cecc', '76486', 'http://srs-f.akamaihd.net', '2036782', 'srs_cecc'],
['commerce', '80177', 'http://commerce1-f.akamaihd.net', '2036779', 'commerce'],
['csce', '75229', 'http://srs-f.akamaihd.net', '2036777', 'srs_srs'],
['dpc', '76590', 'http://dpc-f.akamaihd.net', None, 'dpc'],
['energy', '76448', 'http://energy-f.akamaihd.net', '2036797', 'energy'],
['epw', '76478', 'http://epw-f.akamaihd.net', '2036783', 'environment'],
['ethics', '76449', 'http://ethics-f.akamaihd.net', '2036796', 'ethics'],
['finance', '76450', 'http://finance-f.akamaihd.net', '2036795', 'finance_finance'],
['foreign', '76451', 'http://foreign-f.akamaihd.net', '2036794', 'foreignrelations'],
['govtaff', '76453', 'http://govtaff-f.akamaihd.net', '2036792', 'hsgac'],
['help', '76452', 'http://help-f.akamaihd.net', '2036793', 'help'],
['indian', '76455', 'http://indian-f.akamaihd.net', '2036791', 'indianaffairs'],
['intel', '76456', 'http://intel-f.akamaihd.net', '2036790', 'intelligence'],
['intlnarc', '76457', 'http://intlnarc-f.akamaihd.net', None, 'internationalnarcoticscaucus'],
['jccic', '85180', 'http://jccic-f.akamaihd.net', '2036778', 'jccic'],
['jec', '76458', 'http://jec-f.akamaihd.net', '2036789', 'jointeconomic'],
['judiciary', '76459', 'http://judiciary-f.akamaihd.net', '2036788', 'judiciary'],
['rpc', '76591', 'http://rpc-f.akamaihd.net', None, 'rpc'],
['rules', '76460', 'http://rules-f.akamaihd.net', '2036787', 'rules'],
['saa', '76489', 'http://srs-f.akamaihd.net', '2036780', 'srs_saa'],
['smbiz', '76461', 'http://smbiz-f.akamaihd.net', '2036786', 'smallbusiness'],
['srs', '75229', 'http://srs-f.akamaihd.net', '2031966', 'srs_srs'],
['uscc', '76487', 'http://srs-f.akamaihd.net', '2036781', 'srs_uscc'],
['vetaff', '76462', 'http://vetaff-f.akamaihd.net', '2036785', 'veteransaffairs'],
['arch', '', 'http://ussenate-f.akamaihd.net/', None, None],
['uscp', None, '', '2043685', 'uscp'],
['cio', None, '', '2043686', 'cio']
]
_IE_NAME = 'senate.gov'
_VALID_URL = r'https?://(?:www\.)?senate\.gov/isvp/?\?(?P<qs>.+)'
_TESTS = [{
@ -54,100 +58,120 @@ class SenateISVPIE(InfoExtractor):
'info_dict': {
'id': 'judiciary031715',
'ext': 'mp4',
'title': 'Integrated Senate Video Player',
'thumbnail': r're:^https?://.*\.(?:jpg|png)$',
},
'params': {
# m3u8 download
'skip_download': True,
},
}, {
'url': 'http://www.senate.gov/isvp/?type=live&comm=commerce&filename=commerce011514.mp4&auto_play=false',
'info_dict': {
'id': 'commerce011514',
'ext': 'mp4',
'title': 'Integrated Senate Video Player'
},
'params': {
# m3u8 download
'skip_download': True,
'title': 'judiciary031715',
'thumbnail': 'http://www.judiciary.senate.gov/themes/judiciary/images/video-poster-flash-fit.png',
},
'expected_warnings': ['Failed to download m3u8 information: HTTP Error 404: Not Found'],
}, {
'url': 'http://www.senate.gov/isvp/?type=arch&comm=intel&filename=intel090613&hc_location=ufi',
# checksum differs each time
'info_dict': {
'id': 'intel090613',
'ext': 'mp4',
'title': 'Integrated Senate Video Player'
}
'title': 'intel090613',
},
'expected_warnings': ['Failed to download m3u8 information: HTTP Error 404: Not Found'],
}, {
# From http://www.c-span.org/video/?96791-1
'url': 'http://www.senate.gov/isvp?type=live&comm=banking&filename=banking012715',
'only_matching': True,
'url': 'https://www.senate.gov/isvp/?comm=govtaff&type=archv&stt=975&filename=govtaff111722&auto_play=false&poster=https%3A%2F%2Fwww%2Ehsgac%2Esenate%2Egov%2Fimages%2Fvideo%2Dposter%2Dflash%2Dfit%2Epng',
'info_dict': {
'id': 'govtaff111722',
'ext': 'mp4',
'title': 'govtaff111722',
'thumbnail': 'https://www.hsgac.senate.gov/images/video-poster-flash-fit.png',
},
}, {
'url': 'https://www.senate.gov/isvp/?type=arch&comm=energy&filename=energy111722&stt=00:22:30&auto_play=false&wmode=transparent&poster=https%3A%2F%2Fwww%2Eenergy%2Esenate%2Egov%2Fthemes%2Fenergy%2Fimages%2Fvideo%2Dposter%2Dflash%2Dfit%2Epng',
'info_dict': {
'id': 'energy111722',
'ext': 'mp4',
'title': 'energy111722',
'thumbnail': 'https://www.energy.senate.gov/themes/energy/images/video-poster-flash-fit.png',
},
}, {
'url': 'https://www.senate.gov/isvp/?comm=foreign&type=archv&stt=0&filename=foreign080322&auto_play=false&wmode=transparent&poster=https%3A%2F%2Fwww%2Eforeign%2Esenate%2Egov%2Fthemes%2Fforeign%2Fimages%2Fvideo%2Dposter%2Dflash%2Dfit%2Epng',
'info_dict': {
'id': 'foreign080322',
'ext': 'mp4',
'title': 'foreign080322',
'thumbnail': 'https://www.foreign.senate.gov/themes/foreign/images/video-poster-flash-fit.png',
},
}]
@staticmethod
# returns url from an iframe
def _search_iframe_url(webpage):
mobj = re.search(
r"<iframe[^>]+src=['\"](?P<url>https?://www\.senate\.gov/isvp/?\?[^'\"]+)['\"]",
r'''<iframe\b[^>]+\bsrc\s*=\s*(['"])(?P<url>https?://www\.senate\.gov/isvp/?\?(?:(?!\1)\S)+)''',
webpage)
if mobj:
return mobj.group('url')
@staticmethod
def __dictify(cm):
return dict((row[0], (row[1:] + ['', ''])[:4]) for row in cm)
# returns stream_number, stream_domain, stream_id, msl3
def _get_info_for_comm(self, committee):
for entry in self._COMM_MAP:
if entry[0] == committee:
return entry[1:]
dict = self.__dictify(self._COMM_MAP)
return dict[committee][0:]
def _real_extract(self, url):
# smuggled data may contain a forced title that should be used
url, smuggled_data = unsmuggle_url(url, {})
qs = compat_parse_qs(re.match(self._VALID_URL, url).group('qs'))
if not qs.get('filename') or not qs.get('type') or not qs.get('comm'):
raise ExtractorError('Invalid URL', expected=True)
video_id = re.sub(r'.mp4$', '', qs['filename'][0])
# error handling for invalid URL - specify which error
if not qs.get('filename'):
raise ExtractorError('Invalid URL. Missing filename in query parameters', expected=True)
if not qs.get('comm'):
raise ExtractorError('Invalid URL. Missing committee in query parameters', expected=True)
webpage = self._download_webpage(url, video_id)
committee = qs.get('comm')[0]
filename = qs.get('filename')[0]
video_id = re.sub(r'\.mp4$', '', filename)
if smuggled_data.get('force_title'):
title = smuggled_data['force_title']
else:
title = self._html_search_regex(r'<title>([^<]+)</title>', webpage, video_id)
poster = qs.get('poster')
thumbnail = poster[0] if poster else None
# there is no point in pulling the title from the webpage since it always defaults to 'Integrated Senate Player'
title = smuggled_data.get('force_title') or filename
video_type = qs['type'][0]
committee = video_type if video_type == 'arch' else qs['comm'][0]
stream_num, domain = self._get_info_for_comm(committee)
stream_number, stream_domain, stream_id, msl3 = self._get_info_for_comm(committee)
stream_number = int(stream_number)
stream_id = int(stream_id)
# the possible locations for the video: only the first has been seen in use
possible_manifest_urls = [
'https://www-senate-gov-media-srs.akamaized.net/hls/live/%d/%s/%s/master.m3u8' % (stream_id, committee, filename),
'https://www-senate-gov-msl3archive.akamaized.net/%s/%s_1/master.m3u8' % (msl3, filename),
'%s/i/%s_1@%d/master.m3u8' % (stream_domain, filename, stream_number),
'https://ussenate-f.akamaihd.net/i/%s.mp4/master.m3u8' % video_id,
]
# we iterate through the possible locations until we find formats
formats = []
if video_type == 'arch':
filename = video_id if '.' in video_id else video_id + '.mp4'
formats = [{
# All parameters in the query string are necessary to prevent a 403 error
'url': compat_urlparse.urljoin(domain, filename) + '?v=3.1.0&fp=&r=&g=',
}]
else:
hdcore_sign = 'hdcore=3.1.0'
url_params = (domain, video_id, stream_num)
f4m_url = '%s/z/%s_1@%s/manifest.f4m?' % url_params + hdcore_sign
m3u8_url = '%s/i/%s_1@%s/master.m3u8' % url_params
for entry in self._extract_f4m_formats(f4m_url, video_id, f4m_id='f4m'):
# URLs without the extra param induce an 404 error
entry.update({'extra_param_to_segment_url': hdcore_sign})
formats.append(entry)
for entry in self._extract_m3u8_formats(m3u8_url, video_id, ext='mp4', m3u8_id='m3u8'):
mobj = re.search(r'(?P<tag>(?:-p|-b)).m3u8', entry['url'])
for url in possible_manifest_urls:
entries = self._extract_m3u8_formats(
url, video_id, ext='mp4', m3u8_id='hls',
entry_protocol='m3u8_native', fatal=False)
for entry in entries:
mobj = re.search(r'(?P<tag>-[pb]).m3u8', entry['url'])
if mobj:
entry['format_id'] += mobj.group('tag')
formats.append(entry)
self._sort_formats(formats)
if formats:
break
self._sort_formats(formats)
thumbnail = url_or_none(qs.get('poster', [None])[-1])
start_time = parse_duration(qs.get('stt', [None])[-1])
stop_time = parse_duration(qs.get('dur', [None])[-1])
if stop_time is not None:
stop_time += start_time or 0
return {
'id': video_id,
'title': title,
'formats': formats,
'thumbnail': thumbnail,
'start_time': start_time,
'stop_time': stop_time,
}