Pythonセクション/デコレータ

参考記事

デコレータ

デコレータ

Javaを使用したことのある人なら誰でも、JavaのAOPが非常に便利であることを知っています。Javaは、ログの印刷など、メソッドの呼び出しの前後にいくつかの操作を実行できます。Pythonにも同様の関数があり、それがデコレータです。例として印刷ログを見てみましょう。

ログを印刷する前に

def buy(good_name):
    print("购买商品 %s " % good_name)


def payment(money: int):
    print("支付金额 %d " % money)


if __name__ == "__main__":
    buy("10斤苹果")
    payment(30)

演算結果：

购买商品 10斤苹果 
支付金额 30 

関数呼び出しの前にログを自動的に印刷します

ログ印刷機能

import functools

# 在函数调用前自动打印日志
def log(func):
    @functools.wraps(func)
    def wrapper(*args, **kw):
        print("call  function: %s args: %s kw: %s " % (func.__name__, args, kw))
        return func(*args, **kw)

    return wrapper

呼び出し元の関数にコメントを追加します

@log
def buy(good_name):
    print("购买商品 %s " % good_name)


@log
def payment(money: int):
    print("支付金额 %d " % money)


if __name__ == "__main__":
    buy("10斤苹果")
    payment(30)

演算結果：

call  function: buy args: ('10斤苹果',) kw: {} 
购买商品 10斤苹果 
call  function: payment args: (30,) kw: {} 
支付金额 30 

関数呼び出しの前後にログを出力する

# 在函数调用前后自动打印日志
def log(func):
    @functools.wraps(func)
    def wrapper(*args, **kw):
        print("start call  function: %s args: %s kw: %s " % (func.__name__, args, kw))
        result = func(*args, **kw)
        print("end call  function: %s args: %s kw: %s " % (func.__name__, args, kw))
        return result

    return wrapper

演算結果：

start call  function: buy args: ('10斤苹果',) kw: {} 
购买商品 10斤苹果 
end call  function: buy args: ('10斤苹果',) kw: {} 
start call  function: payment args: (30,) kw: {} 
支付金额 30 
end call  function: payment args: (30,) kw: {} 

関数呼び出しの前後にパラメーターをカスタマイズし、ログを出力します

def log(description="", version="1.0"):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kw):
            print("start %s version=%s . call function: %s args: %s kw: %s " % (
                description, version, func.__name__, args, kw))
            result = func(*args, **kw)
            print("end %s version=%s . call function: %s args: %s kw: %s " % (
                description, version, func.__name__, args, kw))
            return result

        return wrapper

    return decorator

@log(description="购买商品", version="2.2")
def buy(good_name):
    print("购买商品 %s " % good_name)


@log(description="支付金额")
def payment(money: int):
    print("支付金额 %d " % money)


if __name__ == "__main__":
    buy("10斤苹果")
    payment(30)

演算結果：

start 购买商品 version=2.2 . call function: buy args: ('10斤苹果',) kw: {} 
购买商品 10斤苹果 
end 购买商品 version=2.2 . call function: buy args: ('10斤苹果',) kw: {} 
start 支付金额 version=1.0 . call function: payment args: (30,) kw: {} 
支付金额 30 
end 支付金额 version=1.0 . call function: payment args: (30,) kw: {}