Как перехватить функцию в Python?

def decorator(func):
    def wrapper(*args, **kwargs):
        # Действия до вызова функции
        result = func(*args, **kwargs)
        # Действия после вызова функции
        return result
    return wrapper

@decorator
def my_func(arg):
    print(arg)

my_func(""Hello"")

import sys

def my_func(arg):
    print(arg)

def wrapper(*args, **kwargs):
    # Действия до вызова функции
    result = my_func(*args, **kwargs)
    # Действия после вызова функции
    return result

sys.modules[__name__].my_func = wrapper

my_func(""Hello"")