Python：ライブラリ内のすべての関数をラップする

Question

私たちは、別の社内チームが提供するライブラリを使用します。

from externalTeam import dataCreator 
datacreator.createPizza() 
datacreator.createBurger() 
datacreator.createHotDog() 

最近、特定の状況で実行するために、1つの方法が1分以上かかることがわかりました。これをデバッグするには、このメソッドの呼び出しのたびにコードに入り、タイムアウトを追加する必要がありました。後知恵で

import time 
from externalTeam import dataCreator 
start = time.clock() 
datacreator.createPizza() 
stop = time.clock() 
print "It took %s seconds to perform createPizza" % (str(stop-start)) 

我々はすべての場所でcreatePizzaを呼んでいる、と私たちはcreatePizza自体（アナロジーはここで少しを打破するために開始された）をコントロールしていないので、それはです。むしろ1つの場所でcreatePizzaを呼び出し、その周りにタイマーを追加できるようにしたいと思います。これを達成するための私の最初の考えは、自分のラッパークラスですべてのメソッドをラップすることです。それはしかしDRYの反対だ、といつでも彼らは私もあることをラップするために私たちのライブラリを更新する必要があると思います別のメソッドを追加します。私が欲しいもの

import time 
from externalTeam import dataCreator 
def createPizza(self): 
    start = time.clock() 
    datacreator.createPizza() 
    stop = time.clock() 
    print "It took %s seconds to perform createPizza" % (str(stop-start)) 

def createBurger(self): 
    start = time.clock() 
    datacreator.createPizza() 
    stop = time.clock() 
    print "It took %s seconds to perform createBurger" % (str(stop-start)) 

def createHotDog(self): 
    start = time.clock() 
    datacreator.createPizza() 
    stop = time.clock() 
    print "It took %s seconds to perform createHotDog" % (str(stop-start))  

は常に周りのすべての数行のコードを実行するための方法ですdataCreatorから呼び出されている関数です。そのメソッドを動的に定義することができる、またはむしろ定義されていない中間クラスを介して、それを行うにはいくつかの方法が必要です。

Answer 1

私はこのように動作しますdataCreatorアダプタクラスを作成します。

1. デバッグ/タイミング機能にラップする必要がdataCreatorからのメソッドのmethods2wrapリストを持っています。
2. dataCreatorメソッドに1：1をマップし、methods2wrapのメソッドをタイミングデバッグメッセージにラップするオーバーライドされた__getattribute__()を持っています。

プルーフオブコンセプトコード（例えば、クラスlistをラップし、その方法append周りデバッグタイムスタンプを挿入します）。この例で構築し、初期化時に、あなたがラップするとどのような方法が計時されるべきものをクラスに設定できるように、それはパラメトリック作ることができる。もちろん、

import time 

class wrapper(list): 

    def __getattribute__(self, name): 
     TO_OVERRIDE = ['append'] 
     if name in TO_OVERRIDE: 
      start = time.clock() 
     ret = super(list, self).__getattribute__(name) 
     if name in TO_OVERRIDE: 
      stop = time.clock() 
      print "It took %s seconds to perform %s" % (str(stop-start), name) 
     return ret 

profiled_list = wrapper('abc') 
print profiled_list 
profiled_list.append('d') 
print profiled_list 
profiled_list.pop() 
print profiled_list 

...

EDITは：あることに注意してくださいTO_OVERRIDE__getattribute__コールごとに再割り当てされます。これは設計によるものです。クラス属性として作成する場合は、__getattribute__は再帰的にループします（親の__getattribute__メソッドへの明示的な呼び出しを使用して取得する必要がありますが、リストをゼロから再構築するよりも時間がかかるでしょう）。

HTH

Answer 2

Pythonコードをプロファイリングする場合は、手動で行うのではなく、Pythonの組み込みのprofiling librariesを使用する必要があります。

Answer 3

なぜ引数を呼び出す単なるラッパー関数ではないのですか？

def wrapper(func, *args, **kwargs): 
    ... timing logic ... 
    response = func(*args, **kwargs) 
    ... more timing logic 
    return response 

とそれを呼び出す：あなたは関数自体を渡す

wrapper(datacreator.createPizza, arg1, arg2, kwarg1=kwarg) 

ノート、それを呼び出さずに。