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

beautified messages

Browse Source
This commit is contained in:
Ryan Bakkes 2023-11-11 20:30:06 +01:00
parent e0edbfefaf
commit 39c3b09b81
6 changed files with 40 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
goodchain/src/helpers/BlockHelper.py
View File

@ -1,6 +1,5 @@
from classes.Transaction import Tx
from classes.TxBlock import TxBlock
from helpers import SignatureHelper as signatureHelper
from helpers import UtilityHelper as utilityHelper
import time
@ -31,12 +30,12 @@ def exploreBlocks(self):
user_input = int(user_input) -1
except:
utilityHelper.clearScreen()
print("Wrong input, try again")
print(f"{utilityHelper.warningMessage('Wrong input, try again')}")
return
if user_input > len(blocks) - 1 or user_input < 0:
utilityHelper.clearScreen()
print("Wrong input, try again")
print(f"{utilityHelper.warningMessage('Wrong input, try again')}")
return
# print all info of the block
@ -92,7 +91,7 @@ def exploreBlocks(self):
exploreBlocks(self)
case _:
utilityHelper.clearScreen()
print("Wrong input, try again")
print(f"{utilityHelper.warningMessage('Wrong input, try again')}")
return
def createBlock(self):
@ -106,20 +105,20 @@ def createBlock(self):
lastBlock = lastBlock[-1]
if lastBlock != None and time.time() - lastBlock.date < 180:
print(f"You can only mine a block every 3 minutes. Time left: {int(180 - (time.time() - lastBlock.date))}")
print(f"{utilityHelper.bcolors.WARNING}You can only mine a block every 3 minutes. Time left: {int(180 - (time.time() - lastBlock.date))}{utilityHelper.bcolors.ENDC}")
return False
if lastBlock != None and lastBlock.metadata['validated'] == False:
print("You can only mine a block if the last block is validated")
print(f"{utilityHelper.warningMessage('You can only mine a block if the last block is validated')}")
return False
transactions = Tx()
transactions = utilityHelper.loadFile("../data/transaction_pool.dat")
if len(transactions) < 5:
print("You need atleast 5 transactions to mine a block")
print(f"{utilityHelper.warningMessage('You need atleast 5 transactions to mine a block')}")
return False
print("Select items out the pool to mine a block, last 2 values added to list.")
print("Select items out the pool to mine a block, 2 values already selected.")
if input("Press enter to continue... (type b to return)") == "b":
return False
utilityHelper.clearScreen()
@ -132,7 +131,6 @@ def createBlock(self):
fees[x] = transaction.inputs[0][1] - transaction.outputs[0][1]
x+=1
print(fees)
fees_list = list(fees.keys())
selected_transactions = fees_list[:2]
available_transactions = fees_list[2:]
@ -153,20 +151,20 @@ def createBlock(self):
for i in user_input:
try:
if int(i) not in available_transactions:
print("Wrong input, try again")
print(f"{utilityHelper.warningMessage('Wrong input, try again')}")
return False
selected_transactions.append(int(i))
except:
utilityHelper.clearScreen()
print("Wrong input, try again")
print(f"{utilityHelper.warningMessage('Wrong input, try again')}")
return False
if len(selected_transactions) > 10:
print("You can only select up to 10 transactions")
print(f"{utilityHelper.warningMessage('You can only select up to 10 transactions')}")
return False
if len(selected_transactions) < 5:
print("You need atleast 5 transactions to mine a block")
print(f"{utilityHelper.warningMessage('You need atleast 5 transactions to mine a block')}")
return False
only_personal = True
@ -179,7 +177,7 @@ def createBlock(self):
only_personal = False
if only_personal:
print("You need atleast 1 transaction from another user")
print(f"{utilityHelper.warningMessage('You need atleast 1 transaction from another user')}")
return False
if lastBlock == []:
lastBlock = None
@ -200,18 +198,18 @@ def createBlock(self):
utilityHelper.clearScreen()
if not block.good_nonce():
print("ERROR! Bad nonce")
print(f"{utilityHelper.errorMessage('ERROR! Bad nonce')}")
return False
print("Success! Nonce is good!")
print(f"{utilityHelper.successMessage('Success! Nonce is good!')}")
print(f'Accepted Nonce = {str(nonce)}')
print("elapsed time: " + str(elapsed) + " s.")
if elapsed < MIN_MINING_TIME:
print("Mining declined, too fast")
print(f"{utilityHelper.errorMessage('Mining declined, too fast')}")
return False
elif elapsed > MAX_MINING_TIME:
print("Mining declined, too Slow")
print(f"{utilityHelper.errorMessage('Mining declined, too Slow')}")
return False
# add block to blockchain
@ -298,7 +296,7 @@ def validateMinedBlock(self):
try:
del blocks[-1]
except:
print("error biatch")
print(f"{utilityHelper.errorMessage('Something went wrong')}")
return False
utilityHelper.resetFile("../data/ledger.dat")

8
goodchain/src/helpers/DatabaseHelper.py
View File

@ -1,8 +1,8 @@
import sqlite3
from helpers import UtilityHelper as utilityHelper
class DatabaseHelper:
def __init__(self, db=None):
print("Initializing database")
self.conn = None
self.cursor = None
@ -48,7 +48,6 @@ class DatabaseHelper:
self.cursor.execute("SELECT * FROM `users` WHERE (`username` = ? AND `password` = ?)", (username, password,))
return self.cursor.fetchone()
except sqlite3.Error as error:
print(error)
return None
def createUser(self, private_key, public_key, username, password):
@ -59,7 +58,6 @@ class DatabaseHelper:
self.commit()
return True
except sqlite3.Error as error:
print(error)
return False
def changePassword(self, user_privatekey, old_password, password):
@ -75,7 +73,6 @@ class DatabaseHelper:
return True
except sqlite3.Error as error:
print(error)
return None
def changeUsername(self, user_privatekey, username):
@ -90,7 +87,6 @@ class DatabaseHelper:
return True
except sqlite3.Error as error:
print(error)
return None
def deleteUser(self, user_privatekey):
@ -106,7 +102,6 @@ class DatabaseHelper:
return True
except sqlite3.Error as error:
print(error)
return None
def fetchUserByUsername(self, username):
@ -119,7 +114,6 @@ class DatabaseHelper:
return self.cursor.fetchone()
except sqlite3.Error as error:
print(error)
return None
def checkMigrations(self):

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

@ -4,6 +4,7 @@ from cryptography.hazmat.primitives import serialization
from cryptography.hazmat.primitives import hashes
from cryptography.hazmat.primitives.asymmetric import padding
from cryptography.hazmat.backends import default_backend
from helpers import UtilityHelper as utilityHelper
def generateKeys():
private_key = rsa.generate_private_key(public_exponent=65537,key_size=2048)
@ -81,5 +82,5 @@ def verify(message, signature, public_ser):
except InvalidSignature:
return False
except:
print("Error executing 'public_key.verify'")
print(f"{utilityHelper.errorMessage('Error executing <public_key.verify>')}")
return False

1
goodchain/src/helpers/TransactionHelper.py
View File

@ -1,5 +1,4 @@
from classes.Transaction import Tx
from classes.TxBlock import TxBlock
from helpers import UtilityHelper as utilityHelper
from helpers import TaskHelper as taskHelper

12
goodchain/src/helpers/UtilityHelper.py
View File

@ -24,6 +24,18 @@ def computeHash(data):
def clearScreen():
os.system('cls' if os.name == 'nt' else 'clear')
def printLogo():
print(f"""{bcolors.HEADER}
----------------------v1.0.0----------------------
____ _ _ _
/ ___| ___ ___ __| | ___| |__ __ _(_)_ __
| | _ / _ \ / _ \ / _` |/ __| '_ \ / _` | | '_ \
| |_| | (_) | (_) | (_| | (__| | | | (_| | | | | |
\____|\___/ \___/ \__,_|\___|_| |_|\__,_|_|_| |_|
------------------Initializing--------------------
{bcolors.ENDC}""")
def createFile(fileloc):
try:
savefile = open(fileloc, "xb")

12
goodchain/src/main.py
View File

@ -11,17 +11,21 @@
from helpers.DatabaseHelper import DatabaseHelper
from helpers.MenuHelper import MenuHelper
from helpers import UtilityHelper as utilityHelper
from time import sleep
if __name__ == "__main__":
utilityHelper.clearScreen()
utilityHelper.printLogo()
# start db connection
db = DatabaseHelper("../data/goodchain.db")
db.checkMigrations()
# check ledger and pool file
if utilityHelper.createFile("../data/ledger.dat"):
print("Created ledger")
if utilityHelper.createFile("../data/transaction_pool.dat"):
print("Created transaction pool")
utilityHelper.createFile("../data/ledger.dat")
utilityHelper.createFile("../data/transaction_pool.dat")
sleep(1)
# start menu
menu = MenuHelper(db)