[StatTrack] Be more flexible with date parsing

Vexed01 · Nov 5, 2023 · f4545ac · f4545ac
1 parent 9e6a4cc
commit f4545ac
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/stattrack/driver.py b/stattrack/driver.py
@@ -36,7 +36,7 @@ async def get_last_index(self) -> pd.Timestamp:
                 index = await cursor.fetchone()
         if index is None:
             return pd.Timestamp(0)
-        return pd.to_datetime(index[0])
+        return pd.to_datetime(index[0], format="ISO8601")
 
     async def read_all(self) -> pd.DataFrame:
         """Create a Pandas DataFrame from the whole table.
@@ -53,7 +53,7 @@ async def read_all(self) -> pd.DataFrame:
                 data = await cursor.fetchall()
         df = pd.DataFrame(data, columns=columns)
         df.set_index("index", inplace=True)
-        df.index = pd.to_datetime(df.index)
+        df.index = pd.to_datetime(df.index, format="ISO8601")
         return df
 
     async def read_partial(
@@ -86,7 +86,7 @@ async def read_partial(
                 data = await cursor.fetchall()
         df = pd.DataFrame(data, columns=["index"] + list(metrics))
         df.set_index("index", inplace=True)
-        df.index = pd.to_datetime(df.index)
+        df.index = pd.to_datetime(df.index, format="ISO8601")
         return df
 
     async def write(self, df: pd.DataFrame) -> None: