mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-20 01:42:50 +01:00
[cleanup] Misc
This commit is contained in:
parent
c681cb5d93
commit
b868936cd6
@ -11,5 +11,4 @@ else
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
echo python3 -m pytest -k $test_set
|
|
||||||
python3 -m pytest -k "$test_set"
|
python3 -m pytest -k "$test_set"
|
||||||
|
@ -8,13 +8,14 @@
|
|||||||
import unittest
|
import unittest
|
||||||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||||
|
|
||||||
from test.helper import try_rm
|
from test.helper import is_download_test, try_rm
|
||||||
|
|
||||||
|
|
||||||
root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||||||
download_file = join(root_dir, 'test.webm')
|
download_file = join(root_dir, 'test.webm')
|
||||||
|
|
||||||
|
|
||||||
|
@is_download_test
|
||||||
class TestOverwrites(unittest.TestCase):
|
class TestOverwrites(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
# create an empty file
|
# create an empty file
|
||||||
|
@ -515,8 +515,15 @@ def check_deprecated(param, option, suggestion):
|
|||||||
self.report_warning('--merge-output-format will be ignored since --remux-video or --recode-video is given')
|
self.report_warning('--merge-output-format will be ignored since --remux-video or --recode-video is given')
|
||||||
self.params['merge_output_format'] = self.params['final_ext']
|
self.params['merge_output_format'] = self.params['final_ext']
|
||||||
|
|
||||||
if 'overwrites' in self.params and self.params['overwrites'] is None:
|
if self.params.get('overwrites') is None:
|
||||||
del self.params['overwrites']
|
self.params.pop('overwrites', None)
|
||||||
|
elif self.params.get('nooverwrites') is not None:
|
||||||
|
# nooverwrites was unnecessarily changed to overwrites
|
||||||
|
# in 0c3d0f51778b153f65c21906031c2e091fcfb641
|
||||||
|
# This ensures compatibility with both keys
|
||||||
|
self.params['overwrites'] = not self.params['nooverwrites']
|
||||||
|
else:
|
||||||
|
self.params['nooverwrites'] = not self.params['overwrites']
|
||||||
|
|
||||||
if params.get('bidi_workaround', False):
|
if params.get('bidi_workaround', False):
|
||||||
try:
|
try:
|
||||||
@ -889,7 +896,6 @@ def validate_outtmpl(cls, outtmpl):
|
|||||||
def prepare_outtmpl(self, outtmpl, info_dict, sanitize=None):
|
def prepare_outtmpl(self, outtmpl, info_dict, sanitize=None):
|
||||||
""" Make the template and info_dict suitable for substitution : ydl.outtmpl_escape(outtmpl) % info_dict """
|
""" Make the template and info_dict suitable for substitution : ydl.outtmpl_escape(outtmpl) % info_dict """
|
||||||
info_dict.setdefault('epoch', int(time.time())) # keep epoch consistent once set
|
info_dict.setdefault('epoch', int(time.time())) # keep epoch consistent once set
|
||||||
na = self.params.get('outtmpl_na_placeholder', 'NA')
|
|
||||||
|
|
||||||
info_dict = dict(info_dict) # Do not sanitize so as not to consume LazyList
|
info_dict = dict(info_dict) # Do not sanitize so as not to consume LazyList
|
||||||
for key in ('__original_infodict', '__postprocessors'):
|
for key in ('__original_infodict', '__postprocessors'):
|
||||||
@ -970,6 +976,8 @@ def get_value(mdict):
|
|||||||
|
|
||||||
return value
|
return value
|
||||||
|
|
||||||
|
na = self.params.get('outtmpl_na_placeholder', 'NA')
|
||||||
|
|
||||||
def _dumpjson_default(obj):
|
def _dumpjson_default(obj):
|
||||||
if isinstance(obj, (set, LazyList)):
|
if isinstance(obj, (set, LazyList)):
|
||||||
return list(obj)
|
return list(obj)
|
||||||
@ -978,10 +986,7 @@ def _dumpjson_default(obj):
|
|||||||
def create_key(outer_mobj):
|
def create_key(outer_mobj):
|
||||||
if not outer_mobj.group('has_key'):
|
if not outer_mobj.group('has_key'):
|
||||||
return f'%{outer_mobj.group(0)}'
|
return f'%{outer_mobj.group(0)}'
|
||||||
|
|
||||||
prefix = outer_mobj.group('prefix')
|
|
||||||
key = outer_mobj.group('key')
|
key = outer_mobj.group('key')
|
||||||
original_fmt = fmt = outer_mobj.group('format')
|
|
||||||
mobj = re.match(INTERNAL_FORMAT_RE, key)
|
mobj = re.match(INTERNAL_FORMAT_RE, key)
|
||||||
if mobj is None:
|
if mobj is None:
|
||||||
value, default, mobj = None, na, {'fields': ''}
|
value, default, mobj = None, na, {'fields': ''}
|
||||||
@ -990,6 +995,7 @@ def create_key(outer_mobj):
|
|||||||
default = mobj['default'] if mobj['default'] is not None else na
|
default = mobj['default'] if mobj['default'] is not None else na
|
||||||
value = get_value(mobj)
|
value = get_value(mobj)
|
||||||
|
|
||||||
|
fmt = outer_mobj.group('format')
|
||||||
if fmt == 's' and value is not None and key in field_size_compat_map.keys():
|
if fmt == 's' and value is not None and key in field_size_compat_map.keys():
|
||||||
fmt = '0{:d}d'.format(field_size_compat_map[key])
|
fmt = '0{:d}d'.format(field_size_compat_map[key])
|
||||||
|
|
||||||
@ -1021,9 +1027,9 @@ def create_key(outer_mobj):
|
|||||||
if fmt[-1] in 'csr':
|
if fmt[-1] in 'csr':
|
||||||
value = sanitize(mobj['fields'].split('.')[-1], value)
|
value = sanitize(mobj['fields'].split('.')[-1], value)
|
||||||
|
|
||||||
key = '%s\0%s' % (key.replace('%', '%\0'), original_fmt)
|
key = '%s\0%s' % (key.replace('%', '%\0'), outer_mobj.group('format'))
|
||||||
TMPL_DICT[key] = value
|
TMPL_DICT[key] = value
|
||||||
return f'{prefix}%({key}){fmt}'
|
return '{prefix}%({key}){fmt}'.format(key=key, fmt=fmt, prefix=outer_mobj.group('prefix'))
|
||||||
|
|
||||||
return EXTERNAL_FORMAT_RE.sub(create_key, outtmpl), TMPL_DICT
|
return EXTERNAL_FORMAT_RE.sub(create_key, outtmpl), TMPL_DICT
|
||||||
|
|
||||||
@ -1069,7 +1075,6 @@ def prepare_filename(self, info_dict, dir_type='', warn=False):
|
|||||||
self.report_warning('--paths is ignored when an outputting to stdout', only_once=True)
|
self.report_warning('--paths is ignored when an outputting to stdout', only_once=True)
|
||||||
elif os.path.isabs(filename):
|
elif os.path.isabs(filename):
|
||||||
self.report_warning('--paths is ignored since an absolute path is given in output template', only_once=True)
|
self.report_warning('--paths is ignored since an absolute path is given in output template', only_once=True)
|
||||||
self.__prepare_filename_warned = True
|
|
||||||
if filename == '-' or not filename:
|
if filename == '-' or not filename:
|
||||||
return filename
|
return filename
|
||||||
|
|
||||||
@ -1348,15 +1353,12 @@ def process_ie_result(self, ie_result, download=True, extra_info={}):
|
|||||||
'It needs to be updated.' % ie_result.get('extractor'))
|
'It needs to be updated.' % ie_result.get('extractor'))
|
||||||
|
|
||||||
def _fixup(r):
|
def _fixup(r):
|
||||||
self.add_extra_info(
|
self.add_extra_info(r, {
|
||||||
r,
|
'extractor': ie_result['extractor'],
|
||||||
{
|
'webpage_url': ie_result['webpage_url'],
|
||||||
'extractor': ie_result['extractor'],
|
'webpage_url_basename': url_basename(ie_result['webpage_url']),
|
||||||
'webpage_url': ie_result['webpage_url'],
|
'extractor_key': ie_result['extractor_key'],
|
||||||
'webpage_url_basename': url_basename(ie_result['webpage_url']),
|
})
|
||||||
'extractor_key': ie_result['extractor_key'],
|
|
||||||
}
|
|
||||||
)
|
|
||||||
return r
|
return r
|
||||||
ie_result['entries'] = [
|
ie_result['entries'] = [
|
||||||
self.process_ie_result(_fixup(r), download, extra_info)
|
self.process_ie_result(_fixup(r), download, extra_info)
|
||||||
@ -2193,7 +2195,7 @@ def is_wellformed(f):
|
|||||||
format['format'] = '{id} - {res}{note}'.format(
|
format['format'] = '{id} - {res}{note}'.format(
|
||||||
id=format['format_id'],
|
id=format['format_id'],
|
||||||
res=self.format_resolution(format),
|
res=self.format_resolution(format),
|
||||||
note=' ({0})'.format(format['format_note']) if format.get('format_note') is not None else '',
|
note=format_field(format, 'format_note', ' (%s)'),
|
||||||
)
|
)
|
||||||
# Automatically determine file extension if missing
|
# Automatically determine file extension if missing
|
||||||
if format.get('ext') is None:
|
if format.get('ext') is None:
|
||||||
@ -2395,7 +2397,7 @@ def print_optional(field):
|
|||||||
print_optional('thumbnail')
|
print_optional('thumbnail')
|
||||||
print_optional('description')
|
print_optional('description')
|
||||||
print_optional('filename')
|
print_optional('filename')
|
||||||
if self.params.get('forceduration', False) and info_dict.get('duration') is not None:
|
if self.params.get('forceduration') and info_dict.get('duration') is not None:
|
||||||
self.to_stdout(formatSeconds(info_dict['duration']))
|
self.to_stdout(formatSeconds(info_dict['duration']))
|
||||||
print_mandatory('format')
|
print_mandatory('format')
|
||||||
|
|
||||||
@ -2435,8 +2437,6 @@ def process_info(self, info_dict):
|
|||||||
|
|
||||||
assert info_dict.get('_type', 'video') == 'video'
|
assert info_dict.get('_type', 'video') == 'video'
|
||||||
|
|
||||||
info_dict.setdefault('__postprocessors', [])
|
|
||||||
|
|
||||||
max_downloads = self.params.get('max_downloads')
|
max_downloads = self.params.get('max_downloads')
|
||||||
if max_downloads is not None:
|
if max_downloads is not None:
|
||||||
if self._num_downloads >= int(max_downloads):
|
if self._num_downloads >= int(max_downloads):
|
||||||
@ -2637,6 +2637,7 @@ def _write_link_file(extension, template, newline, embed_filename):
|
|||||||
info_dict = self.run_pp(MoveFilesAfterDownloadPP(self, False), info_dict)
|
info_dict = self.run_pp(MoveFilesAfterDownloadPP(self, False), info_dict)
|
||||||
else:
|
else:
|
||||||
# Download
|
# Download
|
||||||
|
info_dict.setdefault('__postprocessors', [])
|
||||||
try:
|
try:
|
||||||
|
|
||||||
def existing_file(*filepaths):
|
def existing_file(*filepaths):
|
||||||
|
@ -320,12 +320,9 @@ def report_retry(self, err, count, retries):
|
|||||||
'[download] Got server HTTP error: %s. Retrying (attempt %d of %s) ...'
|
'[download] Got server HTTP error: %s. Retrying (attempt %d of %s) ...'
|
||||||
% (error_to_compat_str(err), count, self.format_retries(retries)))
|
% (error_to_compat_str(err), count, self.format_retries(retries)))
|
||||||
|
|
||||||
def report_file_already_downloaded(self, file_name):
|
def report_file_already_downloaded(self, *args, **kwargs):
|
||||||
"""Report file has already been fully downloaded."""
|
"""Report file has already been fully downloaded."""
|
||||||
try:
|
return self.ydl.report_file_already_downloaded(*args, **kwargs)
|
||||||
self.to_screen('[download] %s has already been downloaded' % file_name)
|
|
||||||
except UnicodeEncodeError:
|
|
||||||
self.to_screen('[download] The file has already been downloaded')
|
|
||||||
|
|
||||||
def report_unable_to_resume(self):
|
def report_unable_to_resume(self):
|
||||||
"""Report it was impossible to resume download."""
|
"""Report it was impossible to resume download."""
|
||||||
|
@ -35,7 +35,6 @@
|
|||||||
remove_encrypted_media,
|
remove_encrypted_media,
|
||||||
)
|
)
|
||||||
from ..utils import (
|
from ..utils import (
|
||||||
NO_DEFAULT,
|
|
||||||
age_restricted,
|
age_restricted,
|
||||||
base_url,
|
base_url,
|
||||||
bug_reports_message,
|
bug_reports_message,
|
||||||
@ -45,10 +44,11 @@
|
|||||||
determine_protocol,
|
determine_protocol,
|
||||||
dict_get,
|
dict_get,
|
||||||
error_to_compat_str,
|
error_to_compat_str,
|
||||||
ExtractorError,
|
|
||||||
extract_attributes,
|
extract_attributes,
|
||||||
|
ExtractorError,
|
||||||
fix_xml_ampersands,
|
fix_xml_ampersands,
|
||||||
float_or_none,
|
float_or_none,
|
||||||
|
format_field,
|
||||||
GeoRestrictedError,
|
GeoRestrictedError,
|
||||||
GeoUtils,
|
GeoUtils,
|
||||||
int_or_none,
|
int_or_none,
|
||||||
@ -56,6 +56,7 @@
|
|||||||
JSON_LD_RE,
|
JSON_LD_RE,
|
||||||
mimetype2ext,
|
mimetype2ext,
|
||||||
network_exceptions,
|
network_exceptions,
|
||||||
|
NO_DEFAULT,
|
||||||
orderedSet,
|
orderedSet,
|
||||||
parse_bitrate,
|
parse_bitrate,
|
||||||
parse_codecs,
|
parse_codecs,
|
||||||
@ -64,8 +65,8 @@
|
|||||||
parse_m3u8_attributes,
|
parse_m3u8_attributes,
|
||||||
parse_resolution,
|
parse_resolution,
|
||||||
RegexNotFoundError,
|
RegexNotFoundError,
|
||||||
sanitized_Request,
|
|
||||||
sanitize_filename,
|
sanitize_filename,
|
||||||
|
sanitized_Request,
|
||||||
str_or_none,
|
str_or_none,
|
||||||
str_to_int,
|
str_to_int,
|
||||||
strip_or_none,
|
strip_or_none,
|
||||||
@ -75,9 +76,9 @@
|
|||||||
unified_timestamp,
|
unified_timestamp,
|
||||||
update_Request,
|
update_Request,
|
||||||
update_url_query,
|
update_url_query,
|
||||||
urljoin,
|
|
||||||
url_basename,
|
url_basename,
|
||||||
url_or_none,
|
url_or_none,
|
||||||
|
urljoin,
|
||||||
variadic,
|
variadic,
|
||||||
xpath_element,
|
xpath_element,
|
||||||
xpath_text,
|
xpath_text,
|
||||||
@ -1002,7 +1003,7 @@ def _download_socket_json(
|
|||||||
return res if res is False else res[0]
|
return res if res is False else res[0]
|
||||||
|
|
||||||
def report_warning(self, msg, video_id=None, *args, only_once=False, **kwargs):
|
def report_warning(self, msg, video_id=None, *args, only_once=False, **kwargs):
|
||||||
idstr = '' if video_id is None else '%s: ' % video_id
|
idstr = format_field(video_id, template='%s: ')
|
||||||
msg = f'[{self.IE_NAME}] {idstr}{msg}'
|
msg = f'[{self.IE_NAME}] {idstr}{msg}'
|
||||||
if only_once:
|
if only_once:
|
||||||
if f'WARNING: {msg}' in self._printed_messages:
|
if f'WARNING: {msg}' in self._printed_messages:
|
||||||
|
@ -13,16 +13,16 @@
|
|||||||
compat_urllib_parse_urlparse,
|
compat_urllib_parse_urlparse,
|
||||||
)
|
)
|
||||||
from ..utils import (
|
from ..utils import (
|
||||||
dict_get,
|
|
||||||
ExtractorError,
|
ExtractorError,
|
||||||
int_or_none,
|
dict_get,
|
||||||
float_or_none,
|
float_or_none,
|
||||||
|
int_or_none,
|
||||||
OnDemandPagedList,
|
OnDemandPagedList,
|
||||||
parse_duration,
|
parse_duration,
|
||||||
parse_iso8601,
|
parse_iso8601,
|
||||||
PostProcessingError,
|
PostProcessingError,
|
||||||
str_or_none,
|
|
||||||
remove_start,
|
remove_start,
|
||||||
|
str_or_none,
|
||||||
try_get,
|
try_get,
|
||||||
unified_timestamp,
|
unified_timestamp,
|
||||||
urlencode_postdata,
|
urlencode_postdata,
|
||||||
|
@ -6161,8 +6161,11 @@ def to_high_limit_path(path):
|
|||||||
return path
|
return path
|
||||||
|
|
||||||
|
|
||||||
def format_field(obj, field, template='%s', ignore=(None, ''), default='', func=None):
|
def format_field(obj, field=None, template='%s', ignore=(None, ''), default='', func=None):
|
||||||
val = obj.get(field, default)
|
if field is None:
|
||||||
|
val = obj if obj is not None else default
|
||||||
|
else:
|
||||||
|
val = obj.get(field, default)
|
||||||
if func and val not in ignore:
|
if func and val not in ignore:
|
||||||
val = func(val)
|
val = func(val)
|
||||||
return template % val if val not in ignore else default
|
return template % val if val not in ignore else default
|
||||||
|
Loading…
Reference in New Issue
Block a user