decoratorz.py

"""
Playing around with decorators.
"""

__author__ = 'Shay Elmualem'

import time


def calc_func_runtime(func):
    def wrapper():
        start = int(time.time())
        func()
        end = int(time.time())
        return f'Time it took for {func} to finish execution: {end - start}'

    return wrapper


@calc_func_runtime
def sleep_please():
    time.sleep(3)


@calc_func_runtime
def sleep_longer():
    time.sleep(10)


print(sleep_please())
print(sleep_longer())