fix python sdk leader dict

humanprotocol · Sep 7, 2023 · cedc395 · cedc395
1 parent 2c63df6
commit cedc395
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 47 deletions.
diff --git a/packages/sdk/python/human-protocol-sdk/human_protocol_sdk/staking.py b/packages/sdk/python/human-protocol-sdk/human_protocol_sdk/staking.py
@@ -332,17 +332,17 @@ def get_leaders(self, filter: LeaderFilter = LeaderFilter()) -> List[dict]:
             {
                 "id": leader["id"],
                 "address": leader["address"],
-                "amount_staked": leader["amountStaked"],
-                "amount_allocated": leader["amountAllocated"],
-                "amount_locked": leader["amountLocked"],
-                "locked_until_timestamp": leader["lockedUntilTimestamp"],
-                "amount_withdrawn": leader["amountWithdrawn"],
-                "amount_slashed": leader["amountSlashed"],
-                "reputation": leader["reputation"],
-                "reward": leader["reward"],
-                "amount_jobs_launched": leader["amountJobsLaunched"],
+                "amount_staked": int(leader["amountStaked"]),
+                "amount_allocated": int(leader["amountAllocated"]),
+                "amount_locked": int(leader["amountLocked"]),
+                "locked_until_timestamp": int(leader["lockedUntilTimestamp"]),
+                "amount_withdrawn": int(leader["amountWithdrawn"]),
+                "amount_slashed": int(leader["amountSlashed"]),
+                "reputation": int(leader["reputation"]),
+                "reward": int(leader["reward"]),
+                "amount_jobs_launched": int(leader["amountJobsLaunched"]),
                 "role": leader["role"],
-                "fee": leader["fee"],
+                "fee": int(leader["fee"]) if leader["fee"] else None,
                 "public_key": leader["publicKey"],
                 "webhook_url": leader["webhookUrl"],
                 "url": leader["url"],
@@ -377,17 +377,17 @@ def get_leader(self, address: Optional[str] = None) -> dict:
         return {
             "id": leader["id"],
             "address": leader["address"],
-            "amount_staked": leader["amountStaked"],
-            "amount_allocated": leader["amountAllocated"],
-            "amount_locked": leader["amountLocked"],
-            "locked_until_timestamp": leader["lockedUntilTimestamp"],
-            "amount_withdrawn": leader["amountWithdrawn"],
-            "amount_slashed": leader["amountSlashed"],
-            "reputation": leader["reputation"],
-            "reward": leader["reward"],
-            "amount_jobs_launched": leader["amountJobsLaunched"],
+            "amount_staked": int(leader["amountStaked"]),
+            "amount_allocated": int(leader["amountAllocated"]),
+            "amount_locked": int(leader["amountLocked"]),
+            "locked_until_timestamp": int(leader["lockedUntilTimestamp"]),
+            "amount_withdrawn": int(leader["amountWithdrawn"]),
+            "amount_slashed": int(leader["amountSlashed"]),
+            "reputation": int(leader["reputation"]),
+            "reward": int(leader["reward"]),
+            "amount_jobs_launched": int(leader["amountJobsLaunched"]),
             "role": leader["role"],
-            "fee": leader["fee"],
+            "fee": int(leader["fee"]) if leader["fee"] else None,
             "public_key": leader["publicKey"],
             "webhook_url": leader["webhookUrl"],
             "url": leader["url"],

diff --git a/packages/sdk/python/human-protocol-sdk/test/human_protocol_sdk/test_staking.py b/packages/sdk/python/human-protocol-sdk/test/human_protocol_sdk/test_staking.py
@@ -332,15 +332,15 @@ def test_get_leaders(self):
                     {
                         "id": self.gas_payer.address,
                         "address": self.gas_payer.address,
-                        "amount_staked": "100",
-                        "amount_allocated": "50",
-                        "amount_locked": "25",
-                        "locked_until_timestamp": "0",
-                        "amount_withdrawn": "25",
-                        "amount_slashed": "25",
-                        "reputation": "25",
-                        "reward": "25",
-                        "amount_jobs_launched": "25",
+                        "amount_staked": 100,
+                        "amount_allocated": 50,
+                        "amount_locked": 25,
+                        "locked_until_timestamp": 0,
+                        "amount_withdrawn": 25,
+                        "amount_slashed": 25,
+                        "reputation": 25,
+                        "reward": 25,
+                        "amount_jobs_launched": 25,
                         "role": "role",
                         "fee": None,
                         "public_key": None,
@@ -396,15 +396,15 @@ def test_get_leader(self):
                 {
                     "id": staker_address,
                     "address": staker_address,
-                    "amount_staked": "100",
-                    "amount_allocated": "50",
-                    "amount_locked": "25",
-                    "locked_until_timestamp": "0",
-                    "amount_withdrawn": "25",
-                    "amount_slashed": "25",
-                    "reputation": "25",
-                    "reward": "25",
-                    "amount_jobs_launched": "25",
+                    "amount_staked": 100,
+                    "amount_allocated": 50,
+                    "amount_locked": 25,
+                    "locked_until_timestamp": 0,
+                    "amount_withdrawn": 25,
+                    "amount_slashed": 25,
+                    "reputation": 25,
+                    "reward": 25,
+                    "amount_jobs_launched": 25,
                     "role": "role",
                     "fee": None,
                     "public_key": None,
@@ -455,15 +455,15 @@ def test_get_leader_default_account(self):
                 {
                     "id": self.w3.eth.default_account,
                     "address": self.w3.eth.default_account,
-                    "amount_staked": "100",
-                    "amount_allocated": "50",
-                    "amount_locked": "25",
-                    "locked_until_timestamp": "0",
-                    "amount_withdrawn": "25",
-                    "amount_slashed": "25",
-                    "reputation": "25",
-                    "reward": "25",
-                    "amount_jobs_launched": "25",
+                    "amount_staked": 100,
+                    "amount_allocated": 50,
+                    "amount_locked": 25,
+                    "locked_until_timestamp": 0,
+                    "amount_withdrawn": 25,
+                    "amount_slashed": 25,
+                    "reputation": 25,
+                    "reward": 25,
+                    "amount_jobs_launched": 25,
                     "role": "role",
                     "fee": None,
                     "public_key": None,