Python에서 데코레이터 함수를 사용하는 방법

Python에서 데코레이터 기능을 사용하는 방법

Python 프로그래밍에서 데코레이터는 매우 유용한 도구입니다. 이를 통해 원래 함수 코드를 수정하지 않고도 추가 기능으로 함수를 확장할 수 있습니다. 데코레이터 함수는 로깅, 타이밍, 권한 확인 등과 같은 함수 실행 전후에 일부 작업을 자동으로 수행할 수 있습니다. 이 기사에서는 데코레이터 함수의 기본 개념을 소개하고 몇 가지 구체적인 코드 예제를 제공합니다.

1. 데코레이터 함수의 정의

데코레이터 함수는 함수를 매개변수로 받아들이고 새로운 함수를 반환하는 함수입니다. @ 구문을 사용하는 함수에 적용할 수 있습니다. 다음은 간단한 데코레이터 함수 정의의 예입니다.

def decorator_function(original_function):
    def wrapper_function(*args, **kwargs):
        # 在函数执行前的操作
        print("Before executing the function")
        # 执行原始函数
        result = original_function(*args, **kwargs)
        # 在函数执行后的操作
        print("After executing the function")
        return result
    return wrapper_function

2. 데코레이터 함수 적용

데코레이터 함수는 함수 실행 전후에 몇 가지 추가 작업을 수행할 수 있습니다. 다음은 데코레이터 함수를 사용하여 타이밍 함수를 구현하는 예입니다.

import time

def timer_decorator(original_function):
    def wrapper_function(*args, **kwargs):
        start_time = time.time()
        result = original_function(*args, **kwargs)
        end_time = time.time()
        execution_time = end_time - start_time
        print(f"Execution time: {execution_time} seconds")
        return result
    return wrapper_function

@timer_decorator
def my_function():
    time.sleep(1)
    print("Function executed")

my_function()

위 예에서는 my_function 함수에 타이머_데코레이터 데코레이터 함수가 적용되었습니다. my_function이 호출되면 데코레이터는 함수가 실행되기 전의 시작 시간을 기록하고, 함수가 실행된 후의 종료 시간을 계산하고, 함수의 실행 시간을 계산합니다.

3. 데코레이터 함수의 매개변수

데코레이터 함수는 다른 데코레이터를 다른 함수에 적용하기 위해 매개변수를 허용할 수 있습니다. 다음은 매개변수가 포함된 데코레이터 함수의 예입니다.

def prefix_decorator(prefix):
    def decorator_function(original_function):
        def wrapper_function(*args, **kwargs):
            print(f"{prefix}: Before executing the function")
            result = original_function(*args, **kwargs)
            print(f"{prefix}: After executing the function")
            return result
        return wrapper_function
    return decorator_function

@prefix_decorator("LOG")
def my_function():
    print("Function executed")

my_function()

위의 예에서 prefix_ decorator 함수는 함수 실행 전후에 접두사와 함께 로그 정보를 출력하는 데코레이터 함수를 반환합니다.

4. 다중 데코레이터 적용

동일한 함수에 여러 데코레이터 기능을 적용하여 데코레이터의 스택 효과를 형성할 수 있습니다. 다음은 여러 개의 데코레이터를 적용하는 예입니다.

def decorator1(original_function):
    def wrapper_function(*args, **kwargs):
        print("Decorator 1: Before executing the function")
        result = original_function(*args, **kwargs)
        print("Decorator 1: After executing the function")
        return result
    return wrapper_function

def decorator2(original_function):
    def wrapper_function(*args, **kwargs):
        print("Decorator 2: Before executing the function")
        result = original_function(*args, **kwargs)
        print("Decorator 2: After executing the function")
        return result
    return wrapper_function

@decorator1
@decorator2
def my_function():
    print("Function executed")

my_function()

위의 예에서는 decorator1 및 decorator2 데코레이터 함수가 my_function 함수에 순차적으로 적용됩니다. my_function이 호출되면 데코레이터 2가 데코레이터 1 다음에 실행됩니다.

요약:

Decorator 함수는 Python에서 매우 유용한 도구로, 원래 함수 코드를 수정하지 않고도 추가 함수로 함수를 확장할 수 있습니다. 이 기사는 몇 가지 구체적인 코드 예제를 제공함으로써 독자가 데코레이터 함수의 기본 개념과 사용 방법을 이해하는 데 도움이 되기를 바랍니다. 데코레이터 함수를 사용하면 코드 중복이 줄어들고 코드가 더욱 모듈화되고 유지 관리가 쉬워집니다.

위 내용은 Python에서 데코레이터 함수를 사용하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!