Started work on core exceptions and request type

README.md

@@ -3,7 +3,7 @@
 ## Sources
 
 ### Signal catching
-- [Catch SIGTERM](https://stackoverflow.com/a/31464349)
+- [Catch SIGTERM](https://dnmtechs.com/graceful-sigterm-signal-handling-in-python-3-best-practices-and-implementation/)
 - [Get ENUM name from value](https://stackoverflow.com/a/38716384)
 
 ### Networking

src/bank_node/bank_node.py

@@ -3,10 +3,14 @@ import signal
 import sys
 import logging
 
+
 from core.config import BankNodeConfig
+from utils import setup_logger
+
 
 class BankNode():
     def __init__(self):
+        setup_logger()
         self.logger = logging.getLogger(__name__)
         self._setup_signals()
         self.config = BankNodeConfig()
@@ -16,13 +20,6 @@ class BankNode():
         signal.signal(signal.SIGTERM, self.gracefully_exit)
         signal.signal(signal.SIGINT, self.gracefully_exit)
 
-    def gracefully_exit(self, signum, frame):
-        signal_name = signal.Signals(signum).name
-        self.logger.warning("Caught %s. Cleaning up before exiting", signal_name)
-        self.cleanup()
-        sys.exit(0)
-
-
     def start_server(self):
         with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as socket_server:
             socket_server.bind(("127.0.0.1", 6969))
@@ -37,3 +34,9 @@ class BankNode():
 
     def cleanup(self):
         pass
+
+    def gracefully_exit(self, signum, frame):
+        signal_name = signal.Signals(signum).name
+        self.logger.warning("Caught %s. Cleaning up before exiting", signal_name)
+        self.cleanup()
+        sys.exit(0)

src/core/exceptions.py

@@ -0,0 +1,13 @@
+class BankProtocolError(Exception):
+    def __init__(self, message):
+        super().__init__(message)
+        self.message = message
+
+
+class InvalidRequest(BankProtocolError):
+    def __init__(self, message):
+        super().__init__(message)
+        self.message = message
+
+
+__all__ = ["BankProtocolError", "InvalidRequest"]

src/core/request.py

@@ -0,0 +1,10 @@
+from core.exceptions import *
+
+
+class Request():
+    def __init__(self, raw_request):
+        try:
+            self.command_code = raw_request[0:1]
+            self.body = raw_request[2:-1]
+        except IndexError:
+            pass

src/utils/__init__.py

@@ -0,0 +1,5 @@
+from .logger import *
+
+__all__ = [
+    *logger.__all__
+]

src/utils/logger.py

@@ -11,3 +11,6 @@ def setup_logger():
     handler.setFormatter(formatter)
 
     logger.addHandler(handler)
+
+
+__all__ = ["setup_logger"]