Thastertyn/nu
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Error handling

Browse Source
This commit is contained in:
Thastertyn 2025-02-07 08:04:29 +01:00
parent da7c809a16
commit 756d6d3dd4
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/bank_node/bank_worker.py
View File

@ -61,6 +61,11 @@ class BankWorker(multiprocessing.Process):
self.client_socket.shutdown(socket.SHUT_RDWR) self.client_socket.shutdown(socket.SHUT_RDWR)
self.client_socket.close() self.client_socket.close()
break break
except UnicodeDecodeError:
self.logger.warning("Received a non utf-8 message")
response = "ER Not utf-8 message"
self.client_socket.sendall(response.encode("utf-8"))
break
except BankNodeError as e: except BankNodeError as e:
response = "ER " + e.message + "\n\r" response = "ER " + e.message + "\n\r"
self.client_socket.sendall(response.encode("utf-8")) self.client_socket.sendall(response.encode("utf-8"))

3
src/bank_protocol/commands/account_deposit_command.py
View File

@ -5,6 +5,9 @@ from utils.constants import MONEY_AMOUNT_MAXIMUM
def account_deposit(request: Request, config: BankNodeConfig): def account_deposit(request: Request, config: BankNodeConfig):
if request.body is None:
raise InvalidRequest("Invalid request format")
try: try:
split_body = request.body.split("/") split_body = request.body.split("/")
split_ip = split_body[1].split(" ") split_ip = split_body[1].split(" ")