From 89c825c6ed4f9215a5285ebe2bc5ec9e2087175c Mon Sep 17 00:00:00 2001
From: Dinesh Dutt <dd.ps4u@gmail.com>
Date: Mon, 22 Jan 2024 08:24:25 -0800
Subject: [PATCH] utils.py: add support for ios/xe time parsing

Signed-off-by: Dinesh Dutt <dd.ps4u@gmail.com>
---
 suzieq/shared/utils.py | 40 ++++++++++++++++++++++++++++++++++------
 1 file changed, 34 insertions(+), 6 deletions(-)

diff --git a/suzieq/shared/utils.py b/suzieq/shared/utils.py
index 2589362b3e..46ef30aeac 100644
--- a/suzieq/shared/utils.py
+++ b/suzieq/shared/utils.py
@@ -15,7 +15,7 @@
 from logging.handlers import RotatingFileHandler
 from os import getenv
 from time import time
-from typing import Any, Dict, List, Tuple
+from typing import Any, Dict, List, Optional, Tuple, Union
 
 import pandas as pd
 import psutil
@@ -386,9 +386,13 @@ def calc_avg(oldval, newval):
 
 
 def parse_relative_timestamp(
-        uptime: str, relative_to: int = None, ms=False) -> int:
-    """Get a relative time (i.e. with format 10 weeks, 4 days, 3 hours 11 mins)
-    and convert it into a timestamp.
+        uptime: str, relative_to: Optional[int] = None,
+        ms=False) -> Optional[int]:
+    """Convert a string of relative time into a timestamp
+
+    This takes dateparser parsable strings such as
+    '10 weeks, 4 days, 3 hours 11 mins' or '1w4d' or '00h05m040s'
+    and converts it into a timestamp.
 
     Args:
         uptime (str): _description_
@@ -477,8 +481,32 @@ def get_timestamp_from_cisco_time(in_data: str, timestamp: int) -> int:
     return int((datetime.fromtimestamp(timestamp)-delta).timestamp()*1000)
 
 
-def get_timestamp_from_junos_time(in_data: Tuple[Dict, str],
-                                  relative_to: int = None,
+def get_timestamp_from_iosxe_time(in_data: str, timestamp: int) -> int:
+    """Get timestamp in ms from the Cisco IOS/XE-specific timestamp string
+    Examples of Cisco timestamp str are 1w4d, 1d10h, 01:20:35, 00:01:56.
+
+    Args:
+        in_data (str): The IOSXE uptime string
+        timestamp (int): the base unix timestamp IS SECONDS from which
+            subtracting the uptime.
+
+    Returns:
+        int: a unix timestamp in milliseconds
+    """
+    if not in_data:
+        return 0
+
+    if ':' in in_data:
+        # There's time and so we need to convert it into h/m/s
+        in_data = (in_data.replace(':', 'h', 1)
+                          .replace(':', 'm', 1)
+                   ) + 's'
+
+    return parse_relative_timestamp(in_data, timestamp, False) or 0
+
+
+def get_timestamp_from_junos_time(in_data: Union[str, Dict[str, str]],
+                                  relative_to: Optional[int] = None,
                                   ms=True) -> int:
     """Get timestamp in ms from the Junos-specific timestamp string
     The expected input looks like: "attributes" : {"junos:seconds" : "0"}.