Como você escreveria um decorator que recebe os próprios argumentos?

Você adiciona uma terceira camada. A função externa recebe os argumentos do decorator e devolve o decorator de verdade, que recebe a função e devolve um wrapper que a chama. Aplique functools.wraps no wrapper para que nome, docstring e metadados de assinatura sobrevivam. Por exemplo, um decorator de retry recebe a contagem de tentativas na camada externa e o wrapper implementa o loop.

Três camadas, e eu digo em voz alta na ordem para não me perder. A função mais externa recebe a configuração, digamos o número de retries. Ela devolve o decorator, que recebe a função alvo. Esse devolve o wrapper, que recebe star args e star star kwargs, faz o trabalho em volta da chamada e devolve o resultado. Eu sempre coloco functools.wraps no wrapper, senão a função decorada se reporta como wrapper em todo lugar, e o Flask ou o Celery vão registrar alegremente duas rotas com o mesmo nome e falhar de um jeito confuso. O que eu mais escrevi foi um retry com backoff exponencial para chamadas instáveis de upstream: três tentativas, dormindo um intervalo que dobra, relançando na última. Um refinamento útil é aceitar também a função direto, para funcionar com e sem parênteses, embora sinceramente eu costume exigir os parênteses porque a lógica de ramificação custa mais clareza do que economiza.