Spekulaas/Blockchain_School
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added balance check

Browse Source
This commit is contained in:
Ryan Bakkes 2023-11-09 09:03:57 +01:00
parent 711f7b1c63
commit a15cdce0c0
2 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
goodchain/src/helpers/MenuHelper.py
View File

@ -153,7 +153,8 @@ class MenuHelper:
print("TODO")
case "Check balance":
taskHelper.getBalance(self)
print(f"Your balance is: {taskHelper.getBalance(self)}")
case "Check the pool":
transactions = utilityHelper.loadFile("../data/transaction_pool.dat")

13
goodchain/src/helpers/TaskHelper.py
View File

@ -22,7 +22,7 @@ def transaction(self):
return False
try:
amount = int(amount)
amount = float(amount)
except:
print("Wrong input, try again")
return False
@ -32,11 +32,19 @@ def transaction(self):
return False
try:
fee = int(fee)
fee = float(fee)
except:
print("Wrong input, try again")
return False
utilityHelper.clearScreen()
print(f"Processing transaction of {amount + fee} coins, {receiver} will receive {amount} coins")
if amount + fee > getBalance(self):
print("You dont have enough money")
return False
new_tx = Tx()
new_tx.createTransaction(self.user.public_ser, self.user.private_ser, amount, fee, receiver_data[1])
return new_tx
@ -194,5 +202,4 @@ def getBalance(self):
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