From 9755c38eb3884dc5b698d2f4360abc062d545e86 Mon Sep 17 00:00:00 2001 From: Roman Zeyde Date: Wed, 17 Jun 2015 16:51:42 +0300 Subject: [PATCH] trezor: Client should be a context manager --- sshagent/trezor.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/sshagent/trezor.py b/sshagent/trezor.py index 80ad512..ba3f732 100644 --- a/sshagent/trezor.py +++ b/sshagent/trezor.py @@ -39,7 +39,11 @@ class Client(object): log.debug('version : %s', '.'.join([str(v) for v in version])) log.debug('revision : %s', binascii.hexlify(f.revision)) - def close(self): + def __enter__(self): + return self + + def __exit__(self, *args): + log.info('disconnected from Trezor') self.client.close() def get_public_key(self, label):