diff options
Diffstat (limited to 'yt_dlp/utils.py')
-rw-r--r-- | yt_dlp/utils.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/yt_dlp/utils.py b/yt_dlp/utils.py index b9c579cb6..9c16d6601 100644 --- a/yt_dlp/utils.py +++ b/yt_dlp/utils.py @@ -4748,23 +4748,23 @@ def pkcs1pad(data, length): def _base_n_table(n, table): if not table and not n: raise ValueError('Either table or n must be specified') - elif not table: - table = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'[:n] - elif not n or n == len(table): - return table - raise ValueError(f'base {n} exceeds table length {len(table)}') + table = (table or '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ')[:n] + + if n != len(table): + raise ValueError(f'base {n} exceeds table length {len(table)}') + return table def encode_base_n(num, n=None, table=None): """Convert given int to a base-n string""" - table = _base_n_table(n) + table = _base_n_table(n, table) if not num: return table[0] result, base = '', len(table) while num: result = table[num % base] + result - num = num // result + num = num // base return result |