FIX: Error with time.py (#93)

* FIX: Error with time.py

* FIX: Fixed error with time.py, added tests

---------

Co-authored-by: dborodin836 <dborodin836@gmail.com>

Author: teufortressIndustries

modules/utils/time.py

@@ -42,11 +42,14 @@ def get_minutes_from_str(time_str: str) -> int:
     try:
         struct_time = time.strptime(time_str, "%H:%M:%S")
         tm = struct_time.tm_hour * 60 + struct_time.tm_min
+        return tm
     except ValueError:
+        pass
+
+    try:
         struct_time = time.strptime(time_str, "%M:%S")
         tm = struct_time.tm_min
-    except Exception as e:
-        main_logger.warning(f"Unhandled error while parsing time happened. ({e})")
-        tm = 0
-
-    return tm
+        return tm
+    except ValueError as e:
+        main_logger.warning(f"Invalid time format. ({e})")
+        return 0

tests/test_tf_stats.py

@@ -1,4 +1,4 @@
-from modules.utils.time import get_date
+from modules.utils.time import get_date, get_minutes_from_str
 
 
 def test_epoch_time():
@@ -8,3 +8,18 @@ def test_epoch_time():
     assert get_date(1497448943, 1685541046) == "5 years 11 months 21 days"
     assert get_date(1331474543, 1685541046) == "11 years 2 months 21 days"
     assert get_date(1685541046, 1685541046) == "0 years 0 months 0 days"
+
+
+def test_get_minutes_from_str():
+    # Minutes
+    assert get_minutes_from_str("12:20") == 12
+    assert get_minutes_from_str("12:60") == 12
+
+    # Hours
+    assert get_minutes_from_str("1:00:00") == 60
+    assert get_minutes_from_str("1:25:00") == 85
+
+    # Fails
+    assert get_minutes_from_str("TEST") == 0
+    assert get_minutes_from_str("2:") == 0
+    assert get_minutes_from_str("12:62") == 0