Pythonデコレータ(修飾子)

デコレータの役割

Python関数デコレータ@の機能は、既存の関数に関数を追加することです。これらの関数は、ログの挿入、パフォーマンステスト、トランザクション処理などによく使用されます。

デコレータ処理ロジック

インタプリタが@のようなデコレータを読み取ると、最初に@の後のコンテンツを解析し、@の次の行の関数またはクラスを@の背後にある関数のパラメータとして使用してから、で変更された関数オブジェクトに戻り値を割り当てます。次の行

関数修飾子を作成するためのルール

(1)修飾子は関数です
(2)修飾子は変更された関数をパラメーターとして受け取ります
(3)修飾子は新しい関数を返します
(4)修飾子は維持された関数のシグネチャを維持します

例えば

例1:変更された関数はパラメーターを取りません

def log(func):
    def wrapper():
        print('log开始 ...')
        func()
        print('log结束 ...')
    return wrapper
    
@log
def test():
    print('test ..')

test()

演算結果:

log开始 ...
test ..
log结束 ...

例2:変更された関数はパラメーターを取ります

from functools import wraps

def log(func):
    @wraps(func)
    def wrapper(*args,**kwargs):
        print('log开始 ...',func.__name__)
        ret = func(*args,**kwargs)
        print('log结束 ...')
        return ret
    return wrapper
    
@log
def test1(s):
    print('test1 ..', s)
    return s

@log
def test2(s1, s2):
    print('test2 ..', s1, s2)
    return s1 + s2


test1('a')
test2('a','bc')

演算結果:

log开始 ... test1
test1 .. a
log结束 ...
log开始 ... test2
test2 .. a bc
log结束 ...

例3:修飾子にはパラメーターがあり、上記の例以外にラップする必要があります

from functools import wraps

def log(arg):    
    def _log(func):
        @wraps(func)
        def wrapper(*args,**kwargs):
            print('log开始 ...',func.__name__, arg)            
            ret = func(*args,**kwargs)
            print('log结束 ...')
            return ret
        return wrapper
    return _log
 
@log('module1')
def test1(s):
    print('test1 ..', s)
    return s

@log('module1')
def test2(s1, s2):
    print('test2 ..', s1, s2)
    return s1 + s2


test1('a')
test2('a','bc')

演算結果:

log开始 ... test1 module1
test1 .. a
log结束 ...
log开始 ... test2 module1
test2 .. a bc
log结束 ...

おすすめ

転載: blog.csdn.net/baidu_24752135/article/details/114636000