Python Enclosing作用域、閉包、裝飾器的基礎篇,請看Python Enclosing作用域、閉包、裝飾器話聊上篇
Jaglawz: 我經常看到有人的裝飾器是帶引數的,這又是咋回事呢?
Pylego: 這個其實很簡單的,你還記得上次我說:
1 2 3 4 5 6 7 8 9 10 11 |
@deco def foo(): pass # 相當於: foo = deco(foo) # 那麼 @new_deco(*args, **kwargs) def bar(): pass # 相當於: bar = new_deco(*args, **kwargs)(bar) |
Jaglawz: 也就是說,new_deco返回的是一個裝飾器函式,然後再去裝飾其他函式。那類裝飾器又是怎麼回事呢?
Pylego: 你知道Python的物件可以像函式一樣呼叫嗎?
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from hashlib import sha256 class HashCache(object): def __init__(self): self.cache = {} def __call__(self, string): if string not in self.cache: self.cache[string] = sha256(string).hexdigest() return self.cache[string] hc = HashCahce() hc('foo') # 像函式一樣呼叫hc物件 hc('foo') hc('bar') |
Jaglawz: 如果是這樣的話我就明白了。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
class FibCache(object): def __init__(self, func): self.func = func self.cache = {} def __call__(self, n): if n not in self.cache: self.cache[n] = self.func(n) return self.cache[n] @FibCache def fib(n): if n == 0: return 0 elif n == 1: return 1 else: return fib(n-1) + fib(n-2) # 相當於: fib = FibCache(fib) # fib(10)相當於FibCache(fib)(10) # 裝飾後的fib是FibCache的一個物件而已 # 也就是說作為裝飾器的類的構造方法要接收一個待裝飾的函式,然後__call__函式的引數要和待裝飾的函式的引數是一樣的(除了self),這樣的類就可以用來裝飾函式了 |
Pylego: 你這個裝飾器厲害,還給fibnacci函式加了快取,佩服!
Jaglawz: 我不會告訴你我是看了大神的傑作然後嚇嚇唬嚇唬你的!