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