About/Python

[Python] 함수 실행 시간 측정 (Python decorator 활용)

김징어 2021. 7. 19. 18:09

decorator 함수 정의

def logging_time(original_fn):
    import time
    from functools import wraps

    @wraps(original_fn)
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = original_fn(*args, **kwargs)

        end_time = time.time()
        print("WorkingTime[{}]: {} sec".format(original_fn.__name__, end_time - start_time))
        return result
    return wrapper

logging_time() 함수 사용

@logging_time
def func():
    ...
    함수 내용
    ...
    

실행결과

WorkingTime[func]: 2.7772340774536133 sec