Сказка про декораторы в Python

Simple is better than complex.

Материал для тех, кто уже слышал про декораторы, но не понимает зачем они нужны и как их использовать в своих проектах.

Напомню, что декоратор — это функция, которая позволяет обернуть другую функцию для расширения её функциональности без непосредственного изменения её кода. Более подробно можно почитать например тут

Начнем.

Давным давно в большом, уютном проекте жила и трудилась функция fetch_webpage:

import requests

def fetch_webpage():
    webpage = requests.get('https://google.com')
    return webpage

fetch_webpage()

Out:

<Response [200]>

Потом пришли менеджеры и захотели измерить сколько она работает, эта функция, не слишком ли задерживается по вечерам. Позвали менеджеры программиста Олега из большого и уютного проекта. Олег выслушал менеджеров и написал новую версию функции:

import time 
import requests

def fetch_webpage():
    start = time.time()
    webpage = requests.get('https://google.com')
    end = time.time()
    print('Время выполнения: {} секунд.'.format(end-start))
    return webpage

fetch_webpage()

Out:

Время выполнения: 0.1602182388305664 секунд.
<Response [200]>

Менеджеры увидели такую красоту и сразу захотели измерить все функции в проекте и внедрить для них KPI.
Олег как узнал, что от него хотят, сразу стал ругаться и говорить, что он не согласен сейчас все функции переписывать, у него и так еще мультики не смотрены много работы.
А потом подумал и написал функцию wrapper, которая будет измерять время работы другой, чужой функции func и выполнять ее. Посмотрите на них внимательно:

import time 
import requests

def wrapper(func):
    start = time.time()
    out = func()
    end = time.time()
    print('Время выполнения: {} секунд.'.format(end-start))
    return out

def fetch_webpage():
    webpage = requests.get('https://google.com')
    return webpage

wrapper(fetch_webpage)

Out:

Время выполнения: 0.1602182388305664 секунд.
<Response [200]>

Олег обрадовался и захотел переписать весь код большого и уютного проекта, но быстро устал вносить изменения и стал грустить.
3 спринта и 3 дня думал Олег, вспомнил про декораторы и очень обрадовался опять. Вот так можно решить эту проблему с помощью декоратора:

import time 
import requests

def benchmark(func):
    def wrapper():
        start = time.time()
        out = func()
        end = time.time()
        print('Время выполнения: {} секунд.'.format(end-start))
        return out
    return wrapper

@benchmark
def fetch_webpage():
    webpage = requests.get('https://google.com')
    return webpage

fetch_webpage()

Out:

Время выполнения: 0.15940594673156738 секунд.
<Response [200]>

Тут все менеджеры обрадовались, стали поздравлять друг друга, с тем, что теперь все функции будут ручными, а не дикими измеряемыми, а Олегу даже дали медаль или орден, я уже точно не помню.