mirrors/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-10-30 14:15:13 +00:00
Code Issues Projects Releases Wiki Activity

[utils] Move FileDownloader.parse_bytes into utils

Browse Source
This commit is contained in:
pukkandan
2022-11-17 08:40:34 +05:30
parent 4de88a6a36
commit 64c464a144
3 changed files with 25 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
yt_dlp/utils.py
View File

@@ -2289,15 +2289,24 @@ def format_bytes(bytes):
return format_decimal_suffix(bytes, '%.2f%sB', factor=1024) or 'N/A'
def lookup_unit_table(unit_table, s):
def lookup_unit_table(unit_table, s, strict=False):
num_re = NUMBER_RE if strict else NUMBER_RE.replace(R'\.', '[,.]')
units_re = '|'.join(re.escape(u) for u in unit_table)
m = re.match(
r'(?P<num>[0-9]+(?:[,.][0-9]*)?)\s*(?P<unit>%s)\b' % units_re, s)
m = (re.fullmatch if strict else re.match)(
rf'(?P<num>{num_re})\s*(?P<unit>{units_re})\b', s)
if not m:
return None
num_str = m.group('num').replace(',', '.')
num = float(m.group('num').replace(',', '.'))
mult = unit_table[m.group('unit')]
return int(float(num_str) * mult)
return round(num * mult)
def parse_bytes(s):
"""Parse a string indicating a byte quantity into an integer"""
return lookup_unit_table(
{u: 1024**i for i, u in enumerate(['', *'KMGTPEZY'])},
s.upper(), strict=True)
def parse_filesize(s):