get balance

This commit is contained in:
Ryan Bakkes 2023-11-08 13:52:26 +01:00
parent f35f9d1a44
commit 34a78d3e64
2 changed files with 12 additions and 1 deletions

View File

@ -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")

View File

@ -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