From f61109db84a673caeaaabf9e3c7faedfd37b2b34 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Thu, 17 Sep 2015 02:13:42 +0200 Subject: use new netlib exceptions --- libpathod/language/writer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'libpathod/language') diff --git a/libpathod/language/writer.py b/libpathod/language/writer.py index 5e211977..1a27e1ef 100644 --- a/libpathod/language/writer.py +++ b/libpathod/language/writer.py @@ -1,4 +1,5 @@ import time +from netlib.exceptions import TcpDisconnect import netlib.tcp BLOCKSIZE = 1024 @@ -62,5 +63,5 @@ def write_values(fp, vals, actions, sofar=0, blocksize=BLOCKSIZE): return True elif a[1] == "inject": send_chunk(fp, a[2], blocksize, 0, len(a[2])) - except netlib.tcp.NetLibDisconnect: # pragma: no cover + except TcpDisconnect: # pragma: no cover return True -- cgit v1.2.3