Kelder/rubygoldberg-tracker
32
0
Fork 0
Code Issues 4 Pull requests Projects Releases Packages Wiki Activity

added checks to make sure runs and indexes are numbers #5

Browse source
This commit is contained in:
Hannes Klinckaert 2019-10-02 11:29:01 +02:00
parent 37c8e13167
commit a00d93b632
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
watcher/app.py
View file

@ -13,7 +13,7 @@ from flask_cors import CORS
sio = socketio.Server() sio = socketio.Server()
app = Flask(__name__) app = Flask(__name__)
cors = CORS(app) CORS(app)
app.wsgi_app = socketio.WSGIApp(sio, app.wsgi_app) app.wsgi_app = socketio.WSGIApp(sio, app.wsgi_app)
# We request users to send the run number to prevent accidentially getting old requests from a previous run # We request users to send the run number to prevent accidentially getting old requests from a previous run
@ -38,6 +38,9 @@ def index():
@app.route("/start_run/<run_index>") @app.route("/start_run/<run_index>")
def start_run(run_index): def start_run(run_index):
if not run_index.isdigit():
return f'{run_index} is not a number'
db["current_run"] = run_index db["current_run"] = run_index
starttime = time() starttime = time()
if run_index in db["run_data"]: if run_index in db["run_data"]:
@ -50,6 +53,9 @@ def start_run(run_index):
@app.route("/link/start/<run>/<link_index>") @app.route("/link/start/<run>/<link_index>")
def link_start(run, link_index): def link_start(run, link_index):
if not run.isdigit() or not link_index.isdigit():
return f'{run} and/or {link_index} is not a number'
start = time() start = time()
request_data = live_request(run, link_index) request_data = live_request(run, link_index)
sio.emit('live_request', request_data) sio.emit('live_request', request_data)
@ -72,6 +78,9 @@ def link_start(run, link_index):
@app.route("/link/handoff/<run>/<index>") @app.route("/link/handoff/<run>/<index>")
def link_handoff(run, index): def link_handoff(run, index):
if not run.isdigit() or not index.isdigit():
return f'{run} and/or {index} is not a number'
if db["current_run"] != run: if db["current_run"] != run:
return "Wrong run number, check that you updated you run", 404 return "Wrong run number, check that you updated you run", 404
else: else: