diff options
Diffstat (limited to 'pathod/protocols/websockets.py')
-rw-r--r-- | pathod/protocols/websockets.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/pathod/protocols/websockets.py b/pathod/protocols/websockets.py index df83461a..00ae5aa8 100644 --- a/pathod/protocols/websockets.py +++ b/pathod/protocols/websockets.py @@ -1,8 +1,8 @@ import time -from netlib import websockets -from .. import language -from netlib.exceptions import NetlibException +from mitmproxy.net import websockets +from pathod import language +from mitmproxy import exceptions class WebsocketsProtocol: @@ -16,7 +16,7 @@ class WebsocketsProtocol: started = time.time() try: frm = websockets.Frame.from_file(self.pathod_handler.rfile) - except NetlibException as e: + except exceptions.NetlibException as e: lg("Error reading websocket frame: %s" % e) return None, None ended = time.time() |