How to write a custom decorator that works correctly both as a regular decorator (@my_decorator) and as a decorator with arguments (@my_decorator(arg=1))?