def get_next():
"""Generates a new number with a key and returns the information."""
try:
app.logger.info("GET request to get_next action")
data = request.get_json(force=True)
if "key" not in data:
app.logger.info(f"No key present in request body")
return jsonify({
"success": False,
"message": "No key present in request body"
}), 401
key = data["key"]
if os.getenv("KEY") != key:
app.logger.info(f"Invalid key {key} for retrieving queue number")
return jsonify({
"success": False,
"message": "Invalid key"
}), 401
app.logger.info(f"Generating number...")
counter = Counter.get_counter()
number_key = generate_random_key()
number = Number(counter.get(), number_key)
# Save Number in the database
db.session.add(number)
db.session.commit()
# Increment counter
counter.increment()
db.session.commit()
app.logger.info(
f"Number {number.position} generated with key {number_key}"
)
return jsonify({
"success": True,
"number": number.position,
"key": number_key
}), 200
except Exception as err:
app.logger.error(err)
return jsonify({"success": False}), 500
def check_number(position):
"""Checks if number and plain key match."""
try:
app.logger.info("POST request to check_number action")
data = request.get_json(force=True)
if "key" not in data:
app.logger.info(f"No key present in request body")
return jsonify({
"success": False,
"message": "No key present in request body"
}), 401
key = data["key"]
if os.getenv("KEY") != key:
app.logger.info(f"Invalid key {key} for retrieving queue number")
return jsonify({
"success": False,
"message": "Invalid key"
}), 401
if set(["key", "plain_key"]) != set(data.keys()):
app.logger.info(f"Invalid request body keys {data.keys()}")
return jsonify({
"success": False,
"message": "Invalid request body"
}), 400
counter = Counter.get_counter()
if position >= counter.get():
# Number has not been emmited yet
app.logger.info(
f"Number {position} tried to be accessed. Last number created "
f"was {counter.get() - 1}"
)
return jsonify({
"success": False,
"message": "Invalid number"
}), 404
plain_key = data["plain_key"]
app.logger.info(f"Verifying number {position} with key {plain_key}...")
number = Number.query.filter_by(position=position).first()
if not number.check_plain_key(plain_key):
app.logger.info(
f"Number {position} tried to be accessed with incorrect key "
f"{plain_key}"
)
return jsonify({
"success": False,
"message": "Invalid number key"
}), 401
app.logger.info(
f"Number {position} verified correctly with key {plain_key}"
)
return jsonify({"success": True}), 200
except Exception as err:
app.logger.error(err)
return jsonify({"success": False}), 500
