Measure time wrapper

timewrapper.py

import logging
from datetime import datetime
import functools

# Setting up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

def measure_time(func):
    @functools.wraps(func)
    def wrapper_measure_time(*args, **kwargs):
        start_time = datetime.now()
        result = func(*args, **kwargs)
        end_time = datetime.now()
        elapsed = (end_time - start_time).total_seconds()
        logger.info(
            f"{func.__name__} function completed in {int(elapsed // 60)} minutes "
            f"and {elapsed % 60:.2f} seconds."
        )
        return result

    return wrapper_measure_time

@measure_time
def pipeline():
    # Do stuff here, it will be timed
    logger.info("Pipeline started.")


if __name__ == "__main__":
    pipeline()