從另一個庫覆蓋函數的返回值

Question

我正在使用一個函數庫，它在循環中調用函數some_func。

def main(): 
    xs = [] 
    for x in y: 
     xs.append(some_func(x)) 
    return xs 

def some_func(x): 
    return some_data 

我想攔截some_func函數的響應，並用我的值覆蓋它。例如：

def some_func(x): 
    return my_func(some_data) 

這樣做的目的是讓我可以運行一些進度鉤來跟蹤完成該過程所需的時間。

如何在不修改原始庫的源代碼的情況下執行此操作？

Answer 1

您可以使用裝飾器，但不是將其應用於@，而是直接包裝目標函數。這樣的事情：

>>> import math 
>>> def wrap_function(func): 
...  def wrapper(x): 
...   print('Called {0} with {1}'.format(func.__name__, x)) 
...   result = func(x) 
...   print('Returned value is {0!r}'.format(result)) 
...   return result 
...  return wrapper 
... 
>>> math.sin = wrap_function(math.sin) 
>>> r = math.sin(1.1337) 
Called sin with 1.1337 
Returned value is 0.905984636371863 
>>> r 
0.905984636371863 

鑑於你現在可以做的東西之前/之後的通話，你可以實現時間測量等等。

Answer 2

如果我正確理解你，裝飾者是你應該看看。以下是Andreas Jung的摘錄。這允許你在你選擇的任何函數上添加@timeit裝飾器。

import time             

def timeit(method): 

    def timed(*args, **kw): 
     ts = time.time() 
     result = method(*args, **kw) 
     te = time.time() 

     print '%r (%r, %r) %2.2f sec' % \ 
       (method.__name__, args, kw, te-ts) 
     return result 

    return timed 

class Foo(object): 

    @timeit 
    def foo(self, a=2, b=3): 
     time.sleep(0.2) 

@timeit 
def f1(): 
    time.sleep(1) 
    print 'f1' 

@timeit 
def f2(a): 
    time.sleep(2) 
    print 'f2',a 

@timeit 
def f3(a, *args, **kw): 
    time.sleep(0.3) 
    print 'f3', args, kw 

f1() 
f2(42) 
f3(42, 43, foo=2) 
Foo().foo()