Improve logger

.gitignore

@@ -1,4 +1,5 @@
 # temp directory for testing
+last_blockheight.txt
 config.ini
 http_config.ini
 tests/temp/*

BUILD_ARGS

@@ -1,6 +1,6 @@
 
 --build-arg BASE_IMAGE=python:3.9.13-slim-buster  
---build-arg BUILD_DATE=2024-11-25T05:01:50UTC  
+--build-arg BUILD_DATE=2024-11-25T05:53:51UTC  
 --build-arg DOCKER_BUILD_OPTION=--no-cache --rm=true  
 --build-arg ECHO_OPTION=  
 --build-arg GIT_USER=JINWOO-J  
@@ -11,4 +11,4 @@
 --build-arg REPO_HUB=jinwoo  
 --build-arg SED_OPTION=''  
 --build-arg UNAME_S=Darwin  
---build-arg VERSION=2.0.76  
+--build-arg VERSION=2.0.78  

pawnlib/__version__.py

@@ -2,7 +2,7 @@
 __title__ = 'pawnlib'
 __description__ = 'pawnlib is a collection of libraries for IaC.'
 __url__ = 'https://github.com/jinwoo-j/pawnlib'
-__version__ = '2.0.77'
+__version__ = '2.0.78'
 __author__ = 'Jinwoo Jeong'
 __author_email__ = 'jinwoo@parametacorp.com'
 __license__ = 'MIT'

pawnlib/config/__init__.py

@@ -13,5 +13,7 @@
     ConsoleLoggerAdapter,
     setup_logger,
     getPawnLogger,
-    setup_app_logger
+    setup_app_logger,
+    add_logger,
+    LoggerMixin,
 )

pawnlib/config/logging_config.py

@@ -1022,3 +1022,23 @@ def getPawnLogger(name=None, verbose=0):
     new_logger = ConsoleLoggerAdapter(logger_name=name, verbose=verbose)
     ConsoleLoggerAdapter._global_registry[name] = new_logger
     return new_logger
+
+
+def add_logger(cls):
+    """
+    Decorator to add a logger attribute to a class.
+    """
+    class Wrapped(cls):
+        def __init__(self, *args, **kwargs):
+            super().__init__(*args, **kwargs)
+            self.logger = logging.getLogger(f"{self.__module__}.{self.__class__.__name__}")
+    return Wrapped
+
+
+class LoggerMixin:
+    def get_logger(self):
+        """
+        Returns a logger instance with a name in the format 'module_name.ClassName'.
+        """
+        return logging.getLogger(f"{self.__module__}.{self.__class__.__name__}")
+

pawnlib/utils/http.py

@@ -11,7 +11,7 @@
 from pawnlib.exceptions.notifier import notify_exception
 
 from pawnlib.config.globalconfig import pawnlib_config as pawn, global_verbose, pconf, SimpleNamespace, Null
-from pawnlib.config.logging_config import ConsoleLoggerAdapter, setup_logger
+from pawnlib.config.logging_config import ConsoleLoggerAdapter, setup_logger, LoggerMixin
 from pawnlib.output import (
     NoTraceBackException,
     dump, syntax_highlight, kvPrint, debug_logging,
@@ -3030,7 +3030,7 @@ def get_tx_result(self, tx_hash, max_attempts=None):
             return resp
 
 
-class AsyncCallWebsocket:
+class AsyncCallWebsocket(LoggerMixin):
     def __init__(
             self,
             url: str,
@@ -3072,7 +3072,8 @@ def __init__(
         self.on_last_status = ""
 
         # self.logger = ConsoleLoggerAdapter(logger, "AsyncCallWebsocket", verbose > 0)
-        self.logger = setup_logger(logger, "pawnlib.http.AsyncCallWebsocket", verbose)
+        # self.logger = setup_logger(logger, "pawnlib.http.AsyncCallWebsocket", verbose)
+        self.logger = logger or self.get_logger()
 
         if ssl_options is None:
             self.ssl_options = ssl.create_default_context()
@@ -3357,7 +3358,7 @@ def __init__(
         self.check_tx_result_enabled = check_tx_result_enabled
 
         # self.logger = setup_logger(logger, "pawnlib.http.AsyncGoloopWebsocket", verbose)
-        self.logger = logger or logging.getLogger(__name__)
+        self.logger = logger or self.get_logger()
 
         self.logger.info("Start AsyncGoloopWebsocket")