diff options
| author | Aldo Cortesi <aldo@nullcube.com> | 2013-01-25 15:54:41 +1300 |
|---|---|---|
| committer | Aldo Cortesi <aldo@nullcube.com> | 2013-01-25 15:54:41 +1300 |
| commit | 2eb6651e5180035cd3e17f9048b16ea38719a9ac (patch) | |
| tree | d9d95ca726bbacb775b55112882e1ef5123996c4 /netlib/tcp.py | |
| parent | 7248a22d5e381dd57d69c06f8e67e60fd55e55ba (diff) | |
| download | mitmproxy-2eb6651e5180035cd3e17f9048b16ea38719a9ac.tar.gz mitmproxy-2eb6651e5180035cd3e17f9048b16ea38719a9ac.tar.bz2 mitmproxy-2eb6651e5180035cd3e17f9048b16ea38719a9ac.zip | |
Extract TCP test utilities into netlib.test
Diffstat (limited to 'netlib/tcp.py')
| -rw-r--r-- | netlib/tcp.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/netlib/tcp.py b/netlib/tcp.py index d0ca09f3..56cc0dea 100644 --- a/netlib/tcp.py +++ b/netlib/tcp.py @@ -1,4 +1,4 @@ -import select, socket, threading, traceback, sys, time +import select, socket, threading, sys, time, traceback from OpenSSL import SSL import certutils @@ -84,13 +84,14 @@ class _FileLike: def reset_timestamps(self): self.first_byte_timestamp = None + class Writer(_FileLike): def flush(self): - try: - if hasattr(self.o, "flush"): + if hasattr(self.o, "flush"): + try: self.o.flush() - except socket.error, v: - raise NetLibDisconnect(str(v)) + except socket.error, v: + raise NetLibDisconnect(str(v)) def write(self, v): if v: |