Untitled

"""
whenever we are doing compute intensive operation - multi-processing

"""
import time
import multiprocessing


def timeit(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        result = func(*args, **kwargs)
        end = time.time() - start
        print(f"[ INFO ] total time to execute :: {end}")
        return result

    return wrapper


def some_heavy_work(range_):
    return [i ** 2 for i in range(range_)]


@timeit
def main():
    ranges = [10000001, 10000002, 10000003, 10000004, 10000005, 10000006,
              10000071, 10000042, 10000053, 10000024, 10000035, 10000016]

    pool = multiprocessing.Pool(6)
    pool.map(some_heavy_work, ranges)


if __name__ == "__main__":
    main()