[Python]Decorator

데코레이터(@)

어색하고 반복적인 함수의 표현을 줄이기 위해 제안
함수 외에도 클래스, 제너레이터 등 다양한 타입에서도 사용
데코레이터 -> 말 그대로 함수를 '꾸며주는' 혹은 '장식하는', '포장하는' 기능

참고:pep318

함수의 데코레이터(@)

예시부터 보겠습니다.

def one():
    print("Hello")
    print("one이라는 함수")

def two():
    print("Hello")
    print("two라는 함수")

def three():
    print("Hello")
    print("three라는 함수")

위의 예시는 아주 간단하기에 감이 안올 수 있지만, 공통적으로 들어간 print("Hello")를 한번에 묶어서 쓸 수 있습니다.

def Hello_deco(func): # func는 실행할 함수
    def Hello(): # 실행할 함수를 감싸는(wrap) 함수
        print("Hello")
        func()
    return Hello # Hello 함수를 리턴

@Hello_deco
def one():
    print("one이라는 함수")

@Hello_deco
def two():
    print("two라는 함수")

@Hello_deco
def three():
    print("three라는 함수")

위와 같이 사용할 수 있습니다. (지금은 짧지만, 코드가 훨씬 길어졌다고 가정하면 데코레이터의 편리함을 느낄 수 있습니다!)

Hello_deco(one())

# Hello
# one이라는 함수

데코레이터 사용하는 이유

DRY(Don't Repeat Yourself)

파이썬 개발자들이 지키려하는 원칙 중 코드의 반복 사용을 줄이기 위한 DRY 원칙
이 원칙을 지키기 위해 파이썬 개발자들은 상속, 디스크립터, 제너레이터 등 다양한 기술을 사용
특히 데코레이터는 시각적으로 다른 기술들보다 깔끔하고 간결한 코드를 만들어주기에 많이 추천됨

실제 어떤 프로그래밍을 하는 경우가 아니라면 잘 사용안하더라도 패키지를 이용할 때 자주 접할 수 있습니다.

@property 데코레이터

예시)

class test:
    def __init__(self, first, last):
        self.first = first
        self.last = last
        self.full = first + ' ' + last

name = test('Jay', 'Lee')

print(name.first) #-> 'Jay' 출력
print(name.last) #-> 'Lee' 출력
print(name.full) #-> 'Jay Lee' 출력

name.last = 'Den'
print(name.last) #-> 'Den' 출력
print(name.full) #-> 'Jay Lee' 출력

위의 경우, name.last를 바꿔주어도 이미 class를 선언할 때, Jay와 Lee를 받기 때문에 name.full을 프린트해도 Den이 반영된 값이 아닌, 그대로 Jay Lee가 출력됩니다. 이를 반영해주기 위해 아래와 같이 코드를 변경합니다.

class test_1:
    def __init__(self, first, last):
        self.first = first
        self.last = last

    def full(self):
        return self.first + ' ' + self.last

name = test_1('Jay', 'Lee')

위와 같이 full()에 대한 함수를 따로 정의하여 반영해줄 수 있습니다.
다만, 이 때는 name.full()로 호출하여 메소드형식을 취해야합니다. 이를 특성 형태로 바꾸기 위해 property 데코레이터가 사용됩니다.

class test_1:
    def __init__(self, first, last):
        self.first = first
        self.last = last

    @property
    def full(self):
        return self.first + ' ' + self.last

name = test_1('Jay', 'Lee')

위와 같은 경우엔, name.full로 호출하면 값이 리턴됩니다.

getter, setter 데코레이터

위의 property가 getter(가져올 때 사용하는 데코레이터)의 역할을 했으므로 setter(값을 설정해주는 데코레이터)를 알아보겠습니다.

class test_1:
    def __init__(self, first, last):
        self.first = first
        self.last = last
    @property
    def full(self):
        return self.first + ' ' + self.last

    @full.setter
    def full(self, new_full):
        first, last = new_full.split()
        self.first = first
        self.last = last

name1 = test_1('Jay', 'Lee')

print(name1.first) #-> 'Jay' 출력
print(name1.last) #-> 'Lee' 출력
print(name1.full) #-> 'Jay Lee' 출력
name1.full = 'Hyo Lee' #-> name1.full에 'Hye Lee'을 지정하게 되면
print(name1.first) #-> 'Hyo' 출력
print(name1.last) #-> 'Lee' 출력

위와 같이, 이번엔 full을 변경하였을 때, first 및 last 값이 변경되는 것을 확인할 수 있습니다.