From 5e9747eabed904e5ed6f991e1d76868d71c18eb4 Mon Sep 17 00:00:00 2001 From: Andika Wasisto Date: Mon, 7 Sep 2020 20:56:42 +0700 Subject: [PATCH] Better error handling --- quanttp/__main__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/quanttp/__main__.py b/quanttp/__main__.py index f6c8d6a..7744d7e 100644 --- a/quanttp/__main__.py +++ b/quanttp/__main__.py @@ -58,7 +58,7 @@ def randbytes(): if length < 1: return Response('length must be greater than 0', status=400, content_type='text/plain') return Response(qng_wrapper.randbytes(length), content_type='application/octet-stream') - except ValueError as e: + except (TypeError, ValueError) as e: return Response(str(e), status=400, content_type='text/plain') @@ -117,7 +117,7 @@ def handle_ws_message(message, websocket, subscribed): websocket.send('UNSUBSCRIBED') elif split_message[0] == 'CLEAR': qng_wrapper.clear() - except (ValueError, BlockingIOError): + except (IndexError, ValueError, BlockingIOError): pass except Exception as e: websocket.close(code=1011, message=str(e))