import logging import os class Logger: def __init__(self, name, level=logging.INFO): # Create a custom logger self.logger = logging.getLogger(name) self.logger.setLevel(level) # Create handlers c_handler = logging.StreamHandler() c_handler.setLevel(level) # Create formatters and add it to handlers formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s') c_handler.setFormatter(formatter) # Add handlers to the logger self.logger.addHandler(c_handler) def get_logger(self): return self.logger