randogoth/quanttp2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Better error handling

Browse source
This commit is contained in:
Andika Wasisto 2020-09-07 20:56:42 +07:00
parent 915f679105
commit 5e9747eabe
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
quanttp/__main__.py
View file

@ -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))