frank/vat tweaks (#191)

* vat tweaks

* fix black

pyproject.toml

@@ -85,7 +85,7 @@ jito-searcher-client = "0.1.4"
 [tool.poetry.dev-dependencies]
 pytest = "^7.2.0"
 flake8 = "6.0.0"
-black = "^23.3.0"
+black = "24.4.2"
 pytest-asyncio = "^0.21.0"
 mkdocs = "^1.3.0"
 mkdocstrings = "^0.17.0"

src/driftpy/constants/perp_markets.py

@@ -286,7 +286,7 @@ class PerpMarketConfig:
         symbol="RNDR-PERP",
         base_asset_symbol="RNDR",
         market_index=12,
-        oracle=Pubkey.from_string("F3mPHRtJqqq57JPDBmUwUVhpyPLmjE5dAzDfpVgpFkug"),
+        oracle=Pubkey.from_string("8TQztfGcNjHGRusX4ejQQtPZs3Ypczt9jWF6pkgQMqUX"),
         oracle_source=OracleSource.PythPull(),
     ),
     PerpMarketConfig(

src/driftpy/constants/spot_markets.py

@@ -126,7 +126,7 @@ class SpotMarketConfig:
     SpotMarketConfig(
         symbol="RNDR",
         market_index=12,
-        oracle=Pubkey.from_string("F3mPHRtJqqq57JPDBmUwUVhpyPLmjE5dAzDfpVgpFkug"),
+        oracle=Pubkey.from_string("8TQztfGcNjHGRusX4ejQQtPZs3Ypczt9jWF6pkgQMqUX"),
         oracle_source=OracleSource.PythPull(),
         mint=Pubkey.from_string("rndrizKT3MK1iimdxRdWabcF7Zg7AR5T4nud4EkHBof"),
     ),

src/driftpy/pickle/vat.py

@@ -1,5 +1,6 @@
 import pickle
 import os
+from dataclasses import dataclass
 from typing import Optional
 from driftpy.drift_client import DriftClient
 from driftpy.market_map.market_map import MarketMap
@@ -14,7 +15,7 @@ def __init__(
         self,
         drift_client: DriftClient,
         users: UserMap,
-        user_stats: UserStatsMap,
+        user_stats: Optional[UserStatsMap],
         spot_markets: MarketMap,
         perp_markets: MarketMap,
     ):
@@ -31,8 +32,9 @@ async def pickle(self):
         await self.users.sync()
         self.users.dump()
 
-        await self.user_stats.sync()
-        self.user_stats.dump()
+        if self.user_stats is not None:
+            await self.user_stats.sync()
+            self.user_stats.dump()
 
         await self.spot_markets.dump()
         await self.perp_markets.dump()
@@ -54,7 +56,8 @@ async def unpickle(
         self.perp_markets.clear()
 
         await self.users.load(users_filename)
-        await self.user_stats.load(user_stats_filename)
+        if self.user_stats is not None:
+            await self.user_stats.load(user_stats_filename)
         await self.spot_markets.load(spot_markets_filename)
         await self.perp_markets.load(perp_markets_filename)
 

src/driftpy/user_map/user_map.py

@@ -244,11 +244,11 @@ async def load(self, filename: Optional[str] = None):
                 data = decode_user(decompress(user.data))
                 await self.add_pubkey(user.pubkey, DataAndSlot(slot, data))
 
-    def dump(self):
+    def dump(self, filename: Optional[str] = None):
         users = []
         for pubkey, user in self.raw.items():
             users.append(PickledData(pubkey=pubkey, data=compress(user)))
         self.last_dumped_slot = self.get_slot()
-        filename = f"usermap_{self.last_dumped_slot}.pkl"
+        filename = filename or f"usermap_{self.last_dumped_slot}.pkl"
         with open(filename, "wb") as f:
             pickle.dump(users, f, pickle.HIGHEST_PROTOCOL)