Python decorator @

Python 에는 함수를 꾸며줄 수 있는 decorator라는 기능이 존재한다.

Decorator은 @ 기호를 사용하여, 함수의 앞과 뒤에 추가적인 내용을 더해준다.

 

처음에 봤을 땐 @가 뭔가 싶어서 Python @라고 검색했더니 검색결과가 안나와서 당황했다..

 

1) nested function의 형태와, 2) decorator class 두 가지 형태로 사용할 수 있다.

 

1) nested function

- decorator 함수를 정의해준다 (test_decorator).

- decorator 함수는 input으로 함수를 받고, 앞뒤로 취해줄 action을 정의한 후 해당 함수를 호출해 주면 된다.

- decorator을 사용할 함수를 정의하기 전 decorator 기호 @와 함께 decorator 함수의 이름을 적어 주면 된다.

def test_decorator(func):
    def wrapper(*args, **kwargs):
        print('---function starts---')
        func(*args, **kwargs)
        print('---function ends---')
    return wrapper

@test_decorator
def example_f(input_str):
	print(input_str)

example_f('hello')
>> ---function starts---
hello
---function ends---

 

2) decorator class

- decorator class를 정의해준다 (Test_decorator).

- __init__()에서 input으로 함수를 받고,

- __call__()에서 마찬가지로 취해줄 action을 정의한 후 해당 함수를 호출해 주면 된다.

- 마찬가지로 decorator을 사용할 함수 위에 @ 기호와 함께 decorator class 이름을 적어 주면 같은 작용을 한다.

class Test_decorator:
    def __init__(self, func):
        self.func = func
    def __call__(self, *args, **kwargs):
        print('---function starts---')
        func(*args, **kwargs)
        print('---function ends---')

@Test_decorator
def example_f(input_str):
    print(input_str)

example_f('hello')
>> ---function starts---
hello
---function ends---