feat: change directory structure

main.py

@@ -6,7 +6,7 @@
 ##########################################################
 
 # Third-party imports
-from src.athena.server import AthenaServer
+from src.rpc.athena.server import AthenaServer
 
 
 if __name__ == '__main__':

src/athena/server.py → src/rpc/athena/server.py

@@ -10,7 +10,7 @@
 import grpc
 
 # Third-party imports
-from src.base_server import BaseServer
+from src.rpc.base_server import BaseServer
 from protos.athena import athena_pb2
 from protos.athena import athena_pb2_grpc
 

src/base_server.py → src/rpc/base_server.py

@@ -10,6 +10,9 @@
 import grpc
 from concurrent import futures
 
+# Third-party imports
+from src.utils import logger
+
 
 class BaseServer(object):
 
@@ -41,5 +44,6 @@ def start(self) -> None:
         """
             Starts the server in blocking mode.
         """
+        logger.debug(f"Starting server on {self.target}")
         self.server.start()
         self.server.wait_for_termination()

src/utils/logger.py

@@ -0,0 +1,46 @@
+##########################################################
+#
+# Copyright (C) 2023-PRESENT: Keivan Ipchi Hagh
+#
+# Email:    [email protected]
+# GitHub:   https://github.com/keivanipchihagh
+#
+##########################################################
+
+from colorama import Fore
+from pytz import timezone
+from datetime import datetime
+from colorama import init as colorama_init
+
+colorama_init()
+TZ = timezone('Asia/Tehran')    # Timezone
+
+
+def __now() -> str:
+    """ Returns the current datetime as string """
+    return datetime.now(TZ).strftime('%Y-%m-%d %H:%M:%S')
+
+
+def debug(message: str) -> None:
+    """ Prints a message as debug """
+    print(f"[{Fore.MAGENTA}DEBUG{Fore.BLUE} - {Fore.BLACK}{__now()}{Fore.RESET}]\t{message}")
+
+
+def info(message: str) -> None:
+    """ Prints a message as info """
+    print(f"[{Fore.BLUE}INFO{Fore.BLUE} - {Fore.BLACK}{__now()}{Fore.RESET}]\t{message}")
+
+
+def success(message: str) -> None:
+    """ Prints a message as success """
+    print(f"[{Fore.GREEN}SUCCESS{Fore.RESET} - {Fore.BLACK}{__now()}{Fore.RESET}]\t{message}")
+
+
+def error(message: str) -> None:
+    """ Prints a message as error """
+    print(f"[{Fore.RED}ERROR{Fore.RESET} - {Fore.BLACK}{__now()}{Fore.RESET}]\t{message}")
+
+
+def warning(message: str) -> None:
+    """ Prints a message as warning """
+    print(f"[{Fore.YELLOW}WARNING{Fore.RESET} - {Fore.BLACK}{__now()}{Fore.RESET}]\t{message}")