calvin.erfmann/WhatsApp-Chat-Exporter
1
0
Fork 0
mirror of https://github.com/KnugiHK/WhatsApp-Chat-Exporter.git synced 2026-04-25 15:31:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enforce a tighter check on the input of size_str

Browse Source
This commit is contained in:
KnugiHK
2025-06-01 11:54:24 +08:00
parent 24653b8753
commit 706466f63b
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Whatsapp_Chat_Exporter/utility.py
View File

@@ -118,6 +118,9 @@ def readable_to_bytes(size_str: str) -> int:
'YB': 1024**8 'YB': 1024**8
} }
size_str = size_str.upper().strip() size_str = size_str.upper().strip()
match = re.fullmatch(r'^(\d+(\.\d+)?)\s*([KMGTPEZY]?B)?$', size_str)
if not match:
raise ValueError("Invalid size format. Expected format like '10MB', '1024GB', or '512'.")
unit = ''.join(filter(str.isalpha, size_str)).strip() unit = ''.join(filter(str.isalpha, size_str)).strip()
number = ''.join(c for c in size_str if c.isdigit() or c == '.').strip() number = ''.join(c for c in size_str if c.isdigit() or c == '.').strip()
if unit not in SIZE_UNITS: if unit not in SIZE_UNITS: