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

Support arbitrary length for the randbytes method

Browse source
This commit is contained in:
Andika Wasisto 2020-08-12 03:42:19 +07:00
parent a299141f9a
commit 9cf1a177cc
2 changed files with 15 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
quanttp/__main__.py
View file

@ -60,8 +60,6 @@ def api_randbytes():
length = int(request.args.get('length'))
if length < 1:
return Response('length must be greater than 0', status=400, content_type='text/plain')
elif length > 8192:
return Response('length must be less than or equal to 8192', status=400, content_type='text/plain')
qng_wrapper.clear()
return Response(qng_wrapper.randbytes(length), content_type='application/octet-stream')
except ValueError as e:
@ -89,7 +87,7 @@ def handle_ws_message(message, websocket, subscribed):
websocket.send(str(qng_wrapper.randnormal()))
elif split_message[0] == 'RANDBYTES':
length = int(split_message[1])
if not 1 <= length <= 8192:
if length < 1:
raise ValueError()
qng_wrapper.clear()
websocket.send(qng_wrapper.randbytes(length))
@ -113,7 +111,7 @@ def handle_ws_message(message, websocket, subscribed):
websocket.send(str(qng_wrapper.randnormal()))
elif split_message[0] == 'SUBSCRIBEBYTES':
chunk = int(split_message[1])
if not 1 <= chunk <= 8192:
if chunk < 1:
raise ValueError()
qng_wrapper.clear()
if not subscribed[0]: