1
0
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:
pukkandan 2021-08-07 16:50:46 +05:30
parent c681cb5d93
commit b868936cd6
7 changed files with 41 additions and 39 deletions

View File

@ -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"

View File

@ -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

View 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):

View File

@ -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."""

View File

@ -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:

View File

@ -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,

View File

@ -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