refactor(api): Refactor error handling in add() and consume() functions
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Ching 2024-03-03 20:16:24 +08:00
parent 80050f1874
commit f7e90567d7

17
app.py
View File

@ -193,11 +193,16 @@ def add():
spider = BarcodeSpider(x_rapidapi_key=X_RapidAPI_Key) spider = BarcodeSpider(x_rapidapi_key=X_RapidAPI_Key)
good = spider.get_good(barcode) good = spider.get_good(barcode)
if add_product(good, location): try:
response_data = {"message": "New item added successfully"} if add_product(good, location):
return jsonify(response_data), 200 response_data = {"message": "New item added successfully"}
else: return jsonify(response_data), 200
response_data = {"message": "Fail to add new item"} else:
response_data = {"message": "Fail to add new item"}
return jsonify(response_data), 400
except Exception as e:
error_message = str(e)
response_data = {"message": error_message}
return jsonify(response_data), 400 return jsonify(response_data), 400
@ -211,7 +216,7 @@ def consume():
return jsonify(response_data), 200 return jsonify(response_data), 200
except Exception as e: except Exception as e:
error_message = str(e) error_message = str(e)
response_data = {"error": error_message} response_data = {"message": error_message}
return jsonify(response_data), 400 return jsonify(response_data), 400