From 24653b875313ed0a8eaa434c31f6eefde8135c59 Mon Sep 17 00:00:00 2001
From: KnugiHK <24708955+KnugiHK@users.noreply.github.com>
Date: Sun, 1 Jun 2025 11:53:45 +0800
Subject: [PATCH] Fixed integer input for --size not being casted to int #156

---
 Whatsapp_Chat_Exporter/__main__.py | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/Whatsapp_Chat_Exporter/__main__.py b/Whatsapp_Chat_Exporter/__main__.py
index ff44531..1cdb5c4 100644
--- a/Whatsapp_Chat_Exporter/__main__.py
+++ b/Whatsapp_Chat_Exporter/__main__.py
@@ -332,13 +332,18 @@ def validate_args(parser: ArgumentParser, args) -> None:
         parser.error(
             "When --enrich-from-vcards is provided, you must also set --default-country-code")
 
-    # Size validation
-    if args.size is not None and not isinstance(args.size, int) and not args.size.isnumeric():
-        try:
-            args.size = readable_to_bytes(args.size)
-        except ValueError:
-            parser.error(
-                "The value for --split must be ended in pure bytes or with a proper unit (e.g., 1048576 or 1MB)")
+    # Size validation and conversion
+    if args.size is not None:
+        stripped_size = args.size.strip()
+        if stripped_size.isnumeric():
+            args.size = int(stripped_size)
+        else:
+            try:
+                args.size = readable_to_bytes(stripped_size)
+            except ValueError:
+                parser.error(
+                    "The value for --split must be pure bytes or use a proper unit (e.g., 1048576 or 1MB)"
+                )
 
     # Date filter validation and processing
     if args.filter_date is not None: