Decorating one method of a class C with another method of class C?

Chris Angelico rosuav at gmail.com
Sun Jun 8 14:21:14 EDT 2014


On Mon, Jun 9, 2014 at 4:04 AM, Dan Stromberg <drsalists at gmail.com> wrote:
> I have a class that's operating on a socket.
>
> I'd like to have simple operations on that socket like "list
> configured hosts", "allow connection to host", etc.  And I'd like them
> to be decorated with "reconnected_to_server_if_needed".
>
> I'll probably end up putting a try/except in each simple operation,
> making them a bit less simple.

Can you have the decorator outside the class, calling a regular method
to do its main work?

def recon_if_needed(f):
    def inner(self, *a, **kw):
        if self.disconnected: self.connect()
        return f(self, *a, **kw)
    return inner
class SocketWorker:
    @recon_if_needed
    def send_packet(self, packet):
        assert not self.disconnected

ChrisA



More information about the Python-list mailing list