Fixed token balance in the GUI

Tribler · Sep 25, 2019 · 6a16220 · 6a16220
1 parent cdba1d6
commit 6a16220
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 12 deletions.
diff --git a/TriblerGUI/tribler_window.py b/TriblerGUI/tribler_window.py
@@ -603,12 +603,9 @@ def received_trustchain_statistics(self, statistics):
         self.trust_page.received_trustchain_statistics(statistics)
 
         statistics = statistics["statistics"]
-        if 'latest_block' in statistics:
-            balance = (statistics["latest_block"]["transaction"]["total_up"] -
-                       statistics["latest_block"]["transaction"]["total_down"])
-            self.set_token_balance(balance)
-        else:
-            self.token_balance_label.setText("0 MB")
+        balance = (statistics.get("total_up", 0) -
+                   statistics.get("total_down", 0))
+        self.set_token_balance(balance)
 
         # If trust page is currently visible, then load the graph as well
         if self.stackedWidget.currentIndex() == PAGE_TRUST:

diff --git a/TriblerGUI/widgets/trustpage.py b/TriblerGUI/widgets/trustpage.py
@@ -124,15 +124,13 @@ def load_blocks(self):
                                          self.received_trustchain_blocks)
 
     def received_trustchain_statistics(self, statistics):
-        if not statistics:
+        if not statistics or "statistics" not in statistics:
             return
         statistics = statistics["statistics"]
         self.public_key = statistics["id"]
-        total_up = 0
-        total_down = 0
-        if 'latest_block' in statistics:
-            total_up = statistics["latest_block"]["transaction"]["total_up"]
-            total_down = statistics["latest_block"]["transaction"]["total_down"]
+
+        total_up = statistics.get("total_up", 0)
+        total_down = statistics.get("total_down", 0)
 
         self.window().trust_contribution_amount_label.setText("%s MBytes" % (total_up / self.byte_scale))
         self.window().trust_consumption_amount_label.setText("%s MBytes" % (total_down / self.byte_scale))