Function decorator

Function decorator

def decorator_function(original_function):
    def wrapper_function(*arge, **kwargs):
        print('wrapper executed this before {}'.format(original_function.__name__))
        return original_function(*arge, **kwargs)
    return wrapper_function

@decorator_function
def display():
    print('display function ran')
@decorator_function
def display_info(name, age):
    print('display_info ran with arguments ({}, {})'.format(name,age))
display_info('John', 25)
display()

# display = decorator_function(display) equal to @decorator_function

Example: log file

def my_logger(orig_func):
    import logging
    logging.basicConfig(filename='{}.log'.format(orig_func.__name__), level=logging.INFO)

    def wrapper(*args, **kwargs):
        logging.info(
            'Ran with args:{}, and kwargs: {}'.format(args, kwargs))
        return orig_func(*args, **kwargs)
    return wrapper
    
@my_logger
def display_info(name, age):
    print('display_info ran with arguments ({}, {})'.format(name,age))
display_info('John', 25)

Example: timer

def my_timer(orig_func):
    import time

    def wrapper(*args, **kwargs):
        t1 = time.time()
        result = orig_func(*args, **kwargs)
        t2 = time.time() - t1
        print('{} ran in: {} sec'.format(orig_func.__name__,t2))
        return result
    return wrapper

import time

@my_timer
def display_info(name, age):
    time.sleep(1)
    print('display_info ran with arguments ({}, {})'.format(name, age))

Class decorator

class ddecorator_class(object):

    def __init__(self, original_function):
        self.original_function = original_function

    def __call__(self, *args, **kwargs):
        print('call method executed this before {}'.format(self.original_function.__name__))
        return self.original_function(*args, **kwargs)


@ddecorator_class
def display():
    print('display function ran')
@ddecorator_class
def display_info(name, age):
    print('display_info ran with arguments ({}, {})'.format(name,age))
display_info('John', 25)
display()