فهرست منبع

Move initial balance logic to actual dataclass

Kirk Trombley 3 سال پیش
والد
کامیت
220e041a15
1فایلهای تغییر یافته به همراه4 افزوده شده و 4 حذف شده
  1. 4 4
      commands/commands/rollcoin.py

+ 4 - 4
commands/commands/rollcoin.py

@@ -44,7 +44,7 @@ class RollcoinState:
 @initialize_data
 @dataclass
 class RollcoinWallet:
-    balance: float = 0
+    balance: float = 10 # new wallets start with 10 coins
     holdings: float = 0
     cost_basis: float = 0
     nfts: list[str] = field(default_factory=list)
@@ -103,7 +103,7 @@ NFT_MAX_RARITY = "legendary!"
 
 # injection values
 State = Data(RollcoinState).For(Const(GLOBAL_STATE_KEY), treasury=0, mining_puzzle=None, market_state=0)
-SenderWallet = Data(RollcoinWallet).For(Sender, balance=10) # new wallets start with 10 coins
+SenderWallet = Data(RollcoinWallet).For(Sender)
 WalletLookup = Config("rollcoin.wallet_names")
 MarketTransitions = Config("rollcoin.market.transitions")
 MarketMultipliers = Config("rollcoin.market.multipliers")
@@ -174,7 +174,7 @@ async def tip(
     if (target_id := wallet_lookup.get(target_name.lower(), None)) is None:
         RollbotFailure.INVALID_ARGUMENTS.raise_exc(f"Cannot find wallet belonging to {target_name}")
 
-    target_wallet = await wallets.load_or(target_id, balance=10)
+    target_wallet = await wallets.load_or(target_id)
     sender_wallet.balance -= amount
     target_wallet.balance += amount
     await wallets.save(target_id, target_wallet)
@@ -617,7 +617,7 @@ async def giftcha(
     sender_wallet = await get_sender_wallet()
     sender_wallet.balance -= len(pulled)
     await wallet_store.save(sender_id, sender_wallet)
-    target_wallet = await wallet_store.load_or(target_id, balance=10)
+    target_wallet = await wallet_store.load_or(target_id)
     target_wallet.nfts += pulled
     await wallet_store.save(target_id, target_wallet)
     state = await get_state()