¿Cómo escribir un decorador personalizado que funcione correctamente tanto como decorador regular (@my_decorator) como decorador con argumentos (@my_decorator(arg=1))?