-
Notifications
You must be signed in to change notification settings - Fork 60
Expand file tree
/
Copy pathapp.py
More file actions
77 lines (62 loc) · 2.05 KB
/
Copy pathapp.py
File metadata and controls
77 lines (62 loc) · 2.05 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from flask import Flask, jsonify, request
from flask_sqlalchemy import SQLAlchemy
from dotenv import load_dotenv
import os
# Load environment variables from .env file
load_dotenv()
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = os.getenv('DATABASE_URL')
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)
class User(db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80), nullable=False)
def __init__(self, name):
self.name = name
# Create the database tables
@app.before_request
def create_tables():
db.create_all()
# Home route
@app.route('/', methods=['GET'])
def home():
return jsonify({"message": "Welcome to the User Management API!"}), 200
# GET all users
@app.route('/users', methods=['GET'])
def get_users():
users = User.query.all()
return jsonify([{'id': user.id, 'name': user.name} for user in users])
# POST a new user
@app.route('/users', methods=['POST'])
def add_user():
name = request.json.get('name')
if not name:
return jsonify({"error": "Name is required."}), 400
user = User(name=name)
db.session.add(user)
db.session.commit()
return jsonify({"message": f"User {name} added.", "id": user.id}), 201
# PUT to update a user
@app.route('/users/<int:id>', methods=['PUT'])
def update_user(id):
user = User.query.get(id)
if user is None:
return jsonify({"error": "User not found."}), 404
name = request.json.get('name')
if name:
user.name = name
db.session.commit()
return jsonify({"message": f"User {id} updated."})
return jsonify({"error": "Name is required."}), 400
# DELETE a user
@app.route('/users/<int:id>', methods=['DELETE'])
def delete_user(id):
user = User.query.get(id)
if user is None:
return jsonify({"error": "User not found."}), 404
db.session.delete(user)
db.session.commit()
return jsonify({"message": f"User {id} deleted."})
if __name__ == "__main__":
app.run(host="0.0.0.0", port=5000)