CODEWITHSUNDEEP
pythonsetattr
tadasajon
tadasajon

Reputation: 14866

Python - how can I dynamically remove a method from a class -- i.e. opposite of setattr

I don't know if I have a good design here, but I have a class that is derived from unittest.TestCase and the way I have it set up, my code will dynamically inject a bunch of test_* methods into the class before invoking unittest to run through it. I use setattr for this. This has been working well, but now I have a situation in which I want to remove the methods I previously injected and inject a new set of methods. How can I remove all the methods in a class whose names match the pattern test_*?

Upvotes: 23

Views: 23259

Answers (4)

DomTomCat
DomTomCat

Reputation: 8569

As an addition to the previous answers: If the method to be deleted is re-implemented, you'd have to delete the subclass and parent class methods, both:

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

(Or use delattr of @BrenBarn's answer)

Upvotes: 2

kindall
kindall

Reputation: 184280

delattr() is what you want. Loop through vars() of the class and test for attribute names starting with "test_". E.g.,

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

I'd advise against using dir(), as this will show you names from your parent classes as well, so not all the names you get from dir() may defined on the class you're targeting.

Upvotes: 1

Ashwini Chaudhary
Ashwini Chaudhary

Reputation: 251061

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

Upvotes: 7

BrenBarn
BrenBarn

Reputation: 251448

It's called delattr and is documented here.

Upvotes: 30

Related Questions