From 34a78d3e644b7d085aea61b3911a5f5a722a729c Mon Sep 17 00:00:00 2001 From: spekulaas Date: Wed, 8 Nov 2023 13:52:26 +0100 Subject: [PATCH] get balance --- goodchain/src/helpers/MenuHelper.py | 2 +- goodchain/src/helpers/TaskHelper.py | 11 +++++++++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/goodchain/src/helpers/MenuHelper.py b/goodchain/src/helpers/MenuHelper.py index 2f76794..022e343 100644 --- a/goodchain/src/helpers/MenuHelper.py +++ b/goodchain/src/helpers/MenuHelper.py @@ -153,7 +153,7 @@ class MenuHelper: print("TODO") case "Check balance": - print("TODO") + taskHelper.getBalance(self) case "Check the pool": transactions = utilityHelper.loadFile("../data/transaction_pool.dat") diff --git a/goodchain/src/helpers/TaskHelper.py b/goodchain/src/helpers/TaskHelper.py index 717194f..09508bd 100644 --- a/goodchain/src/helpers/TaskHelper.py +++ b/goodchain/src/helpers/TaskHelper.py @@ -185,3 +185,14 @@ def exploreBlocks(self): print("Wrong input, try again") return +def getBalance(self): + balance = 0 + blocks = utilityHelper.loadFile("../data/ledger.dat") + for block in blocks: + for transaction in block.data: + if transaction.outputs[0][0] == self.user.public_ser: + balance += transaction.outputs[0][1] + elif transaction.inputs[0][0] == self.user.public_ser: + balance -= transaction.inputs[0][1] + print(f"Your balance is: {balance}") + return balance \ No newline at end of file