python - measure execution time function

timeit.py

# author - https://github.com/s3rgeym

import time
import functools
import typing


def timeit(fn: typing.Callable) -> typing.Callable:
    @functools.wraps(fn)
    def timed(*args: typing.Any, **kwargs: typing.Any) -> typing.Any:
        try:
            dt = -time.monotonic()
            return fn(*args, **kwargs)
        finally:
            dt += time.monotonic()
            print("function %s tooks %.3fs", fn.__name__, dt)

    return timed

# example
@timeit
def delete_forbidden_chars(string):
    # deleting forbidden chars
    FORBIDDEN_CHARS = re.escape(punctuation)
    cleared_string = re.sub('['+FORBIDDEN_CHARS+']', '', string).replace('"', "")
    
    return cleared_string