diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2016-06-03 12:06:36 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2016-06-03 12:06:36 +1200 |
commit | e60860e65d06d2b4452b7ea94902d79eed11d78c (patch) | |
tree | 91a635bef6421248fa63a26255b52a6e96f98631 /pathod/pathoc.py | |
parent | 36383a6146d605879bf8c2834370b09f42bfec06 (diff) | |
download | mitmproxy-e60860e65d06d2b4452b7ea94902d79eed11d78c.tar.gz mitmproxy-e60860e65d06d2b4452b7ea94902d79eed11d78c.tar.bz2 mitmproxy-e60860e65d06d2b4452b7ea94902d79eed11d78c.zip |
Make tcp.Client.connect return a context manager that closes the connection
Diffstat (limited to 'pathod/pathoc.py')
-rw-r--r-- | pathod/pathoc.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/pathod/pathoc.py b/pathod/pathoc.py index 2b7d053c..5cfb4591 100644 --- a/pathod/pathoc.py +++ b/pathod/pathoc.py @@ -286,7 +286,7 @@ class Pathoc(tcp.TCPClient): if self.use_http2 and not self.ssl: raise NotImplementedError("HTTP2 without SSL is not supported.") - tcp.TCPClient.connect(self) + ret = tcp.TCPClient.connect(self) if connect_to: self.http_connect(connect_to) @@ -324,6 +324,7 @@ class Pathoc(tcp.TCPClient): if self.timeout: self.settimeout(self.timeout) + return ret def stop(self): if self.ws_framereader: |