Madeorsk/youtube-dl
1
0
Fork 0
mirror of https://codeberg.org/polarisfm/youtube-dl synced 2024-11-26 02:14:32 +01:00
Code Issues Releases Wiki Activity

Update code style for cspan (flake8)

Browse Source
This commit is contained in:
Grabien 2020-07-29 14:00:08 +04:00
parent a9c21fa9ed
commit b30047431e
1 changed files with 15 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
youtube_dl/extractor/cspan.py
View File

@ -16,13 +16,14 @@ from ..utils import (
from .senateisvp import SenateISVPIE from .senateisvp import SenateISVPIE
from .ustream import UstreamIE from .ustream import UstreamIE
remove_row = re.compile("^\s+|\n|\r|\t|\s+$") remove_row = re.compile(r"^\s+|\n|\r|\t|\s+$")
remove_all_space = re.compile("\s+") remove_all_space = re.compile(r"\s+")
m3u8_files_start = re.compile('''jwsetup=\{playlist:\[\{sources:\[''') m3u8_files_start = re.compile(r"jwsetup={playlist:\[{sources:\[")
m3u8_files_end = re.compile("],") m3u8_files_end = re.compile(r"],")
m3u8_file = re.compile("{file:'(.+)'}") m3u8_file = re.compile(r"{file:'(.+)'}")
def getM3u8Files(data):
def get_m3u8_files(data):
data = remove_row.sub("", data) data = remove_row.sub("", data)
data = remove_all_space.sub("", data) data = remove_all_space.sub("", data)
si = m3u8_files_start.search(data) si = m3u8_files_start.search(data)
@ -34,6 +35,7 @@ def getM3u8Files(data):
files.append(ff.group(1)) files.append(ff.group(1))
return files return files
class CSpanIE(InfoExtractor): class CSpanIE(InfoExtractor):
_VALID_URL = r'https?://(?:www\.)?c-span\.org/video/\?(?P<id>[0-9a-f]+)' _VALID_URL = r'https?://(?:www\.)?c-span\.org/video/\?(?P<id>[0-9a-f]+)'
IE_DESC = 'C-SPAN' IE_DESC = 'C-SPAN'
@ -48,7 +50,6 @@ class CSpanIE(InfoExtractor):
'skip': 'Regularly fails on travis, for unknown reasons', 'skip': 'Regularly fails on travis, for unknown reasons',
}, { }, {
'url': 'http://www.c-span.org/video/?c4486943/cspan-international-health-care-models', 'url': 'http://www.c-span.org/video/?c4486943/cspan-international-health-care-models',
'url': 'https://www.c-span.org/video/?465817-1/president-trump-addresses-international-association-chiefs-police-conference',
# md5 is unstable # md5 is unstable
'info_dict': { 'info_dict': {
'id': 'c4486943', 'id': 'c4486943',
@ -147,7 +148,8 @@ class CSpanIE(InfoExtractor):
def get_text_attr(d, attr): def get_text_attr(d, attr):
return d.get(attr, {}).get('#text') return d.get(attr, {}).get('#text')
json_url = "http://www.c-span.org/assets/player/ajax-player.php?os=android&html5=%s&id=%s" % (video_type, video_id) json_url = "http://www.c-span.org/assets/player/ajax-player.php?os=android&html5=%s&id=%s" \
% (video_type, video_id)
xml_url = 'http://www.c-span.org/common/services/flashXml.php?%sid=%s' % (video_type, video_id) xml_url = 'http://www.c-span.org/common/services/flashXml.php?%sid=%s' % (video_type, video_id)
data = self._download_json(json_url, video_id)['video'] data = self._download_json(json_url, video_id)['video']
@ -160,11 +162,12 @@ class CSpanIE(InfoExtractor):
entries = [] entries = []
if data['@status'] != 'Success': if data['@status'] != 'Success':
try: try:
files = getM3u8Files(webpage) files = get_m3u8_files(webpage)
for partnum, path in enumerate(files): for partnum, path in enumerate(files):
formats = self._extract_m3u8_formats( formats = self._extract_m3u8_formats(
path, video_id, 'mp4', path, video_id, 'mp4',
entry_protocol='m3u8_native', m3u8_id='hls') if determine_ext(path) == 'm3u8' else [{'url': path, }] entry_protocol='m3u8_native', m3u8_id='hls') if determine_ext(path) == 'm3u8' \
else [{'url': path, }]
for f in formats: for f in formats:
f.setdefault('http_headers', {})['Referer'] = url f.setdefault('http_headers', {})['Referer'] = url
self._sort_formats(formats) self._sort_formats(formats)
@ -201,8 +204,8 @@ class CSpanIE(InfoExtractor):
formats = self._extract_m3u8_formats( formats = self._extract_m3u8_formats(
path, video_id, 'mp4', entry_protocol='m3u8_native', path, video_id, 'mp4', entry_protocol='m3u8_native',
m3u8_id='hls') if determine_ext(path) == 'm3u8' else [{'url': path, }] m3u8_id='hls') if determine_ext(path) == 'm3u8' else [{'url': path, }]
for f in formats: for ff in formats:
f.setdefault('http_headers', {})['Referer'] = url ff.setdefault('http_headers', {})['Referer'] = url
self._sort_formats(formats) self._sort_formats(formats)
entries.append({ entries.append({
'id': '%s_%d' % (video_id, partnum + 1), 'id': '%s_%d' % (video_id, partnum + 1),
@ -226,12 +229,6 @@ class CSpanIE(InfoExtractor):
entry['id'] = 'c' + video_id if video_type == 'clip' else video_id entry['id'] = 'c' + video_id if video_type == 'clip' else video_id
return entry return entry
else: else:
res = {
'_type': 'playlist',
'entries': entries,
'title': title,
'id': 'c' + video_id if video_type == 'clip' else video_id,
}
return { return {
'_type': 'playlist', '_type': 'playlist',
'entries': entries, 'entries': entries,