diff options
author | Miguel Grinberg <miguel.grinberg@gmail.com> | 2017-08-11 15:29:16 -0700 |
---|---|---|
committer | Sergey Shepelev <temotor@gmail.com> | 2017-08-12 19:48:54 +0300 |
commit | b3037873144e5e33d754266cee38c8e61b000aa3 (patch) | |
tree | 6a75af8c4ed02407f77ed9c34f3a93d1378bc1ad | |
parent | f1b63abd6db186c978077499f9670600da599d1a (diff) | |
download | eventlet-b3037873144e5e33d754266cee38c8e61b000aa3.tar.gz |
socket: context manager support
Fixes #430
-rw-r--r-- | eventlet/greenio/base.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/eventlet/greenio/base.py b/eventlet/greenio/base.py index b36d834..e5c16bb 100644 --- a/eventlet/greenio/base.py +++ b/eventlet/greenio/base.py @@ -431,6 +431,12 @@ class GreenSocket(object): def gettimeout(self): return self._timeout + def __enter__(self): + return self + + def __exit__(self, *args): + self.close() + if "__pypy__" in sys.builtin_module_names: def _reuse(self): getattr(self.fd, '_sock', self.fd)._reuse() |