From 001bcdbd38fb8c4f56065875ea3c99a32367d7db Mon Sep 17 00:00:00 2001 From: Xevion Date: Thu, 11 May 2023 02:32:47 -0500 Subject: [PATCH] Allow lowercase in convert_to_bytes, return 0 for invalid suffixes --- phototag/helpers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/phototag/helpers.py b/phototag/helpers.py index ec926f4..346c84d 100644 --- a/phototag/helpers.py +++ b/phototag/helpers.py @@ -80,7 +80,7 @@ def convert_to_bytes(size_string: str) -> int: :return: The number of bytes the given string is equivalent to. """ match = re.match(r"(\d+)\s*(\w{1,2})", size_string) - return int(match.group(1)) * byte_magnitudes[match.group(2)] + return int(match.group(1)) * byte_magnitudes.get(match.group(2).upper(), 0) def select_files(files: List[str], regex: Optional[str], glob_pattern: Optional[str]) -> List[str]: