Logger to record the use of class methods. Use as decorator above method def and init class with verbose=True

class_logger.py

# Logger gist available here:
# https://gist.github.com/TJMac93/cc17e0aa4546723cfba0d7165dbd9651

from functools import wraps
from .logger import logger
import pandas as pd

def log_step(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        f_args = args[1:]
        f_args = ["Pandas DF" if isinstance(arg, pd.DataFrame) else arg for arg in f_args]


        if args[0].verbose:
            # logger.info(f"Running function: {args[0].name}.{func.__name__}() with arguments: {f_args}")
            pass
        try:
            result = func(*args, **kwargs)
        except Exception as e:
            logger.error(f"Failed to run function: {args[0].name}.{func.__name__}().\n\t\t\t\tException: {repr(e)}")
            raise e
        else:
            if args[0].verbose:
                logger.info(f"Successfully ran function: {args[0].name}.{func.__name__}().")
            return result
    return wrapper