Exception handling and logging added

Author: GoJo-Rika

network_security/exception/exception.py

@@ -0,0 +1,24 @@
+import sys
+
+from network_security.logging import logger
+
+
+class NetworkSecurityError(Exception):
+    def __init__(self, error_message: str, error_details: sys) -> None:
+        self.error_message = error_message
+        _, _, exc_tb = error_details.exc_info()
+
+        self.lineno = exc_tb.tb_lineno
+        self.file_name = exc_tb.tb_frame.f_code.co_filename
+
+    def __str__(self) -> str:
+        return f"Error occured in python script name [{self.file_name}] line number [{self.lineno}] error message [{self.error_message!s}]"
+
+
+# if __name__ == "__main__":
+#     try:
+#         logger.logging.info("Enter the try block")
+#         a = 1 / 0
+#         print("This will not be printed", a)
+#     except Exception as e:
+#         raise NetworkSecurityError(e, sys) from e

network_security/logging/logger.py

@@ -0,0 +1,16 @@
+import logging
+from datetime import UTC, datetime
+from pathlib import Path
+
+LOG_FILE = f"{datetime.now(UTC).strftime('%m_%d_%Y_%H_%M_%S')}.log"
+
+logs_path = Path.cwd() / "logs"
+logs_path.mkdir(parents=True, exist_ok=True)
+
+LOG_FILE_PATH = logs_path / LOG_FILE
+
+logging.basicConfig(
+    filename=LOG_FILE_PATH,
+    format="[ %(asctime)s ] %(lineno)d %(name)s - %(levelname)s - %(message)s",
+    level=logging.INFO,
+)