progress

lib/pt_trading/results.py

@@ -170,7 +170,24 @@ def store_config_in_database(
 
         traceback.print_exc()
 
+def convert_timestamp(timestamp: Any) -> Optional[datetime]:
+    """Convert pandas Timestamp to Python datetime object for SQLite compatibility."""
+    if timestamp is None:
+        return None
+    if isinstance(timestamp, pd.Timestamp):
+        return timestamp.to_pydatetime()
+    elif isinstance(timestamp, datetime):
+        return timestamp
+    elif isinstance(timestamp, date):
+        return datetime.combine(timestamp, datetime.min.time())
+    elif isinstance(timestamp, str):
+        return datetime.strptime(timestamp, "%Y-%m-%d %H:%M:%S")
+    elif isinstance(timestamp, int):
+        return datetime.fromtimestamp(timestamp)
+    else:
+        raise ValueError(f"Unsupported timestamp type: {type(timestamp)}")
 
+ 
 def store_results_in_database(
     db_path: str, datafile: str, bt_result: "BacktestResult"
 ) -> None:
@@ -180,14 +197,6 @@ def store_results_in_database(
     if db_path.upper() == "NONE":
         return
 
-    def convert_timestamp(timestamp: Any) -> Optional[datetime]:
-        """Convert pandas Timestamp to Python datetime object for SQLite compatibility."""
-        if timestamp is None:
-            return None
-        if hasattr(timestamp, "to_pydatetime"):
-            return timestamp.to_pydatetime()
-        return timestamp
-
     try:
         # Extract date from datafile name (assuming format like 20250528.mktdata.ohlcv.db)
         filename = os.path.basename(datafile)
@@ -489,7 +498,11 @@ class BacktestResult:
             price = row.price
             disequilibrium = getattr(row, "disequilibrium", None)
             scaled_disequilibrium = getattr(row, "scaled_disequilibrium", None)
-            timestamp = getattr(row, "time", None)
+            if hasattr(row, "time"):
+                timestamp = getattr(row, "time")
+            else:
+                timestamp = convert_timestamp(row.Index)
+
             self.add_trade(
                 pair_nm=str(row.pair),
                 action=str(action),