Add timestamp to GMOCoinDataStore (#88)

pybotters · Nov 10, 2021 · 1dca568 · 1dca568
1 parent 192735a
commit 1dca568
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pybotters/models/gmocoin.py b/pybotters/models/gmocoin.py
@@ -289,6 +289,9 @@ def _onmessage(self, mes: Ticker) -> None:
 class OrderBookStore(DataStore):
     _KEYS = ["symbol", "side", "price"]
 
+    def _init(self) -> None:
+        self.timestamp: Optional[datetime] = None
+
     def sorted(self, query: Optional[Item] = None) -> dict[OrderSide, list[OrderLevel]]:
         if query is None:
             query = {}
@@ -308,6 +311,7 @@ def _onmessage(self, mes: OrderBook) -> None:
         result = self.find({"symbol": mes["symbol"]})
         self._delete(result)
         self._insert(cast(list[Item], data))
+        self.timestamp = mes["timestamp"]
 
 
 class TradeStore(DataStore):