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

Clean code

Browse source
This commit is contained in:
Andika Wasisto 2020-05-25 12:56:13 +07:00
parent 4106510a25
commit 85196980d7
1 changed files with 14 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

27
quanttp/__main__.py
View file

@ -25,13 +25,14 @@ from flask_sockets import Sockets
from gevent import pywsgi from gevent import pywsgi
from geventwebsocket.handler import WebSocketHandler from geventwebsocket.handler import WebSocketHandler
from quanttp.data.qng_wrapper import QngWrapper
from quanttp.data.qng_wrapper_mock import QngWrapperMock
app = Flask(__name__) app = Flask(__name__)
sockets = Sockets(app) sockets = Sockets(app)
from quanttp.data.qng_wrapper import QngWrapper
qng_wrapper = QngWrapper() qng_wrapper = QngWrapper()
# from quanttp.data.qng_wrapper_mock import QngWrapperMock
# qng_wrapper = QngWrapperMock() # qng_wrapper = QngWrapperMock()
@ -72,46 +73,46 @@ def ws(websocket):
def handle_ws_message(message, websocket, subscribed): def handle_ws_message(message, websocket, subscribed):
try: try:
split_message = message.strip().lower().split() split_message = message.strip().upper().split()
if split_message[0] == 'randint32': if split_message[0] == 'RANDINT32':
qng_wrapper.clear() qng_wrapper.clear()
websocket.send(str(qng_wrapper.randint32())) websocket.send(str(qng_wrapper.randint32()))
elif split_message[0] == 'randuniform': elif split_message[0] == 'RANDUNIFORM':
qng_wrapper.clear() qng_wrapper.clear()
websocket.send(str(qng_wrapper.randuniform())) websocket.send(str(qng_wrapper.randuniform()))
elif split_message[0] == 'randnormal': elif split_message[0] == 'RANDNORMAL':
qng_wrapper.clear() qng_wrapper.clear()
websocket.send(str(qng_wrapper.randnormal())) websocket.send(str(qng_wrapper.randnormal()))
elif split_message[0] == 'randbytes': elif split_message[0] == 'RANDBYTES':
length = int(split_message[1]) length = int(split_message[1])
qng_wrapper.clear() qng_wrapper.clear()
websocket.send(qng_wrapper.randbytes(length)) websocket.send(qng_wrapper.randbytes(length))
elif split_message[0] == 'subscribeint32': elif split_message[0] == 'SUBSCRIBEINT32':
if not subscribed[0]: if not subscribed[0]:
subscribed[0] = True subscribed[0] = True
qng_wrapper.clear() qng_wrapper.clear()
while subscribed[0] and not websocket.closed: while subscribed[0] and not websocket.closed:
websocket.send(str(qng_wrapper.randint32())) websocket.send(str(qng_wrapper.randint32()))
elif split_message[0] == 'subscribeuniform': elif split_message[0] == 'SUBSCRIBEUNIFORM':
if not subscribed[0]: if not subscribed[0]:
subscribed[0] = True subscribed[0] = True
qng_wrapper.clear() qng_wrapper.clear()
while subscribed[0] and not websocket.closed: while subscribed[0] and not websocket.closed:
websocket.send(str(qng_wrapper.randuniform())) websocket.send(str(qng_wrapper.randuniform()))
elif split_message[0] == 'subscribenormal': elif split_message[0] == 'SUBSCRIBENORMAL':
if not subscribed[0]: if not subscribed[0]:
subscribed[0] = True subscribed[0] = True
qng_wrapper.clear() qng_wrapper.clear()
while subscribed[0] and not websocket.closed: while subscribed[0] and not websocket.closed:
websocket.send(str(qng_wrapper.randnormal())) websocket.send(str(qng_wrapper.randnormal()))
elif split_message[0] == 'subscribebytes': elif split_message[0] == 'SUBSCRIBEBYTES':
chunk = int(split_message[1]) chunk = int(split_message[1])
qng_wrapper.clear() qng_wrapper.clear()
if not subscribed[0]: if not subscribed[0]:
subscribed[0] = True subscribed[0] = True
while subscribed[0] and not websocket.closed: while subscribed[0] and not websocket.closed:
websocket.send(qng_wrapper.randbytes(chunk)) websocket.send(qng_wrapper.randbytes(chunk))
elif split_message[0] == 'unsubscribe': elif split_message[0] == 'UNSUBSCRIBE':
subscribed[0] = False subscribed[0] = False
websocket.send('UNSUBSCRIBED') websocket.send('UNSUBSCRIBED')
except (ValueError, BlockingIOError): except (ValueError, BlockingIOError):