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

CLI arguments for servername and port number

Browse source
This commit is contained in:
kflux 2020-09-18 15:22:26 +02:00
parent 242d44178c
commit 337f765eed
1 changed files with 181 additions and 158 deletions
Download patch file Download diff file Expand all files Collapse all files

33
quanttp/__main__.py
View file

@ -20,6 +20,7 @@
# SOFTWARE. # SOFTWARE.
import os import os
import sys
import threading import threading
import json import json
@ -36,6 +37,25 @@ sockets = Sockets(app)
qng_wrapper = QngWrapperWindows() if (os.name == 'nt') else QngWrapperLinux() qng_wrapper = QngWrapperWindows() if (os.name == 'nt') else QngWrapperLinux()
def main():
# Commandline Arguments (servername, port)
argNo = len(sys.argv) - 1
if argNo < 2:
print("--------------------------------------------")
print("Please provide arguments: <servername> <port>")
print("--------------------------------------------")
else:
servername = sys.argv[1]
port = int(sys.argv[2])
print("----------------------------------------------------------------------------------------")
print("Serving Entropy as pod \"", servername, "\" on http://localhost:", port, "/api/...", sep='')
print("----------------------------------------------------------------------------------------")
serve(servername, port)
def serve(servername, port):
# Original API ---------------------------------------------- # Original API ----------------------------------------------
@app.route('/api/randint32') @app.route('/api/randint32')
@ -83,7 +103,7 @@ def randjsonint32():
int32array = [] int32array = []
for x in range(0, length): for x in range(0, length):
int32array.append(qng_wrapper.randint32()) int32array.append(qng_wrapper.randint32())
return Response(json.dumps({"type": "string", "format": "int32", "length":length, "data": int32array, "success": "true"}), content_type='text/plain') return Response(json.dumps({"server" : servername, "type": "string", "format": "int32", "length":length, "data": int32array, "success": "true"}), content_type='text/plain')
except (TypeError, ValueError) as e: except (TypeError, ValueError) as e:
return Response(json.dumps({"error": str(e), "success":"false"}), status=400, content_type='text/plain') return Response(json.dumps({"error": str(e), "success":"false"}), status=400, content_type='text/plain')
@ -96,7 +116,7 @@ def randjsonuniform():
uniformarray = [] uniformarray = []
for x in range(0, length): for x in range(0, length):
uniformarray.append(qng_wrapper.randuniform()) uniformarray.append(qng_wrapper.randuniform())
return Response(json.dumps({"type": "string", "format": "uniform", "length":length, "data": uniformarray, "success": "true"}), content_type='text/plain') return Response(json.dumps({"server" : servername, "type": "string", "format": "uniform", "length":length, "data": uniformarray, "success": "true"}), content_type='text/plain')
except (TypeError, ValueError) as e: except (TypeError, ValueError) as e:
return Response(json.dumps({"error": str(e), "success":"false"}), status=400, content_type='text/plain') return Response(json.dumps({"error": str(e), "success":"false"}), status=400, content_type='text/plain')
@ -109,7 +129,7 @@ def randjsonnormal():
normarray = [] normarray = []
for x in range(0, length): for x in range(0, length):
normarray.append(qng_wrapper.randnormal()) normarray.append(qng_wrapper.randnormal())
return Response(json.dumps({"type": "string", "format": "normal", "length":length, "data": normarray, "success": "true"}), content_type='text/plain') return Response(json.dumps({"server" : servername, "type": "string", "format": "normal", "length":length, "data": normarray, "success": "true"}), content_type='text/plain')
except (TypeError, ValueError) as e: except (TypeError, ValueError) as e:
return Response(json.dumps({"error": str(e), "success":"false"}), status=400, content_type='text/plain') return Response(json.dumps({"error": str(e), "success":"false"}), status=400, content_type='text/plain')
@ -125,7 +145,7 @@ def randjsonhex():
hexarray = [] hexarray = []
for x in range(0, length): for x in range(0, length):
hexarray.append(qng_wrapper.randbytes(size).hex()) hexarray.append(qng_wrapper.randbytes(size).hex())
return Response(json.dumps({"type": "string", "format": "hex", "length":length, "size": size, "data": hexarray, "success": "true"}), content_type='text/plain') return Response(json.dumps({"server" : servername, "type": "string", "format": "hex", "length":length, "size": size, "data": hexarray, "success": "true"}), content_type='text/plain')
except (TypeError, ValueError) as e: except (TypeError, ValueError) as e:
return Response(json.dumps({"error": str(e), "success":"false"}), status=400, content_type='text/plain') return Response(json.dumps({"error": str(e), "success":"false"}), status=400, content_type='text/plain')
@ -201,5 +221,8 @@ def handle_ws_message(message, websocket, subscribed):
def handle_exception(e): def handle_exception(e):
return Response(e.description, status=e.code, content_type='text/plain') return Response(e.description, status=e.code, content_type='text/plain')
server = pywsgi.WSGIServer(('0.0.0.0', 62456), application=app, handler_class=WebSocketHandler) server = pywsgi.WSGIServer(('0.0.0.0', port), application=app, handler_class=WebSocketHandler)
server.serve_forever() server.serve_forever()
if __name__ == "__main__":
main()