기본 콘텐츠로 건너뛰기

[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

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

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

댓글

댓글 쓰기

이 블로그의 인기 게시물

[node.js] 파일 리더기 만들기 - 사용 모듈 정리, pdf 구조, hwp 구조

pdf -> html 변환 가장 무난하다. 기본적으로 pdf는 htm와 비슷한 형태의 구조를 가지고 있다. 크게 header , body , xref table , trailer 의 구조로 되어있다. pdf는 환경에 상관없이 표현을 하기 위한 목적을 가지고 있는 파일이다. 이런 이유 때문에 무난히 진행이 된 것 같다. pdf2htmlex와 pdftohtmljs라는 모듈을 이용을 했다. var pdftohtml = require ( 'pdftohtmljs' ) ; var converter = new pdftohtml ( 'test.pdf' , "sample.html" ) ; converter . convert ( 'ipad' ) . then ( function ( ) { console . log ( "Success" ) ; } ) . catch ( function ( err ) { console . error ( "Conversion error: " + err ) ; } ) ; 이미지나, text같은 것들이 거의 100%로 변환이 된다. docx -> html 변환 docx파일을 html파일로 변환을 할 때는 style 적용과 한글이 깨지는 문제가 있다. 텍스트들을 전부 잘 읽기는 하는데 스타일 정보를 제대로 가져오지 못하기 때문에 좀 애매하다 - Headings. - Lists. - Customisable mapping from your own docx styles to HTML. For instance, you could convert WarningHeading to h1.warning by providing an appropriate style ...
댓글 쓰기
자세한 내용 보기

[css] css로 프린트 방지하기

웹에서 프린트 제어가 불가능 한 줄 알았는데 프린트 클릭 시  스크립트로 해당 이벤트를 받아올 수 있다. 하지만 스크립트를 사용하는 방법은 브라우저마다 작동을 하지 않을 수 있다. 좀 더 찾아보니 css로 인쇄되는 영역을 제어를 해줄 수 있다.  @media print 를 이용하면 된다. < html > < head > < title > print test page < / title > < style > @media print { . np { display : none ; } } < / style > < / head > < body > < div class = "np" > test < / div > < div > test1 < / div > < div > test1 < / div > < div > test1 < / div > < / body > < / html > 위 코드를 보면 np 클래스를 @media print에 넣어주었다. @media print는 인쇄됐을 때의 스타일을 지정을 해주는 것이다.  위에서는 해당 페이지를 인쇄할 때 p를 display : none으로 가려주었다. @media print를 이용하면 좀 더 멋진 인쇄물을 만들 수 ...
댓글 쓰기
자세한 내용 보기

[알고리즘] 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...
댓글 쓰기
자세한 내용 보기