refactor: break parse_iso into parse_iso_date and parse_iso_time

Author: launeh

typed_python/lib/datetime/date_parser.py

@@ -184,6 +184,28 @@ def parse_tz_offset(tokens: ListOf(str)) -> int:
         else:
             raise ValueError('Invalid tz offset: ')
 
+    @Entrypoint
+    @staticmethod
+    def parse_iso_time(time_tokens: ListOf(str)):
+        hour, minute, second = 0, 0, 0.0
+
+        if len(time_tokens) == 1:
+            if len(time_tokens[0]) == 6:
+                hour, minute, second = int(time_tokens[0][:2]), int(time_tokens[0][2:4]), float(time_tokens[0][4:6])
+            elif len(time_tokens[0]) == 4:
+                hour, minute, second = int(time_tokens[0][:2]), int(time_tokens[0][2:4]), 0.0
+            elif len(time_tokens[0]) == 2:
+                hour, minute, second = int(time_tokens[0][:2]), 0, 0.0
+        elif len(time_tokens) == 2:
+            hour, minute, second = int(time_tokens[0]), int(time_tokens[1]), 0.0
+        elif len(time_tokens) == 3:
+            hour, minute, second = int(time_tokens[0]), int(time_tokens[1]), float(time_tokens[2])
+
+        if not is_time(hour, minute, second):
+            raise ValueError('Invalid time: ', time_tokens)
+
+        return time_to_seconds(hour, minute, second)
+
     @Entrypoint
     @staticmethod
     def parse_iso(date_str: str) -> float:
@@ -224,6 +246,11 @@ def parse_iso(date_str: str) -> float:
         if not is_date(year, month, day):
             raise ValueError('Invalid date_tokens: ', date_tokens)
 
+        dt = date_to_seconds(year, month, day)
+
+        if cursor >= len(tokens):
+            return dt
+
         # Process time segement
         time_tokens = ListOf(str)([])
         while cursor < len(tokens):
@@ -236,32 +263,15 @@ def parse_iso(date_str: str) -> float:
                 time_tokens.append(tokens[cursor])
                 cursor += 1
 
-        hour, minute, second = 0, 0, 0.0
-
-        if len(time_tokens) == 1:
-            if len(time_tokens[0]) == 6:
-                hour, minute, second = int(time_tokens[0][:2]), int(time_tokens[0][2:4]), float(time_tokens[0][4:6])
-            elif len(time_tokens[0]) == 4:
-                hour, minute, second = int(time_tokens[0][:2]), int(time_tokens[0][2:4]), 0.0
-            elif len(time_tokens[0]) == 2:
-                hour, minute, second = int(time_tokens[0][:2]), 0, 0.0
-        elif len(time_tokens) == 2:
-            hour, minute, second = int(time_tokens[0]), int(time_tokens[1]), 0.0
-        elif len(time_tokens) == 3:
-            hour, minute, second = int(time_tokens[0]), int(time_tokens[1]), float(time_tokens[2])
-
-        if not is_time(hour, minute, second):
-            raise ValueError('Invalid time: ', time_tokens)
+        dt += DateParser.parse_iso_time(time_tokens)
+        if cursor >= len(tokens):
+            return dt
 
         tz_tokens = ListOf(str)([])
         while cursor < len(tokens):
             tz_tokens.append(tokens[cursor])
             cursor += 1
 
-        dt = date_to_seconds(year, month, day) + time_to_seconds(hour, minute, second)
-        if len(tz_tokens) == 0:
-            return dt
-
         if len(tz_tokens) == 1:
             tz_offset = tz_abbr_to_utc_offset(tz_tokens[0], dt)
         elif tz_tokens[0] == PLUS or tz_tokens[0] == DASH: