diff options
| -rw-r--r-- | src/utils/__init__.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/src/utils/__init__.py b/src/utils/__init__.py index a700dd95..0cb131ff 100644 --- a/src/utils/__init__.py +++ b/src/utils/__init__.py @@ -85,6 +85,25 @@ def to_pretty_time(total_seconds, minimum_unit=UNIT_SECOND, max_units=6): units += 1 return out +def parse_number(s): + if s.isdigit(): + return s + + unit = s[-1].lower() + number = s[:-1] + if not number.isdigit(): + raise ValueError("Invalid format '%s' passed to parse_number") + + if unit == "k": + number *= 1_000 + elif unit == "m": + number *= 1_000_000 + elif unit == "b": + number *= 1_000_000_000 + else: + raise ValueError("Unknown unit '%s' given to parse_number") + return str(number) + IS_TRUE = ["true", "yes", "on", "y"] IS_FALSE = ["false", "no", "off", "n"] def bool_or_none(s): |