2

@wrapt デコレータを使用した条件付きモンキー パッチ

Wraptには2つの特徴があります

  • デコレータのベスト プラクティス
  • モンキーパッチ

呼び出されたメソッドを条件付きで変更できるように、両方を行っています。パンダのバージョンに基づく例を提供しました。これは機能しますが、私はベアボーン モンキー パッチを適用し、wrapt 機能を使用しませんでした。代わりの方法にモンキー パッチを適用するには、どのように wrapt を使用できますか?

import pandas as pd
import wrapt, inspect
from distutils.version import StrictVersion
def alt_impl(alt_fn, cond):
    @wrapt.decorator
    def wrapper(wrapped, instance, args, kwargs):
        nonlocal alt_fn
        if cond():
            # @staticmethod and @classmethod need to navigate to actual function
            if not inspect.isfunction(alt_fn):
                alt_fn = alt_fn.__func__
            # class instance methods need to be bound to class instance
            if instance is not None:
                alt_fn = alt_fn.__get__(instance, instance.__class__)
            return alt_fn(*args, **kwargs)
        else:
            return wrapped(*args, **kwargs)
    return wrapper

使用例

class alt_impl_example():
    def upgraded_pandas(self, args):
        print(f"upgraded {pd.__version__}") 

    @alt_impl(upgraded_pandas, lambda: StrictVersion(pd.__version__) >= StrictVersion("1.0.0")) 
    def pandas(self, args): 
        print(pd.__version__) 

t = alt_impl_example()
t.pandas({})
4

0 に答える 0