auto-analyst-backend-2

Running

auto-analyst-backend-2 / src /utils /logger.py

Arslan1997

adding free tools

1d17209 3 months ago

3.93 kB

import os
import time
import logging
import sys
from dotenv import load_dotenv

load_dotenv()

class Logger:
    def __init__(self, name: str, see_time: bool = False, console_log: bool = False, level: int = logging.INFO):
        self.is_dev = os.getenv("ENVIRONMENT", "development") == "development"
        self.logger = logging.getLogger(name)
        self.logger.setLevel(level if self.is_dev else logging.WARNING)

        # Logger(name) is constructed in many modules; logging.getLogger returns the
        # same underlying logger each time, so adding handlers unconditionally would
        # duplicate every line once per construction.
        if self.logger.handlers:
            return

        if not self.is_dev:
            # Production: no log files (container filesystems are ephemeral), but
            # warnings and errors must reach stdout or they are invisible in the
            # platform logs and every 500 becomes undiagnosable.
            prod_handler = logging.StreamHandler(sys.stdout)
            prod_handler.setFormatter(
                logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
            )
            prod_handler.setLevel(logging.WARNING)
            self.logger.addHandler(prod_handler)
            return

        os.makedirs("./logs", exist_ok=True)

        formatter = logging.Formatter(
            "%(asctime)s - %(name)s - %(levelname)s - %(message)s" if see_time else "%(message)s,"
        )

        # File handler with UTF-8 encoding
        file_handler = logging.FileHandler(f"./logs/{name}.log", encoding='utf-8')
        file_handler.setFormatter(formatter)
        self.logger.addHandler(file_handler)

        if console_log:
            # Console handler with UTF-8 encoding
            console_handler = logging.StreamHandler(sys.stdout)
            console_handler.setFormatter(formatter)
            # Force UTF-8 encoding for console output
            if hasattr(console_handler.stream, 'reconfigure'):
                console_handler.stream.reconfigure(encoding='utf-8')
            self.logger.addHandler(console_handler)

    def log_message(self, message: str, level: int = logging.INFO):
        # Level filtering is handled by the logger/handler levels set in __init__,
        # so production keeps WARNING and above and drops the INFO/DEBUG chatter.
        try:
            if level == logging.INFO:
                self.logger.info(message)
            elif level == logging.ERROR:
                self.logger.error(message)
            elif level == logging.WARNING:
                self.logger.warning(message)
            elif level == logging.DEBUG:
                self.logger.debug(message)
            else:
                self.logger.info(message)
        except UnicodeEncodeError:
            # Fallback: remove emoji characters if encoding fails
            safe_message = message.encode('ascii', 'ignore').decode('ascii')
            if level == logging.INFO:
                self.logger.info(safe_message)
            elif level == logging.ERROR:
                self.logger.error(safe_message)
            elif level == logging.WARNING:
                self.logger.warning(safe_message)
            elif level == logging.DEBUG:
                self.logger.debug(safe_message)
            else:
                self.logger.info(safe_message)

    def disable_logging(self):
        self.logger.disabled = True


def log_time(func):
    def wrapper(*args, **kwargs):
        if os.getenv("ENV", "development") != "development":
            return func(*args, **kwargs)
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        logger = Logger(func.__name__ + "_time", see_time=True, level=logging.INFO)
        logger.log_message(f"Function: {func.__name__}, Execution time: {round(end_time - start_time, 5)} seconds")
        return result
    return wrapper

	import os
	import time
	import logging
	import sys
	from dotenv import load_dotenv

	load_dotenv()

	class Logger:
	def __init__(self, name: str, see_time: bool = False, console_log: bool = False, level: int = logging.INFO):
	self.is_dev = os.getenv("ENVIRONMENT", "development") == "development"
	self.logger = logging.getLogger(name)
	self.logger.setLevel(level if self.is_dev else logging.WARNING)

	# Logger(name) is constructed in many modules; logging.getLogger returns the
	# same underlying logger each time, so adding handlers unconditionally would
	# duplicate every line once per construction.
	if self.logger.handlers:
	return

	if not self.is_dev:
	# Production: no log files (container filesystems are ephemeral), but
	# warnings and errors must reach stdout or they are invisible in the
	# platform logs and every 500 becomes undiagnosable.
	prod_handler = logging.StreamHandler(sys.stdout)
	prod_handler.setFormatter(
	logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
	)
	prod_handler.setLevel(logging.WARNING)
	self.logger.addHandler(prod_handler)
	return

	os.makedirs("./logs", exist_ok=True)

	formatter = logging.Formatter(
	"%(asctime)s - %(name)s - %(levelname)s - %(message)s" if see_time else "%(message)s,"
	)

	# File handler with UTF-8 encoding
	file_handler = logging.FileHandler(f"./logs/{name}.log", encoding='utf-8')
	file_handler.setFormatter(formatter)
	self.logger.addHandler(file_handler)

	if console_log:
	# Console handler with UTF-8 encoding
	console_handler = logging.StreamHandler(sys.stdout)
	console_handler.setFormatter(formatter)
	# Force UTF-8 encoding for console output
	if hasattr(console_handler.stream, 'reconfigure'):
	console_handler.stream.reconfigure(encoding='utf-8')
	self.logger.addHandler(console_handler)

	def log_message(self, message: str, level: int = logging.INFO):
	# Level filtering is handled by the logger/handler levels set in __init__,
	# so production keeps WARNING and above and drops the INFO/DEBUG chatter.
	try:
	if level == logging.INFO:
	self.logger.info(message)
	elif level == logging.ERROR:
	self.logger.error(message)
	elif level == logging.WARNING:
	self.logger.warning(message)
	elif level == logging.DEBUG:
	self.logger.debug(message)
	else:
	self.logger.info(message)
	except UnicodeEncodeError:
	# Fallback: remove emoji characters if encoding fails
	safe_message = message.encode('ascii', 'ignore').decode('ascii')
	if level == logging.INFO:
	self.logger.info(safe_message)
	elif level == logging.ERROR:
	self.logger.error(safe_message)
	elif level == logging.WARNING:
	self.logger.warning(safe_message)
	elif level == logging.DEBUG:
	self.logger.debug(safe_message)
	else:
	self.logger.info(safe_message)

	def disable_logging(self):
	self.logger.disabled = True


	def log_time(func):
	def wrapper(args, *kwargs):
	if os.getenv("ENV", "development") != "development":
	return func(args, *kwargs)
	start_time = time.time()
	result = func(args, *kwargs)
	end_time = time.time()
	logger = Logger(func.__name__ + "_time", see_time=True, level=logging.INFO)
	logger.log_message(f"Function: {func.__name__}, Execution time: {round(end_time - start_time, 5)} seconds")
	return result
	return wrapper