fix: logger working properly now

Author: PeriniM

scrapegraph-py/examples/async_smartscraper_example.py

@@ -1,9 +1,9 @@
 import asyncio
 
 from scrapegraph_py import AsyncClient
-from scrapegraph_py.logger import get_logger
+from scrapegraph_py.logger import sgai_logger
 
-get_logger(level="DEBUG")
+sgai_logger.set_logging(level="INFO")
 
 
 async def main():

scrapegraph-py/examples/feedback_credits_example.py

@@ -1,7 +1,7 @@
 from scrapegraph_py import SyncClient
-from scrapegraph_py.logger import get_logger
+from scrapegraph_py.logger import sgai_logger
 
-get_logger(level="DEBUG")
+sgai_logger.set_logging(level="INFO")
 
 # Initialize the client
 sgai_client = SyncClient(api_key="your-api-key-here")

scrapegraph-py/examples/smartscraper_example.py

@@ -1,7 +1,7 @@
 from scrapegraph_py import SyncClient
-from scrapegraph_py.logger import get_logger
+from scrapegraph_py.logger import sgai_logger
 
-get_logger(level="DEBUG")
+sgai_logger.set_logging(level="INFO")
 
 # Initialize the client
 sgai_client = SyncClient(api_key="your-api-key-here")

scrapegraph-py/scrapegraph_py/logger.py

@@ -22,53 +22,98 @@ def format(self, record: logging.LogRecord) -> str:
         return super().format(record)
 
 
-def get_logger(
-    name: str = "scrapegraph",
-    level: str = "INFO",
-    log_file: Optional[str] = None,
-    log_format: Optional[str] = None,
-) -> logging.Logger:
-    """
-    Get a configured logger instance with emoji support.
-
-    Args:
-        name: Name of the logger (default: 'scrapegraph')
-        level: Logging level (default: 'INFO')
-        log_file: Optional file path to write logs to
-        log_format: Optional custom log format string
-
-    Returns:
-        logging.Logger: Configured logger instance
-    """
-    logger = logging.getLogger(name)
-
-    # Return existing logger if already configured
-    if logger.handlers:
-        return logger
-
-    # Set log level
-    level = getattr(logging, level.upper(), logging.INFO)
-    logger.setLevel(level)
-
-    # Default format if none provided
-    if not log_format:
-        log_format = "%(levelname)-6s %(asctime)-15s %(message)s"
-
-    formatter = EmojiFormatter(log_format)
-
-    # Console handler
-    console_handler = logging.StreamHandler()
-    console_handler.setFormatter(formatter)
-    logger.addHandler(console_handler)
-
-    # File handler if log_file specified
-    if log_file:
-        file_handler = logging.FileHandler(log_file)
-        file_handler.setFormatter(formatter)
-        logger.addHandler(file_handler)
-
-    return logger
-
-
-# Default sgai logger instance
-sgai_logger = get_logger()
+class ScrapegraphLogger:
+    """Class to manage Scrapegraph logging configuration"""
+
+    _instance = None
+    _initialized = False
+
+    def __new__(cls):
+        if cls._instance is None:
+            cls._instance = super(ScrapegraphLogger, cls).__new__(cls)
+        return cls._instance
+
+    def __init__(self):
+        if not self._initialized:
+            self.logger = logging.getLogger("scrapegraph")
+            self.logger.setLevel(logging.INFO)
+            self.enabled = False
+            self._initialized = True
+
+    def set_logging(
+        self,
+        level: Optional[str] = None,
+        log_file: Optional[str] = None,
+        log_format: Optional[str] = None,
+    ) -> None:
+        """
+        Configure logging settings. If level is None, logging will be disabled.
+
+        Args:
+            level: Logging level (e.g., 'DEBUG', 'INFO'). None to disable logging.
+            log_file: Optional file path to write logs to
+            log_format: Optional custom log format string
+        """
+        # Clear existing handlers
+        self.logger.handlers.clear()
+
+        if level is None:
+            # Disable logging
+            self.enabled = False
+            return
+
+        # Enable logging with specified level
+        self.enabled = True
+        level = getattr(logging, level.upper(), logging.INFO)
+        self.logger.setLevel(level)
+
+        # Default format if none provided
+        if not log_format:
+            log_format = "%(emoji)s %(asctime)-15s %(message)s"
+
+        formatter = EmojiFormatter(log_format)
+
+        # Console handler
+        console_handler = logging.StreamHandler()
+        console_handler.setFormatter(formatter)
+        self.logger.addHandler(console_handler)
+
+        # File handler if log_file specified
+        if log_file:
+            file_handler = logging.FileHandler(log_file)
+            file_handler.setFormatter(formatter)
+            self.logger.addHandler(file_handler)
+
+    def disable(self) -> None:
+        """Disable all logging"""
+        self.logger.handlers.clear()
+        self.enabled = False
+
+    def debug(self, message: str) -> None:
+        """Log debug message if logging is enabled"""
+        if self.enabled:
+            self.logger.debug(message)
+
+    def info(self, message: str) -> None:
+        """Log info message if logging is enabled"""
+        if self.enabled:
+            self.logger.info(message)
+
+    def warning(self, message: str) -> None:
+        """Log warning message if logging is enabled"""
+        if self.enabled:
+            self.logger.warning(message)
+
+    def error(self, message: str) -> None:
+        """Log error message if logging is enabled"""
+        if self.enabled:
+            self.logger.error(message)
+
+    def critical(self, message: str) -> None:
+        """Log critical message if logging is enabled"""
+        if self.enabled:
+            self.logger.critical(message)
+
+
+# Default logger instance
+sgai_logger = ScrapegraphLogger()