api/logger.py

import logging

def setup_logger(name):
    logger = logging.getLogger(name)
    if not logger.handlers:
        logger.setLevel(logging.INFO)
        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        
        # Console handler
        console_handler = logging.StreamHandler()
        console_handler.setFormatter(formatter)
        logger.addHandler(console_handler)
        
        # File Handler - Error Level
        # error_file_handler = logging.FileHandler('error.log')
        # error_file_handler.setFormatter(formatter)
        # error_file_handler.setLevel(logging.ERROR)
        # logger.addHandler(error_file_handler)
    
    return logger

logger = setup_logger(__name__)

# New functions for logging different scenarios
def log_generated_chat_id(chat_id, chat_url):
    logger.info(f"Generated Chat ID: {chat_id} - URL: {chat_url}")

def log_model_delay(delay_seconds, chat_id, model):
    logger.info(f"Introducing a delay of {delay_seconds} seconds for model '{model}' (Chat ID: {chat_id})")

def log_http_error(chat_id, error):
    logger.error(f"HTTP error occurred for Chat ID {chat_id}: {error}")

def log_request_error(chat_id, error):
    logger.error(f"Error occurred during request for Chat ID {chat_id}: {error}")

def log_strip_prefix(model_prefix):
    logger.debug(f"Stripping prefix '{model_prefix}' from content.")