From 612f2be5d3924540158dfbe5f25d841f04cff8c6 Mon Sep 17 00:00:00 2001 From: pukkandan Date: Mon, 20 Jun 2022 11:55:54 +0530 Subject: Bugfix for 7b2c3f47c6b586a208655fcfc716bba3f8619d1e --- yt_dlp/utils.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'yt_dlp/utils.py') 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 -- cgit v1.2.3