Maple - Stephanie & Ayaka

app/__init__.py

@@ -12,6 +12,7 @@ def create_app(test_config=None):
     app = Flask(__name__)
     app.url_map.strict_slashes = False
     app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
+    app.config["SQLALCHEMY_ECHO"] = True
 
     if test_config is None:
         app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get(
@@ -21,7 +22,7 @@ def create_app(test_config=None):
         app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get(
             "SQLALCHEMY_TEST_DATABASE_URI")
 
-    
+
     # import models for Alembic Setup
     from app.models.customer import Customer
     from app.models.video import Video
@@ -32,5 +33,9 @@ def create_app(test_config=None):
     migrate.init_app(app, db)
 
     #Register Blueprints Here
+    from .routes import videos_bp, customers_bp, rentals_bp
+    app.register_blueprint(videos_bp)
+    app.register_blueprint(customers_bp)
+    app.register_blueprint(rentals_bp)
 
-    return app
+    return app

app/models/customer.py

@@ -1,4 +1,17 @@
 from app import db
 
 class Customer(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(40))
+    postal_code = db.Column(db.String)
+    phone = db.Column(db.String)
+    register_at = db.Column(db.DateTime)
+    videos = db.relationship("Video", secondary="rental", backref="customers")
+
+    def to_dict(self):
+        return {
+            "id": self.id,
+            "name": self.name,
+            "phone": self.phone,
+            "postal_code": self.postal_code
+        }

app/models/rental.py

@@ -1,4 +1,19 @@
 from app import db
+from app.models.video import Video
 
 class Rental(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
+    id = db.Column(db.Integer, primary_key=True)
+    customer_id = db.Column(db.Integer, db.ForeignKey('customer.id'))
+    video_id = db.Column(db.Integer, db.ForeignKey('video.id'))
+    due_date = db.Column(db.DateTime)
+    is_checked_in = db.Column(db.Boolean, default=False)
+
+    def to_dict(self):
+        video = Video.query.get(self.video_id)
+        checked_out = Rental.query.filter_by(video_id = self.video_id, is_checked_in=False).all()
+        return {
+            "video_id": self.video_id,
+            "customer_id": self.customer_id,
+            "videos_checked_out_count": len(checked_out),
+            "available_inventory": video.total_inventory - len(checked_out)
+        }

app/models/video.py

@@ -1,4 +1,20 @@
 from app import db
 
 class Video(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
+    id = db.Column(db.Integer, primary_key=True)
+    title = db.Column(db.String)
+    release_date = db.Column(db.DateTime)
+    total_inventory = db.Column(db.Integer)
+
+    def to_dict(self):
+        return {
+            "id": self.id,
+            "title": self.title,
+            "total_inventory": self.total_inventory
+        }
+
+    def to_dict_using_rentals(self):
+        return {
+            "title": self.title,
+            "release_date": self.release_date
+        }

app/routes/__init__.py

@@ -0,0 +1,3 @@
+from .video_routes import videos_bp
+from .customer_routes import customers_bp
+from .rental_routes import rentals_bp

app/routes/customer_routes.py

@@ -0,0 +1,99 @@
+from app import db
+from app.models.customer import Customer
+from app.models.rental import Rental
+from flask import Blueprint, request, jsonify
+
+customers_bp = Blueprint("customers", __name__, url_prefix="/customers")
+
+@customers_bp.route("", methods=["GET"])
+def get_all_customers():
+    customers = Customer.query.all()
+    response_body = [customer.to_dict() for customer in customers]
+    return jsonify(response_body)
+
+@customers_bp.route("", methods=["POST"])
+def create_customer():
+    request_body = request.get_json()
+
+    error_message = validate_request(request_body)
+    if error_message:
+        return error_message
+
+    new_customer = Customer(
+        name= request_body["name"],
+        phone= request_body["phone"],
+        postal_code= request_body["postal_code"]
+    )
+
+    db.session.add(new_customer)
+    db.session.commit()
+
+    return new_customer.to_dict(), 201
+
+@customers_bp.route("/<customer_id>", methods=["GET"])
+def get_customer(customer_id):
+    try:
+        customer = Customer.query.get(customer_id)
+    except:
+        return jsonify(None), 400
+
+    if customer is None:
+        return {"message": f"Customer {customer_id} was not found"}, 404
+
+    return customer.to_dict()
+
+@customers_bp.route("/<customer_id>", methods=["DELETE"])
+def delete_customer(customer_id):
+    customer = Customer.query.get(customer_id)
+
+    if customer is None:
+        return {"message": f"Customer {customer_id} was not found"}, 404
+
+    db.session.delete(customer)
+    db.session.commit()
+
+    return customer.to_dict()
+
+@customers_bp.route("/<customer_id>", methods=["PUT"])
+def update_customer(customer_id):
+    customer = Customer.query.get(customer_id)
+    request_body = request.get_json()
+
+    if customer is None:
+        return {"message": f"Customer {customer_id} was not found"}, 404
+
+    error_message = validate_request(request_body)
+    if error_message:
+        return error_message
+
+    customer.name = request_body["name"]
+    customer.phone = request_body["phone"]
+    customer.postal_code = request_body["postal_code"]
+
+    db.session.commit()
+
+    return customer.to_dict()
+
+@customers_bp.route("/<customer_id>/rentals", methods=["GET"])
+def get_rentals_by_customer(customer_id):
+    customer = Customer.query.get(customer_id)
+
+    if not customer:
+        return {"message": f"Customer {customer_id} was not found"}, 404
+
+    videos= []
+    for video in customer.videos:
+        rental = Rental.query.filter_by(customer_id = customer.id, video_id = video.id).first()
+        video = video.to_dict_using_rentals()
+        video["due_date"] = rental.due_date
+        videos.append(video)
+
+    return jsonify(videos)
+
+def validate_request(request_body):
+    attributes = ["postal_code", "name", "phone"]
+    for attribute in attributes:
+        if attribute not in request_body:
+            return {"details": f"Request body must include {attribute}."}, 400
+
+

app/routes/rental_routes.py

@@ -0,0 +1,51 @@
+from app import db
+from app.models.rental import Rental
+from app.models.customer import Customer
+from app.models.video import Video
+from flask import Blueprint, request, jsonify
+from datetime import datetime, timedelta
+
+rentals_bp = Blueprint("rentals", __name__, url_prefix="/rentals")
+
+@rentals_bp.route("/check-out", methods=["POST"])
+def check_out_video():
+    request_body = request.get_json()
+
+    try:
+        customer = Customer.query.get_or_404(request_body["customer_id"])
+        video = Video.query.get_or_404(request_body["video_id"])
+    except KeyError:
+        return jsonify(None), 400
+
+    if video.total_inventory - len(video.customers) == 0:
+        return {"message": "Could not perform checkout"}, 400
+
+    new_rental = Rental(
+        customer_id = customer.id,
+        video_id = video.id,
+        due_date = datetime.now() + timedelta(days=7)
+    )
+
+    db.session.add(new_rental)
+    db.session.commit()
+
+    return jsonify(new_rental.to_dict())
+
+@rentals_bp.route("/check-in", methods=["POST"])
+def check_in_video():
+    request_body = request.get_json()
+
+    try:
+        customer = Customer.query.get_or_404(request_body["customer_id"])
+        video = Video.query.get_or_404(request_body["video_id"])
+    except KeyError:
+        return jsonify(None), 400
+
+    rental = Rental.query.filter_by(customer_id = customer.id, video_id = video.id).first()
+    if not rental:
+        return {"message": f"No outstanding rentals for customer {customer.id} and video {video.id}"}, 400
+
+    rental.is_checked_in = True
+    db.session.commit()
+
+    return jsonify(rental.to_dict())

app/routes/video_routes.py

@@ -0,0 +1,99 @@
+from app import db
+from app.models.video import Video
+from app.models.rental import Rental
+from flask import Blueprint, request, jsonify
+
+videos_bp = Blueprint("videos", __name__, url_prefix="/videos")
+
+@videos_bp.route("", methods=["GET"])
+def get_all_videos():
+    videos = Video.query.all()
+    response_body = [video.to_dict() for video in videos]
+    return jsonify(response_body)
+
+@videos_bp.route("", methods=["POST"])
+def create_video():
+    request_body = request.get_json()
+
+    error_message = validate_request(request_body)
+    if error_message:
+        return error_message
+
+    new_video = Video(
+        title= request_body["title"],
+        release_date= request_body["release_date"],
+        total_inventory= request_body["total_inventory"]
+    )
+    db.session.add(new_video)
+    db.session.commit()
+
+    return jsonify(new_video.to_dict()), 201
+
+
+@videos_bp.route("/<video_id>", methods=["GET"])
+def get_video(video_id):
+    try:
+        video = Video.query.get(video_id)
+    except:
+        return jsonify(None), 400
+
+    if video is None:
+        return jsonify({"message": f"Video {video_id} was not found"}), 404
+
+    return video.to_dict()
+
+
+@videos_bp.route("/<video_id>", methods=["DELETE"])
+def delete_video(video_id):
+    video = Video.query.get(video_id)
+
+    if video is None:
+        return {"message": f"Video {video_id} was not found"}, 404
+
+    db.session.delete(video)
+    db.session.commit()
+
+    return video.to_dict()
+
+@videos_bp.route("/<video_id>", methods=["PUT"])
+def update_video(video_id):
+    request_body = request.get_json()
+    video = Video.query.get(video_id)
+
+    if video is None:
+        return {"message": f"Video {video_id} was not found"}, 404
+
+    error_message = validate_request(request_body)
+    if error_message:
+        return error_message
+
+    video.title = request_body["title"]
+    video.total_inventory = request_body["total_inventory"]
+    video.release_date = request_body["release_date"]
+
+    db.session.commit()
+
+    return video.to_dict()
+
+@videos_bp.route("/<video_id>/rentals", methods=["GET"])
+def get_rentals_by_video(video_id):
+    video = Video.query.get(video_id)
+
+    if video is None:
+        return {"message": f"Video {video_id} was not found"}, 404
+
+    customers = []
+    for customer in video.customers:
+        rental = Rental.query.filter_by(customer_id = customer.id, video_id = video_id).first()
+        customer = customer.to_dict()
+        customer["due_date"] = rental.due_date
+        customers.append(customer)
+
+    return jsonify(customers)
+
+def validate_request(request_body):
+    attributes = ["title", "release_date", "total_inventory"]
+
+    for attribute in attributes:
+        if attribute not in request_body:
+            return {"details": f"Request body must include {attribute}."}, 400

migrations/README

@@ -0,0 +1 @@
+Generic single-database configuration.