/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

5 votes291 views3 answers
Pandas show duplicate rows
Python
0 vote266 views1 answer
Heatmap choos format for annotation
Python
0 vote322 views1 answer
Plotly heatmap with label
Python
1 vote284 views3 answers
Uninstall python ubuntu 18.04
Python
0 vote261 views1 answer
Django snippet 800
Python
0 vote265 views1 answer
How to know if i'm in a virtual environment
Python
0 vote296 views3 answers
Pow
Python
3 votes328 views1 answer
Only keep few key value from dict
Python
0 vote291 views1 answer
How to know the length of a dataset tensorflow
Python
0 vote339 views1 answer
Flask - store object directly in a session [duplicate]
Python Flask