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

added close connection / exit function

Browse Source
This commit is contained in:
Spekulaas 2024-01-05 10:23:06 +01:00
parent 210a41de64
commit 3804a7448b
2 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -64,8 +64,7 @@ class MenuHelper:
# check if choice is in menu # check if choice is in menu
match self.start_menu[choice]: match self.start_menu[choice]:
case "Exit": case "Exit":
print(f"{utilityHelper.blinkMessage('Exiting system')}") utilityHelper.closeConnection(self.peer_ip_addr, self.peer_port)
exit()
case "Explore the Blockchain": case "Explore the Blockchain":
blockHelper.exploreBlocks(self) blockHelper.exploreBlocks(self)
@ -114,8 +113,7 @@ class MenuHelper:
# check if choice is in menu # check if choice is in menu
match self.user_main_menu[choice]: match self.user_main_menu[choice]:
case "Exit": case "Exit":
print(f"{utilityHelper.blinkMessage('Exiting system')}") utilityHelper.closeConnection(self.peer_ip_addr, self.peer_port)
exit()
case "Log out": case "Log out":
self.user.logout() self.user.logout()

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

@ -3,6 +3,7 @@ import hashlib
import os import os
from helpers import BlockHelper as blockHelper from helpers import BlockHelper as blockHelper
from helpers import TaskHelper as taskHelper from helpers import TaskHelper as taskHelper
from helpers import SocketHelper as socketHelper
class bcolors: class bcolors:
HEADER = '\033[95m' HEADER = '\033[95m'
@ -94,5 +95,16 @@ def headerMessage(text):
def blinkMessage(text): def blinkMessage(text):
return bcolors.BLINK + text + bcolors.ENDC return bcolors.BLINK + text + bcolors.ENDC
def closeConnection(ip, port):
socketHelper.sendObj(ip, "EXIT", port)
print(f"{bcolors.WARNING}Closing connection{bcolors.ENDC}")
close()
def close():
print(f"{bcolors.WARNING}Exiting system{bcolors.ENDC}")
os._exit(0)
# close connection