Python - come posso rimuovere dinamicamente un metodo da una classe - cioè opposto di setattr

Question

Non so se ho un buon design qui, ma ho una classe derivata da unittest.TestCase e il modo in cui io averlo impostato, il mio codice inserirà dinamicamente un gruppo di metodi test_* nella classe prima di richiamare l'unittest per eseguirlo. Io uso setattr per questo. Questo ha funzionato bene, ma ora ho una situazione in cui voglio rimuovere i metodi precedentemente iniettati e iniettare una nuova serie di metodi. Come posso rimuovere tutti i metodi in una classe i cui nomi corrispondono al modello test_*?

Answer 1

Si chiama delattr ed è documentato here.

Answer 2

>>> class Foo: 
    def func(self): 
     pass 
...  
>>> dir(Foo) 
['__doc__', '__module__', 'func'] 
>>> del Foo.func 
>>> dir(Foo) 
['__doc__', '__module__'] 

Answer 3

delattr() è quello che vuoi. Passare attraverso vars() della classe e verificare i nomi degli attributi che iniziano con "test_". Per esempio,

@classmethod 
def remove_test_methods(cls): 
    for name in list(vars(cls)): 
     if name.startswith("test_") and callable(getattr(cls, name)): 
      delattr(cls, name) 

vorrei consigliare di non usare dir(), in quanto questo vi mostrerà i nomi dei tuoi classi genitore e, in modo non tutti i nomi che si ottiene da dir() possono definita sulla classe che stai targeting.

Answer 4

In aggiunta alle risposte precedenti: se il metodo da cancellare è re-implementato, che avrebbe dovuto eliminare i metodi della classe sottoclasse e genitori, entrambi:

class A: 
    def method(self): 
     print("TestA") 

class B(A): 
    def method(self): 
     A.method(self) 
     print("TestB") 

instance = B() 
instance.method() # will call B.method() 
del B.method 
instance.method() # will now call A.method() 
del A.method 
instance.method() # will now raise AttributeError 

(o usare delattr di @ Risposta di BrenBarn)