def handle_goal():
if request.method == "GET":
goals = Goal.query.all()
response = []
if not goals:
return jsonify(response), 200
for goal in goals:
response.append({"id": goal.goal_id, "title": goal.title})
return jsonify(response), 200
elif request.method == "POST":
request_body = request.get_json()
if request_body:
new_goal = Goal(title=request_body["title"])
db.session.add(new_goal)
db.session.commit()
response = {"goal": new_goal.json_response()}
return jsonify(response), 201
else:
response = {"details": "Invalid data"}
return jsonify(response), 400
def create_goals():
request_body = request.get_json()
if "title" not in request_body.keys():
return make_response({"details": "Invalid data"}, 400)
goal = Goal(title=request_body["title"])
db.session.add(goal)
db.session.commit()
return {"goal": goal.json_response()}, 201
