Decorators example

decorators.py

def my_decortator(func):
    def wrapper(*args):
        print("Run before func")
        func(*args)
        print("Run after func")

    return wrapper


def decorator_with_tags(tags):
    def decorator(func):
        def wrapper(name):
            if tags == 'div':
                return "<div>{}</div>".format(func(name))
            elif tags == 'p':
                return "<p>{}</p>".format(func(name))
            else:
                return "<h1>{}</h1>".format(func(name))

        return wrapper

    return decorator


@my_decortator
def test_func(name):
    print("Hello {}".format(name))


@decorator_with_tags("div")
def test_tagged_decorator(name):
    return name


test_func("World!")
print(test_tagged_decorator("My Name"))