Как да напишете персонализиран декоратор, който работи правилно както като обикновен декоратор (@my_decorator), така и като декоратор с аргументи (@my_decorator(arg=1))?