use decorator to count processing time

timeit_decorator.py

from time import time


def timeit(func):
    def wrap(*args, **kwargs):
        start = time()
        result = func(*args, **kwargs)
        print('[Time]: {:6f}'.format(time() - start))
        return result
    return wrap


@timeit
def multiply(n, nums=None):
    if nums:
        for i in nums:
            n *= i ** 2

    return n


if __name__ == '__main__':
    print(multiply(2, range(1, 10)))
    print(multiply(5, range(1, 10)))