aboutsummaryrefslogtreecommitdiffstats
path: root/yt_dlp
diff options
context:
space:
mode:
authorpukkandan <pukkandan.ytdlp@gmail.com>2021-04-15 18:01:16 +0530
committerpukkandan <pukkandan.ytdlp@gmail.com>2021-04-22 04:19:33 +0530
commita439a3a45ca884956cacc104680e0a32aa3faba5 (patch)
tree19f9b1fbc36d6d39c2d6819fdbc5a0d3d4f2af0c /yt_dlp
parent26e2805c3f11066c79ed29cf78511eae7fdc7a7b (diff)
downloadhypervideo-pre-a439a3a45ca884956cacc104680e0a32aa3faba5.tar.lz
hypervideo-pre-a439a3a45ca884956cacc104680e0a32aa3faba5.tar.xz
hypervideo-pre-a439a3a45ca884956cacc104680e0a32aa3faba5.zip
Improve output template (see desc)
* Objects can be traversed like `%(field.key1.key2)s` * A number can be added to the field as `%(field+n)s` * Deprecates `--autonumber-start`
Diffstat (limited to 'yt_dlp')
-rw-r--r--yt_dlp/YoutubeDL.py59
-rw-r--r--yt_dlp/options.py2
-rw-r--r--yt_dlp/utils.py25
3 files changed, 54 insertions, 32 deletions
diff --git a/yt_dlp/YoutubeDL.py b/yt_dlp/YoutubeDL.py
index a3d7968ff..5aea1b807 100644
--- a/yt_dlp/YoutubeDL.py
+++ b/yt_dlp/YoutubeDL.py
@@ -99,6 +99,7 @@ from .utils import (
strftime_or_none,
subtitles_filename,
to_high_limit_path,
+ traverse_dict,
UnavailableVideoError,
url_basename,
version_tuple,
@@ -796,6 +797,7 @@ class YoutubeDL(object):
def prepare_outtmpl(self, outtmpl, info_dict, sanitize=None):
""" Make the template and info_dict suitable for substitution (outtmpl % info_dict)"""
template_dict = dict(info_dict)
+ na = self.params.get('outtmpl_na_placeholder', 'NA')
# duration_string
template_dict['duration_string'] = ( # %(duration>%H-%M-%S)s is wrong if duration > 24hrs
@@ -821,18 +823,10 @@ class YoutubeDL(object):
elif template_dict.get('width'):
template_dict['resolution'] = '%dx?' % template_dict['width']
- if sanitize is None:
- sanitize = lambda k, v: v
- template_dict = dict((k, v if isinstance(v, compat_numeric_types) else sanitize(k, v))
- for k, v in template_dict.items()
- if v is not None and not isinstance(v, (list, tuple, dict)))
- na = self.params.get('outtmpl_na_placeholder', 'NA')
- template_dict = collections.defaultdict(lambda: na, template_dict)
-
# For fields playlist_index and autonumber convert all occurrences
# of %(field)s to %(field)0Nd for backward compatibility
field_size_compat_map = {
- 'playlist_index': len(str(template_dict['n_entries'])),
+ 'playlist_index': len(str(template_dict.get('n_entries', na))),
'autonumber': autonumber_size,
}
FIELD_SIZE_COMPAT_RE = r'(?<!%)%\((?P<field>autonumber|playlist_index)\)s'
@@ -844,32 +838,51 @@ class YoutubeDL(object):
outtmpl)
numeric_fields = list(self._NUMERIC_FIELDS)
+ if sanitize is None:
+ sanitize = lambda k, v: v
- # Format date
- FORMAT_DATE_RE = FORMAT_RE.format(r'(?P<key>(?P<field>\w+)>(?P<format>.+?))')
- for mobj in re.finditer(FORMAT_DATE_RE, outtmpl):
- conv_type, field, frmt, key = mobj.group('type', 'field', 'format', 'key')
- if key in template_dict:
- continue
- value = strftime_or_none(template_dict.get(field), frmt, na)
- if conv_type in 'crs': # string
- value = sanitize(field, value)
- else: # number
- numeric_fields.append(key)
- value = float_or_none(value, default=None)
+ # Internal Formatting = name.key1.key2+number>strf
+ INTERNAL_FORMAT_RE = FORMAT_RE.format(
+ r'''(?P<final_key>
+ (?P<fields>\w+(?:\.[-\w]+)*)
+ (?:\+(?P<add>-?\d+(?:\.\d+)?))?
+ (?:>(?P<strf_format>.+?))?
+ )''')
+ for mobj in re.finditer(INTERNAL_FORMAT_RE, outtmpl):
+ mobj = mobj.groupdict()
+ # Object traversal
+ fields = mobj['fields'].split('.')
+ final_key = mobj['final_key']
+ value = traverse_dict(template_dict, fields)
+ # Offset the value
+ if mobj['add']:
+ value = float_or_none(value)
+ if value is not None:
+ value = value + float(mobj['add'])
+ # Datetime formatting
+ if mobj['strf_format']:
+ value = strftime_or_none(value, mobj['strf_format'])
+ if mobj['type'] in 'crs' and value is not None: # string
+ value = sanitize('%{}'.format(mobj['type']) % fields[-1], value)
+ else: # numeric
+ numeric_fields.append(final_key)
+ value = float_or_none(value)
if value is not None:
- template_dict[key] = value
+ template_dict[final_key] = value
# Missing numeric fields used together with integer presentation types
# in format specification will break the argument substitution since
# string NA placeholder is returned for missing fields. We will patch
# output template for missing fields to meet string presentation type.
for numeric_field in numeric_fields:
- if numeric_field not in template_dict:
+ if template_dict.get(numeric_field) is None:
outtmpl = re.sub(
FORMAT_RE.format(re.escape(numeric_field)),
r'%({0})s'.format(numeric_field), outtmpl)
+ template_dict = collections.defaultdict(lambda: na, (
+ (k, v if isinstance(v, compat_numeric_types) else sanitize(k, v))
+ for k, v in template_dict.items() if v is not None))
return outtmpl, template_dict
def _prepare_filename(self, info_dict, tmpl_type='default'):
diff --git a/yt_dlp/options.py b/yt_dlp/options.py
index c21c453ed..60ae6c7fe 100644
--- a/yt_dlp/options.py
+++ b/yt_dlp/options.py
@@ -908,7 +908,7 @@ def parseOpts(overrideArguments=None):
filesystem.add_option(
'--autonumber-start',
dest='autonumber_start', metavar='NUMBER', default=1, type=int,
- help='Specify the start value for %(autonumber)s (default is %default)')
+ help=optparse.SUPPRESS_HELP)
filesystem.add_option(
'--restrict-filenames',
action='store_true', dest='restrictfilenames', default=False,
diff --git a/yt_dlp/utils.py b/yt_dlp/utils.py
index 3e566285f..461f64db0 100644
--- a/yt_dlp/utils.py
+++ b/yt_dlp/utils.py
@@ -6092,11 +6092,20 @@ def load_plugins(name, type, namespace):
def traverse_dict(dictn, keys, casesense=True):
- if not isinstance(dictn, dict):
- return None
- first_key = keys[0]
- if not casesense:
- dictn = {key.lower(): val for key, val in dictn.items()}
- first_key = first_key.lower()
- value = dictn.get(first_key, None)
- return value if len(keys) < 2 else traverse_dict(value, keys[1:], casesense)
+ keys = list(keys)[::-1]
+ while keys:
+ key = keys.pop()
+ if isinstance(dictn, dict):
+ if not casesense:
+ dictn = {k.lower(): v for k, v in dictn.items()}
+ key = key.lower()
+ dictn = dictn.get(key)
+ elif isinstance(dictn, (list, tuple, compat_str)):
+ key, n = int_or_none(key), len(dictn)
+ if key is not None and -n <= key < n:
+ dictn = dictn[key]
+ else:
+ dictn = None
+ else:
+ return None
+ return dictn