Error handling

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

View File

@ -61,6 +61,11 @@ class BankWorker(multiprocessing.Process):
self.client_socket.shutdown(socket.SHUT_RDWR)
self.client_socket.close()
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:
response = "ER " + e.message + "\n\r"
self.client_socket.sendall(response.encode("utf-8"))

View File

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