Decorador para qualquer função com args e kwargs

Por que o wrapper precisa de args e kwargs

O decorador da aula anterior tinha um limite escondido. O wrapper foi escrito como def wrapper(), sem parâmetros, e por dentro chamava func() também sem argumentos. Isso funciona para dizer_ola, que não recebe nada. Mas tente decorar uma função que soma dois números e tudo desmorona: quando você chama a versão decorada passando os dois números, quem recebe esses argumentos é o wrapper, que não os espera. O Python reclama na hora que o wrapper não aceita argumentos. Um decorador de verdade precisa servir para qualquer função, e é aí que entram *args e **kwargs.

def com_moldura(func):
    def wrapper():          # sem parâmetros: problema!
        print("--- antes ---")
        func()
        print("--- depois ---")
    return wrapper

@com_moldura
def somar(a, b):
    return a + b

somar(2, 3)
# TypeError: wrapper() takes 0 positional arguments but 2 were given

O wrapper sem parâmetros não sabe o que fazer com os argumentos de somar.

O wrapper que aceita qualquer função

A solução é fazer o wrapper aceitar tudo e repassar tudo. Você já viu *args e **kwargs no módulo de funções avançadas: *args recolhe os argumentos posicionais numa tupla e **kwargs recolhe os nomeados num dicionário. Escrevendo def wrapper(*args, **kwargs), o wrapper aceita qualquer chamada. Dentro dele, chamamos a função original com func(*args, **kwargs), o que desempacota tudo de volta e entrega os argumentos exatos que a função esperava. Falta um detalhe fácil de esquecer e caro de depurar: devolver o resultado. Se a função original retorna um valor, o wrapper precisa fazer return func(...); caso contrário, a versão decorada passa a devolver None em silêncio.

def com_moldura(func):
    def wrapper(*args, **kwargs):
        print("--- antes ---")
        resultado = func(*args, **kwargs)   # repassa tudo
        print("--- depois ---")
        return resultado                    # devolve o valor original
    return wrapper

@com_moldura
def somar(a, b):
    return a + b

@com_moldura
def cumprimentar(nome, saudacao="Olá"):
    return f"{saudacao}, {nome}!"

print(somar(2, 3))
# --- antes ---
# --- depois ---
# 5
print(cumprimentar("Ana", saudacao="Oi"))
# --- antes ---
# --- depois ---
# Oi, Ana!

Com *args e **kwargs, o mesmo decorador serve para somar e para cumprimentar.

Verificando o padrão universal

A partir de agora, todo wrapper que você escrever nasce com a assinatura def wrapper(*args, **kwargs) e termina com return func(*args, **kwargs) em algum ponto. Esse é o formato genérico que funciona com qualquer função, receba ela zero, dois ou dez argumentos, posicionais ou nomeados. Guarde os dois cuidados: aceitar tudo na entrada com *args e **kwargs, e devolver o resultado no fim. Com isso resolvido, o decorador já é utilizável de verdade. Ainda sobra um detalhe cosmético importante, que a próxima aula ataca: o nome da função decorada passa a mostrar wrapper, e não o nome original.