/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

6 votes337 views6 answers
How to check if a value is nan in python
Python
0 vote294 views1 answer
Pandas use map lambda to fillna python
Python
2 votes328 views2 answers
Print labels on confusion_matrix
Python
20 votes280 views6 answers
Sort list in python
Python
21 votes279 views4 answers
Floor function in python
Python
0 vote304 views1 answer
How to end a while loop python
Python
0 vote295 views1 answer
How to sort values in numpy by one column
Python
4 votes314 views1 answer
How to find the last item of a list
Python
0 vote280 views1 answer
__all__ python
Python
2 votes279 views1 answer
Embed variables python
Python