Merge pull request #1 from Edanflame/main

添加存储成功返回
vnpy · Sep 16, 2021 · 2cd89dd · 2cd89dd
2 parents d2728bc + 69a74ca
commit 2cd89dd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/vnpy_sqlite/sqlite_database.py b/vnpy_sqlite/sqlite_database.py
@@ -186,6 +186,8 @@ def save_bar_data(self, bars: List[BarData]) -> bool:
 
         overview.save()
 
+        return True
+
     def save_tick_data(self, ticks: List[TickData]) -> bool:
         """保存TICK数据"""
         # 将TickData数据转换为字典，并调整时区
@@ -205,6 +207,8 @@ def save_tick_data(self, ticks: List[TickData]) -> bool:
             for c in chunked(data, 10):
                 DbTickData.insert_many(c).on_conflict_replace().execute()
 
+        return True
+
     def load_bar_data(
         self,
         symbol: str,
@@ -229,7 +233,7 @@ def load_bar_data(
             bar = BarData(
                 symbol=db_bar.symbol,
                 exchange=Exchange(db_bar.exchange),
-                datetime=db_bar.datetime.astimezone(DB_TZ),
+                datetime=datetime.fromtimestamp(db_bar.datetime.timestamp(), DB_TZ),
                 interval=Interval(db_bar.interval),
                 volume=db_bar.volume,
                 turnover=db_bar.turnover,
@@ -266,7 +270,7 @@ def load_tick_data(
             tick = TickData(
                 symbol=db_tick.symbol,
                 exchange=Exchange(db_tick.exchange),
-                datetime=db_tick.datetime.astimezone(DB_TZ),
+                datetime=datetime.fromtimestamp(db_tick.datetime.timestamp(), DB_TZ),
                 name=db_tick.name,
                 volume=db_tick.volume,
                 turnover=db_tick.turnover,