Playing with python decorators

decorator.py

from datetime import datetime
from time import sleep
from sys import argv


def measure(func):
    def decorator(n):
        start = datetime.now()
        res = func(n)
        end = datetime.now()
        print('%s completed in %.2f seconds' % (func.__name__, (end - start).total_seconds()))
        return res
    return decorator


@measure
def lazy(n):
    print('i am going to take a snap!')
    sleep(float(n))


if __name__ == '__main__':
    n = argv[1] if len(argv) > 1 else 1
    lazy(n)