Refactor balance example to remove raw response print and streamline output

Author: theshadow76

examples/balance.py

@@ -15,8 +15,7 @@ async def main():
 
         # 1. Get your balance
         balance = await client.get_user_balance()
-        print("Raw response:", balance)
-        print("\n💰 Your Balance:")
+        print("💰 Your Balance:")
         if balance["available"]:
             for currency, amount in balance["available"].items():
                 if amount > 0:

stakeapi/client.py

@@ -188,10 +188,6 @@ async def _graphql_request(
 
         response = await self._request("POST", "/_api/graphql", data=payload)
 
-        # Debug: print raw response
-        import sys
-        print(f"[DEBUG] Raw GraphQL response: {response}", file=sys.stderr)
-        
         # Check for GraphQL errors
         if "errors" in response:
             error_messages = [error.get("message", "Unknown error") for error in response["errors"]]
@@ -302,20 +298,14 @@ async def get_user_balance(self) -> Dict[str, Dict[str, float]]:
         }
 
         if "user" in data and data["user"] and "balances" in data["user"]:
-            balances = data["user"]["balances"]
-            
-            # Process available balances
-            if "available" in balances:
-                for balance in balances["available"]:
-                    currency = balance.get("currency", "").lower()
-                    amount = float(balance.get("amount", 0))
+            for entry in data["user"]["balances"]:
+                if "available" in entry:
+                    currency = entry["available"].get("currency", "").lower()
+                    amount = float(entry["available"].get("amount", 0))
                     result["available"][currency] = amount
-            
-            # Process vault balances
-            if "vault" in balances:
-                for balance in balances["vault"]:
-                    currency = balance.get("currency", "").lower()
-                    amount = float(balance.get("amount", 0))
+                if "vault" in entry:
+                    currency = entry["vault"].get("currency", "").lower()
+                    amount = float(entry["vault"].get("amount", 0))
                     result["vault"][currency] = amount
 
         return result