import time
from functools import wraps

def timeit(func):
    @wraps(func)
    def measure_time(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print("@timefn: {} took {} seconds.".format(func.__name__, end_time - start_time))
        return result
    return measure_time

@timeit
def func(a, b):
    c = a + b
    return c

print(func)

if __name__ == "__main__":
    func(10, 20)