Hogyan írjunk egy egyedi dekorátort, amely helyesen működik mind normál dekorátorként (@my_decorator), mind argumentumokkal (@my_decorator(arg=1))?