如何裝飾班級中的所有方法？我能裝飾課堂嗎？

Question

我有幾個類，他們有相同的實現名稱，但不同的實現。我想在一些類中修飾所有方法，但其他類不是。我曾想過繼承，但有些類有一些方法不需要進行修飾。問題是我不想一個接一個地裝飾方法，有些類需要用同一個裝飾器來裝飾，有沒有解決方法來解決它？

Answer 1

你可以啓動需要一些前綴裝飾，然後用這樣的方法都：

class Xobject(object): 

    def __init__(self, decorator): 
     for method_name in dir(self): 
      if method_name.startswith("dec_"): 
       attr = getattr(self, method_name) 
       wrapped = decorator(attr) 
       setattr(self, method_name, wrapped) 

    def dec_me_1(self): 
     print("In dec_me1") 
     return 0 

    def dec_me_2(self): 
     print("In dec_me2") 
     return 1 


def decorator(func): 

    def wrapped(*args): 
     print("TEST") 
     return func(*args) 

    return wrapped 


x = Xobject(decorator) 

x.dec_me_1() 
x.dec_me_2() 

UPDATE：

您可以通過以下功能均裝飾類。在使用Python時，您應該知道Python中的class也是對象，因此您可以將其更改並傳遞給其他函數。

def decorator(func): 

    def wrapped(*args): 
     print("TEST") 
     return func(*args) 

    return wrapped 


def decorate_object(p_object, decorator): 
    for method_name in dir(p_object): 
     if method_name.startswith("dec_"): 
      attr = getattr(p_object, method_name) 
      wrapped = decorator(attr) 
      setattr(p_object, method_name, wrapped) 

decorate_object(Xobject, decorator) 

x = Xobject() 

x.dec_me_1() 
x.dec_me_2() 

而且你可以裝飾已經實例化對象相同的方法：

x = Xobject() 

x.dec_me_1() 
x.dec_me_2() 

decorate_object(x, decorator) 

x.dec_me_1() 
x.dec_me_2() 

Answer 2

我敢肯定，有幾個方法可供選擇，但主要領導選項有：

• 創建一個自定義元類，其中__new__方法遍歷屬性字典，標識方法並對其進行裝飾。有關Python元類編程的示例，請參閱http://eli.thegreenplace.net/2011/08/14/python-metaclasses-by-example/。缺點：這可能比我們想要進入的更復雜。
• 在普通班級的__init__方法中做同樣的事情。缺點：只裝飾實例方法而不裝入類或靜態方法，而且速度較慢，因爲每次創建新實例時都會運行該方法。

• 做它的類外：

  class Foo(object): 
      def bar(self): 
      print 'bar' 
  
  for name, ref in vars(Foo): 
      if callable(ref): ... 
  

  缺點：你只有一次機會做正確的事：在導入時。子類不會被修改。

• 在類級裝飾器中執行此操作。與在課堂以外一樣的缺點（我認爲）。

Answer 3

在某些時候，你必須明確什麼被包裹，什麼不包含。

如果我理解正確的話，你，我想你可以做這樣的事情：

def wrapper(func): 
    def inner(*args, **kwargs): 
     print "%s was called" func.__name__ 
     return func(*args, **kwargs) 
    return inner 

class A(object): 
    def foo(self): 
     print "foo called" 
    def bar(self): 
     print "BAR CALLED" 

class B(A): 
    @wrapper 
    def foo(self): 
     super(B, self).foo() 

class C(A): 
    @wrapper 
    def bar(self): 
     super(C, self).bar() 

Stick = A() 
Dave = B() 
Jupiter = C() 

Jupiter.foo() #prints "foo called" 
Jupiter.bar() #prints "bar wrapped" and "BAR CALLED"