from functools import wraps
import time

def timefn(fn):
    """計算性能的修飾器"""
    @wraps(fn)
    def measure_time(*args, **kwargs):
        t1 = time.perf_counter()
        result = fn(*args, **kwargs)
        t2 = time.perf_counter()
        print(f"{fn.__name__}:{t2 - t1:.2f}s")
        return result

    return measure_time

# 在定義函數前進行裝飾
@timefn
def test():
    s = 0
    for i in range(1000000):
        s += 1
    return s


test()