기본 콘텐츠로 건너뛰기

[python] decorator를 이용하여 함수 확장

python에서는 @라는 키워드를 이용해서 decorator라는 기능을 쓰면 함수의 기능을 확장하여 사용 할 수 있다.
python에서도 javascript와 동일하게 함수를 일급객체 취급을 한다. 즉 함수를 변수에 담아서 인자로 넘길 수 있다는 의미가 된다.

이제 decorator의 기능에 대해서 알아보자. 
함수를 다른 함수의 기능을 덧 붙일수 있고, 클래스의 기능을 덧 붙일수 있다.
첫번쨰로 함수의 기능을 덧 붙여보자.


1. function

p.py
1
2
3
4
5
6
7
8
9
10
11
12
13
def decorator_test(func):
    def new_func():
        print ("Begin")
        func()
        print ("End")
    return new_func
 
@decorator_test
def ppp():
    print('test')
 
 
ppp()
cs

p1.py
1
2
3
4
5
6
7
8
9
10
11
12
13
def decorator_test(func):
    def new_func():
        print ("Begin")
        func()
        print ("End")
    return new_func
 
# @decorator_test
def ppp():
    print('test')
 
test = decorator_test(ppp)
test()
cs

$python p.py

>>> Begin
test
End

$python p1.py

>>> Begin
test
End
두 코드를 실행을 해보면 실행결과가 똑같다. 
아마 decorator를 보다 클로저에 익숙한 개발자라면 p.py코드보단 p1.py가 더 익숙할 것이다.

python에서는 decorator를 사용하면 p.py를 p1.py와 같은 방식으로 코드가 동작한다.



2. class


p.py
1
2
3
4
5
6
7
8
9
10
11
12
13
class decorator_test(object):
    def __init__(self, func):
        self.func = func
 
    def __call__(self, *args, **kwargs):
        name = kwargs.get('name').lower()
        self.func(name)
 
@decorator_test
def good(name):
    print("good ", name)
 
good(name='mung')
cs

$python p.py

>>> good mung
위와 같이 실행이 된다.
클래스에서는 데코레이터를 사용하면 @[클래스] 해당 클래스를 아래에 있는 함수를 인자로 넘겨 init을 실행을 한다.
이후 해당 함수를 실행을 하면  클래스명으로 호출이 되어 __call__이 호출이 된다.


데코레이터는 인자도 넘길 수 있다.
p.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
def out(param1, param2):
    def decorator_test(func):
        def new_func():
            print ("Begin", param1)
            func()
            print ("End", param2)
        return new_func
    return decorator_test
@out(1,2)
def ppp():
    print('test')
 
 
ppp()
cs

$python p.py

>>> Begin 1
test
End2

데코레이터에서 인자를 넘기기 위해서는 parameter를 받는 함수로 감싸주어야 한다.


데코레이터로 이것저것 테스트 하다가 발견한 것인데

1
2
3
4
5
6
7
8
9
10
def decorator_test():
    def new_func(func):
        print ("Begin")
        func()
        print ("End")
    return new_func
 
@decorator_test()
def ppp():
    print('test')
cs
$python p.py

>>> Begin
test
End
위 코드에서는 어느 함수도 호출 하는 것이 없는데 바로 실행이 된다.



그럼 먼가 복잡해 보이는 데코레이터를 왜 이용을 할 까????

1
2
3
4
5
6
7
8
9
10
11
def process1():
    start = time.time()
    for i in range(10000):
        pass
    end = time.time()
    
def process2():
    start = time.time()
    for i in range(10000):
        pass
    end = time.time()
cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
def decorator_test(func):
 
    def new_func():
        start = time.time()
        func()
        end = time.time()
        print(end - start)
    return new_func
 
@decorator_test
def proccess1():
    for i in range(10000):
        pass
 
@decorator_test
def proccess2():
    for i in range(10000):
        pass
cs

위코드를 아래 처럼 바꿀 수 있다.

위코드와 다르게 시간은 찍어주는 코드를 재사용하여 사용하고 있다.

댓글

이 블로그의 인기 게시물

[javascript] 자스민(Jasmine)을 이용하여 TDD 작성후 웹 페이지에 적용해보자.

 다음 코드는  멍개의 코드 저장소 에 있습니다. 코드를 확인하고 싶으면 여기를 click 하세요 시작하기 전에... 책을 시작하기 전에 책에서 사용하는 자스민을 어떤식으로 이용하여 테스트 코드를 작성할 수 있는지 봐보자. 자스민 사용방법 테스트 러너파일 작성 테스트 러너란 자스민 코드와 소스파일, 스펙을 참조하는 html 파일 app_spec.html 테스트 코드를 작성하기 전에 첫째로 js 파일을 UI에서 완전히 분리하여 작성하는 것이 중요하다. 모듈패턴 테스트 코드가 가능한 코드를 작성하기 위해서는  모듈패턴 을 사용하도록 합니다. 모듈패턴이란? 함수로 데이터를 감추고, 모듈 API를 담고있는 객체를 반환하는 형태다. 모듈패턴은 자바스크립트에서 가장 많이 사용하는 패턴입니다. 두 가지 형태로 모듈패턴을 사용할 수 있습니다. 첫 번째는 호출하는 방식으로 사용합니다. 두 번째는 즉시 실행 함수 기반으로 사용할 수 있습니다. 여기서 즉시 실행 함수를  IIFE(Immediately Invoked Function Expression) 라고 합니다. 모듈 패턴 코드를 어떤식으로 작성할 수 있는지 간단하게 확인해보겠습니다. // 공간생성 let App = App || {}; // 생성된 공간에 함수를 추가한다. 인자로 함수를 넘긴다.( == 의존성 있는 함수를 주입) App.Person = function(God){ let name = God.makeName(); // API 노출 return { getName: function(){return name;}, setName: function(newName){name = newName;} } }; App 변수를 생성합니다. 생선된 변수에 Person() 함수를 만듭니다. person() 함수는 첫 번째 인자로 함수를 전달받습니다. retur...

[알고리즘] snake게임 알고리즘

막무가네로 알고리즘을 공부하면 재미가 없으니 게임을 접목하여 다루어 보도록 하겠습니다. 게임의 대상은 스네이크 게임입니다. 많은 사람들은 어릴 때 뱀게임을 많이 해봤을 것 입니다. 이번에 다뤄볼 주제는 뱀이 움직임을 어떻게 구현을 할지 알아보겠습니다. 뱀은 크게 3가지의 경우가 있습니다 1. 가장 중요한 뱀을 움직이기 2. 음식먹기 이때 뱀은 크기가 늘어나야 합니다. 3. 뱀이 움직이는 정책   - 뱀이 움직이지 못하는 경우는 : 우측방향에서 좌측 방향으로 OR 위에 아래 방향고 같이 180도 반전되는 움직임은 막겠습니다. 순수한 알고리즘을 만드는 과정이기 때문에 음식을 먹었는지 안먹었는지 판단하는 부분은 랜덤으로 판단을 하도록 하겠습니다. def is_eat(): return random.choice([1, 0]) 랜덤으로 1, 0을 반환을 해줍니다. 실제로 게임을 만든다면 해당 함수는 뱀의 머리가 음식의 좌표와 같은지 검사를 해주면 되겠습니다. key_position_map = { 'w': [-1, 0], # up 's': [1, 0], # down 'a': [0, -1], # left 'd': [0, 1] # right } direction = key_position_map.get('d') 다음으로는 키맵핑을 한 오브젝트 입니다. direction은 현재 뱀의 방향을 나타냅니다. snake_body = [[2, 3], [1, 3],[1, 2], [1, 1]] 주인공이 되는 뱀의 좌표들 입니다. while True: key = input() new_direction = key_position_map.get(key) if new_direction and direction_check(direction, new_direction): directi...

[react] react 작업환경 설정

react-boilerplate 사용방법 해당 프로젝트는 일반적인 react를 사용하기 전에 이런저런 환경설정을 미리 셋팅을 해둔 프로젝트 입니다. 해당 프로젝트는 아래의 설정들을 거쳐 해당 명령어만 실행을 해주면 일반적인 react 프로젝트를 실행 시킬 수 있습니다. 1. git clone https://github.com/pjt3591oo/react-boilerplate.git - 프로젝트 클론하기 2. npm install – Dependency 설치하기 3. npm start – 개발 서버 실행하기 (포트: 7777) 위 명령어를 실행 하였으면 http://localhost:7777로 접속해보기 해당 프로젝트의 자세한 설명은 아래에 작성하였습니다. 시작하기 앞서서 npm 3.10.6이상 버전 설치 권장 : npm의 버전마다 설치 방식이 달라 2.x버전에서 아래에 나와있는 모듈을 설치 할 때 상당히 오래 걸릴 수 있습니다. 1. Global Package 설치하기 babel  : 아직 ECMAScript6를 지원하지 않는 환경에서 ECMAScript6 Syntax를 사용할 수 있게 해줌. webpack  : 모듈 번들러로서, Browserify처럼 브라우저 위에서 import(require)을 할 수 있게 해주고 자바스크립트 파일들을 하나로 합쳐준다. webpack-dev-server  : webpack에서 지원하는 간단한 개발서버로서 별도의 서버를 구축하지 않고도 웹서버를 열 수 있으며 hot-=loader를 통하여 코드가 수정될때마다 자동으로 리로드 되게 할 수 있다. $ npm install -g babel webpack webpack-dev-server 2. project생성 $ mkdir react-tutorial && cd react-tutorial $ npm init 3. Dependency 및 Plugin 설치 react사용을 위한 ...