如何创建一个可以使用或不使用参数的Python装饰器？

import functools

def my_decorator(func=None, *, arg1=None, arg2=None):
    def decorator_wrapper(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            print(f"Before function: {arg1=}, {arg2=}")
            result = func(*args, **kwargs)
            print(f"After function: {arg1=}, {arg2=}")
            return result
        return wrapper

    if func is None:
        return decorator_wrapper
    else:
        return decorator_wrapper(func)

# 使用装饰器，不使用参数
@my_decorator
def my_function():
    print("Inside my_function")

# 使用装饰器，并使用参数
@my_decorator(arg1="value1", arg2="value2")
def my_function_with_args():
    print("Inside my_function_with_args")

my_function()
my_function_with_args()

Before function: arg1=None, arg2=None
Inside my_function
After function: arg1=None, arg2=None
Before function: arg1=value1, arg2=value2
Inside my_function_with_args
After function: arg1=value1, arg2=value2