[250103] 데코레이터란?

데코레이터란? ✨

---

1. 데코레이터 정의

• 데코레이터는 기존의 함수나 메서드에 기능을 추가하거나 수정할 수 있는 파이썬의 고급 기능이다
• 다른 함수를 입력으로 받아 실행 결과를 반환하는 고차 함수(Higher-Order Function)의 일종임
• 주로 함수나 메서드의 전후에 실행될 로직을 추가할 때 사용한다

---

2. 데코레이터의 기본 구조

• 기본 함수

def my_function():
    print("Hello, World!")

• 데코레이터 함수 정의

def my_decorator(func):
    def wrapper():
        print("Before the function call")
        func()  # 원래 함수 실행
        print("After the function call")
    return wrapper

• 함수에 데코레이터 적용

@my_decorator
def my_function():
    print("Hello, World!")

• 실행 결과

my_function()
# 출력:
# Before the function call
# Hello, World!
# After the function call

---

3. 데코레이터의 주요 특징

1. 함수를 입력으로 받음
   데코레이터는 원래의 함수를 매개변수로 전달받아 감싸는 래퍼(wrapper) 함수에 기능을 추가하는 것이다
2. 래핑(Wrapping)
   원래 함수의 동작을 감싸서 확장하거나 변경할 수 있음
3. `@` 구문
   데코레이터를 사용할 때 `@decorator_name`을 함수 정의 바로 위에 추가해준다

---

4. 데코레이터의 응용 예시

1. 함수 실행 시간 측정

import time

def timer_decorator(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"Function {func.__name__} took {end_time - start_time:.2f} seconds")
        return result
    return wrapper

@timer_decorator
def slow_function():
    time.sleep(2)
    print("Function complete")

slow_function()
# 출력:
# Function complete
# Function slow_function took 2.00 seconds

2. 권한 확인

def requires_permission(permission):
    def decorator(func):
        def wrapper(user, *args, **kwargs):
            if user.get("permission") == permission:
                return func(user, *args, **kwargs)
            else:
                print("Permission denied!")
        return wrapper
    return decorator

@requires_permission("admin")
def admin_task(user):
    print("Performing admin task")

user1 = {"name": "Alice", "permission": "admin"}
user2 = {"name": "Bob", "permission": "user"}

admin_task(user1)  # 출력: Performing admin task
admin_task(user2)  # 출력: Permission denied!

3. 로깅 추가

def log_decorator(func):
    def wrapper(*args, **kwargs):
        print(f"Calling function: {func.__name__} with arguments {args} and {kwargs}")
        result = func(*args, **kwargs)
        print(f"Function {func.__name__} returned {result}")
        return result
    return wrapper

@log_decorator
def add(a, b):
    return a + b

add(3, 4)
# 출력:
# Calling function: add with arguments (3, 4) and {}
# Function add returned 7

---

5. 데코레이터의 매개변수 처리

• 데코레이터가 매개변수를 받을 수 있는 경우에는 3단계 중첩 함수를 사용함

def repeat(n):  # 매개변수를 받는 데코레이터
    def decorator(func):
        def wrapper(*args, **kwargs):
            for _ in range(n):
                func(*args, **kwargs)
        return wrapper
    return decorator

@repeat(3)
def say_hello():
    print("Hello!")

say_hello()
# 출력:
# Hello!
# Hello!
# Hello!​

---

6. 클래스에 데코레이터 사용

• 클래스 메서드 데코레이터

def uppercase_decorator(func):
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        return result.upper()
    return wrapper

class Greeting:
    @uppercase_decorator
    def say_hello(self):
        return "hello, world"

g = Greeting()
print(g.say_hello())  # 출력: HELLO, WORLD

• 정적 메서드 데코레이터

class Math:
    @staticmethod
    @log_decorator
    def add(a, b):
        return a + b

Math.add(5, 7)
# 출력:
# Calling function: add with arguments (5, 7) and {}
# Function add returned 12

---

7. 데코레이터 체이닝

• 여러 데코레이터를 동시에 사용할 수 있음
• 위에서 아래로 순차적으로 적용됨

def decorator1(func):
    def wrapper(*args, **kwargs):
        print("Decorator 1")
        return func(*args, **kwargs)
    return wrapper

def decorator2(func):
    def wrapper(*args, **kwargs):
        print("Decorator 2")
        return func(*args, **kwargs)
    return wrapper

@decorator1
@decorator2
def my_function():
    print("Original function")

my_function()
# 출력:
# Decorator 1
# Decorator 2
# Original function​

---

정리 🧹

• 데코레이터는 코드 재사용성과 유지보수를 높여주는 역할을 한다~
• 사용 예시:
  • 로깅, 권한 검증, 실행 시간 측정, 데이터 검증 등•••
• 함수와 클래스에 모두 적용 가능하며 다양한 유틸리티 함수를 구현할 때 유용하다~