import logging import sys logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') logger = logging.getLogger(__name__) def log_message(message): logger.info(message) print(message, flush=True) sys.stdout.flush()