add helper method for ms timestamp

Author: ceciliastevens

src/_incydr_sdk/queries/utils.py

@@ -40,6 +40,11 @@ def parse_ts_to_posix_ts(timestamp: Union[str, datetime]):
     dt = timestamp if isinstance(timestamp, datetime) else parse_str_to_dt(timestamp)
     return dt.timestamp()
 
+def parse_ts_to_ms_ts(timestamp: Union[str, datetime]):
+    """
+    Parse epoch ms timestamp from DATE/DATETIME str or datetime obj.
+    """
+    return parse_ts_to_posix_ts(timestamp) * 1000
 
 def parse_str_to_dt(timestamp: str):
     try:

src/_incydr_sdk/sessions/client.py

@@ -9,6 +9,7 @@
 from _incydr_sdk.enums.sessions import SessionStates
 from _incydr_sdk.enums.sessions import SortKeys
 from _incydr_sdk.queries.utils import parse_ts_to_posix_ts
+from _incydr_sdk.queries.utils import parse_ts_to_ms_ts
 from _incydr_sdk.sessions.models.models import SessionsChangeStateRequest
 from _incydr_sdk.sessions.models.models import SessionsCriteriaRequest
 from _incydr_sdk.sessions.models.models import SessionsQueryRequest
@@ -76,9 +77,9 @@ def get_page(
 
         # Parse timestamps
         if start_time and not isinstance(start_time, (int, float)):
-            start_time = parse_ts_to_posix_ts(start_time) * 1000
+            start_time = parse_ts_to_ms_ts(start_time)
         if end_time and not isinstance(end_time, (int, float)):
-            end_time = parse_ts_to_posix_ts(end_time) * 1000
+            end_time = parse_ts_to_ms_ts(end_time)
 
         if states and not isinstance(states, List):
             states = [states]
@@ -253,9 +254,9 @@ def update_state_by_criteria(
 
         # Parse timestamps
         if start_time and not isinstance(start_time, (int, float)):
-            start_time = parse_ts_to_posix_ts(start_time) * 1000
+            start_time = parse_ts_to_ms_ts(start_time)
         if end_time and not isinstance(end_time, (int, float)):
-            end_time = parse_ts_to_posix_ts(end_time) * 1000
+            end_time = parse_ts_to_ms_ts(end_time)
 
         if states and not isinstance(states, List):
             states = [states]