/Python/Examples of function decorators in Python

Examples of function decorators in Python

from functools import wraps

def logit(func):
    @wraps(func)
    def with_logging(*args, **kwargs):
        print(func.__name__ + " was called")
        return func(*args, **kwargs)
    return with_logging

@logit
def addition_func(x):
   """Do some math."""
   return x + x


result = addition_func(4)
# Output: addition_func was called

Source: book.pythontips.com

Examples of function decorators in Python

from functools import wraps

def logit(logfile='out.log'):
    def logging_decorator(func):
        @wraps(func)
        def wrapped_function(*args, **kwargs):
            log_string = func.__name__ + " was called"
            print(log_string)
            # Open the logfile and append
            with open(logfile, 'a') as opened_file:
                # Now we log to the specified logfile
                opened_file.write(log_string + '\n')
            return func(*args, **kwargs)
        return wrapped_function
    return logging_decorator

@logit()
def myfunc1():
    pass

myfunc1()
# Output: myfunc1 was called
# A file called out.log now exists, with the above string

@logit(logfile='func2.log')
def myfunc2():
    pass

myfunc2()
# Output: myfunc2 was called
# A file called func2.log now exists, with the above string

Source: book.pythontips.com

Similar codes

0 vote214 views1 answer
How to check nth prime in python
Python
4 votes188 views2 answers
Get stock data in python
Python
1 vote222 views1 answer
Python cv2 imwrite
Python
97 votes254 views5 answers
How to make a python exe
Python
6 votes200 views4 answers
Uninstall python 2.7 in ubuntu
Python
0 vote184 views1 answer
Rotate 90 degrees clockwise counter python
Python
0 vote215 views1 answer
Wxpython mainloop
Python
11 votes186 views3 answers
How to pause a python script
Python
0 vote222 views1 answer
Django array of dates
Python Django
1 vote180 views2 answers
Pandas count rows with value
Python