Decorators in Python are functions that modify other functions. If we need to modify a lot of functions all at once with a minimum of coding, we should use a decorator. A decorator allows us to modify the behavior of a function without changing the codes inside of the function permanently.<\/p>\n\n\n\n\n\n\n\n
A decorator is defined just like a function, but it must take another function as input argument. The decorator at the end must return the modified function. To modify the function, the typical way to this is to define the new function inside the decorator. The new function inside the decorator is sometimes called a “wrapper”.<\/p>\n\n\n\n
Suppose we have made a bunch of one-variable functions that simply “return” some output. But somehow later we actually want those functions to “print” the output instead. In this case, we can make just make a decorator that modifies functions to accomplish this for all functions at once instead of changing the codes in every function.<\/p>\n\n\n\n
Suppose the functions that we made look like this:<\/p>\n\n\n\n
# function that takes one variable and return some output\ndef my_func(x):\n return (x + 3) ** 2<\/code><\/pre><\/div>\n\n\n\nAnd a decorator that prints the output instead of returning:<\/p>\n\n\n\n
# a decorator definition\ndef my_dec(fun):\n def wrapper(x):\n print(fun(x))\n return wrapper<\/code><\/pre><\/div>\n\n\n\nNotice that a decorator must take a function as input argument, and define a new function that modifies that original function. In the end, the decorator must return the name of the function (NOT<\/strong> INCLUDE<\/strong> any arguments). What’s more, the decorator must be above the function that is being applied.<\/p>\n\n\n\nTo apply a decorator to a function definition, simply put @<name of decorator> right above the “def”. When we call the original function, the behavior has been changed.<\/p>\n\n\n\n
@my_dec\ndef my_func(x):\n return (x + 3) ** 2\n\nmy_func(3)<\/code><\/pre><\/div>\n\n\n\nOutput:<\/p>\n\n\n\n
36\n<\/pre>\n\n\n\nThe number of input argument of new function inside a decorator is not necessarily match the original function. Instead, when we call the function that has decorator, the input arguments need to match the new function inside the decorator. For example:<\/p>\n\n\n\n
# a wrapper that takes no input argument\ndef take_input(fun):\n def wrapper():\n x = eval(input("Please enter a number: "))\n print(fun(x))\n return wrapper\n\n# a function take takes one input argument\n@take_input\ndef my_func(x):\n return (x + 3) ** 2\n\n\nmy_func()<\/code><\/pre><\/div>\n\n\n\nOutput:<\/p>\n\n\n\n
Please enter a number: 5\n64\n<\/pre>\n\n\n\nWe can apply multiple decorators to a function. But we need to pay attention to the order of decorators (ORDER MATTERS<\/strong>). <\/p>\n\n\n\n# a decorator that prints some logs\ndef logger(fun):\n def wrapper():\n print("before the function")\n fun()\n print("after the function")\n return wrapper\n\n# a decorator that runs the function twice\ndef do_twice(fun):\n def wrapper():\n fun()\n fun()\n return wrapper\n\n# the closer decorator will be applied earlier\n@logger\n@do_twice\ndef new_func():\n print("hello world")<\/code><\/pre><\/div>\n\n\n\nOutput:<\/p>\n\n\n\n
before the function\nhello world\nhello world\nafter the function\n<\/pre>\n\n\n\nTo make our decorator as flexible as possible to handler all inputs, use *args and **kwargs to allow for arbitrarily many arguments and keyword arguments. <\/p>\n\n\n\n
def error_handler(fun):\n def wrapper(*args, **kwargs):\n try:\n return fun(*args, **kwargs)\n except Exception as e:\n return f"An error occurs! The error is {e}"\n return wrapper\n\n\n@error_handler\ndef division(a, b):\n return a \/ b<\/code><\/pre><\/div>\n\n\n\nOutput:<\/p>\n\n\n\n
2.5\nAn error occurs! The error is division by zero\n<\/pre>\n\n\n\nDecorators can be used for all sort of things, like graphing or debugging or timing your functions or suppressing\/handling errors or printing output, etc. Another common used of a decorator is timing our functions. For example:<\/p>\n\n\n\n
def time_dec(fun):\n def wrapper(*args, **kwargs):\n start = time.time()\n out = fun(*args, **kwargs)\n end = time.time()\n print(f"It took {end - start} seconds for {fun.__name__} to finish")\n return out\n return wrapper\n\n\n@time_dec\ndef slow_fun(n):\n ret = 0\n for i in range(n ** n):\n ret += 1\n ret -= 1\n return ret\n\n\nslow_fun(9)<\/code><\/pre><\/div>\n\n\n\nOutput:<\/p>\n\n\n\n
It took 11.597818851470947 seconds for slow_fun to finish\n<\/pre>\n","protected":false},"excerpt":{"rendered":"Decorators in Python are functions that modify other functions. If we need to modify a lot of functions all at once with a minimum of coding, we should use a decorator. A decorator allows us to modify the behavior of a function without changing the codes inside of the function permanently.<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[4],"tags":[],"class_list":["post-241","post","type-post","status-publish","format-standard","hentry","category-python"],"_links":{"self":[{"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=\/wp\/v2\/posts\/241","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=241"}],"version-history":[{"count":2,"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=\/wp\/v2\/posts\/241\/revisions"}],"predecessor-version":[{"id":270,"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=\/wp\/v2\/posts\/241\/revisions\/270"}],"wp:attachment":[{"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=241"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=241"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.baizhao666.com\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=241"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}