from flask import Flask, request, jsonify
app = Flask(__name__)
votes = {"option1": 0, "option2": 0}
@app.route('/vote', methods=['POST'])
def vote():
data = request.get_json()
option = data['option']
if option in votes:
votes[option] += 1
return jsonify({"status": "success", "votes": votes})
else:
return jsonify({"status": "failure", "message": "Invalid option"})
if __name__ == '__main__':
app.run(debug=True)
]]>
from functools import wraps
def login_required(f):
@wraps(f)
def decorated_function(*args, **kwargs):
if not session.get('logged_in'):
return jsonify({"status": "failure", "message": "Login required"}), 401
return f(*args, **kwargs)
return decorated_function
]]>