from flask import Flask, jsonify, request
app = Flask(__name__)
# Sample data
users = [
{"id": 1, "name": "John Doe"},
{"id": 2, "name": "Jane Doe"}
]
# GET endpoint - Fetch all users
@app.route('/users', methods=['GET'])
def get_users():
return jsonify(users)
# GET endpoint - Fetch a user by ID
@app.route('/users/', methods=['GET'])
def get_user(user_id):
user = next((u for u in users if u["id"] == user_id), None)
return jsonify(user) if user else jsonify({"error": "User not found"}), 404
# POST endpoint - Add a new user
@app.route('/users', methods=['POST'])
def add_user():
new_user = request.json # Get data from request body
users.append(new_user)
return jsonify({"message": "User added successfully", "user": new_user}), 201
if __name__ == '__main__':
app.run(debug=True)