Spruce - Rebecca Z. and Shaina B.

app/__init__.py

@@ -32,5 +32,11 @@ def create_app(test_config=None):
     migrate.init_app(app, db)
 
     #Register Blueprints Here
+    from app.routes.customer_routes import customers_bp
+    from app.routes.video_routes import video_bp
+    from app.routes.rental_routes import rentals_bp
+    app.register_blueprint(customers_bp)
+    app.register_blueprint(video_bp)
+    app.register_blueprint(rentals_bp)
 
     return app

app/models/customer.py

@@ -1,4 +1,19 @@
+from flask import current_app
 from app import db
 
 class Customer(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    name = db.Column(db.String)
+    postal_code = db.Column(db.String)
+    phone = db.Column(db.String)
+    register_at = db.Column(db.DateTime)
+
+    videos = db.relationship("Video", secondary="rentals", backref="checked_out")
+
+    def to_json(self):
+        return {
+            "id": self.id,
+            "name": self.name,
+            "postal_code": self.postal_code,
+            "phone": self.phone
+        }

app/models/helper.py

@@ -0,0 +1,53 @@
+from flask import request
+from app.models.video import Video
+from app.models.rental import Rental
+from app.models.customer import Customer
+
+def invalid_cust_data(request_body):
+    if "name" not in request_body:
+        valid = {"details": "Request body must include name."}
+    elif "phone" not in request_body:
+        valid = {"details": "Request body must include phone."}
+    elif "postal_code" not in request_body:
+        valid = {"details": "Request body must include postal_code."}
+    else:
+        valid = False
+    return valid
+
+def invalid_customer(customer_id):
+    invalid = False
+    if not customer_id.isnumeric():
+        invalid = {"message": f"Invalid customer id"}, 400
+    elif customer_id.isnumeric():
+        if Customer.query.get(customer_id) is None:
+            invalid = {"message": f"Customer {customer_id} was not found"}, 404
+    return invalid
+
+def invalid_video_data(request_body):
+    if "title" not in request_body:
+        valid = {"details": "Request body must include title."}
+    elif "release_date" not in request_body:
+        valid = {"details": "Request body must include release_date."}
+    elif "total_inventory" not in request_body:
+        valid = {"details": "Request body must include total_inventory."}
+    else:
+        valid = False
+    return valid
+
+def invalid_video(video_id):
+    invalid = False
+    if not video_id.isnumeric():
+        invalid = {"message": f"Invalid video id"}, 400
+    elif video_id.isnumeric():
+        if Video.query.get(video_id) is None:
+            invalid = {"message": f"Video {video_id} was not found"}, 404
+    return invalid
+
+def invalid_rental_data(request_body):
+    if "customer_id" not in request_body:
+        invalid = {"details": "Request body must include customer_id."}
+    elif "video_id" not in request_body:
+        invalid = {"details": "Request body must include video_id."}
+    else:
+        invalid = False
+    return invalid

app/models/rental.py

@@ -1,4 +1,24 @@
+from sqlalchemy.orm import backref
 from app import db
+from app.models.video import Video
+from app.models.customer import Customer
 
 class Rental(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
+    __tablename__ = "rentals"
+    id = db.Column(db.Integer, primary_key=True, autoincrement = True, nullable=False)
+    customer_id = db.Column(db.Integer, db.ForeignKey('customer.id'), primary_key=True, nullable=False)
+    video_id = db.Column(db.Integer, db.ForeignKey('video.id'), primary_key=True, nullable=False)
+    due_date = db.Column(db.DateTime)
+    checked_out = db.Column(db.Boolean, default = False)
+
+    def rental_by_title(self):
+        video = Video.query.get(self.video_id)
+        return{
+            "title": video.title
+        }
+
+    def cust_by_name(self):
+        customer = Customer.query.get(self.customer_id)
+        return {
+            "name": customer.name
+        }

app/models/video.py

@@ -1,4 +1,17 @@
 from app import db
 
 class Video(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    title = db.Column(db.String, nullable=False)
+    release_date = db.Column(db.DateTime, nullable=False)
+    total_inventory = db.Column(db.Integer, nullable=False)
+
+    # customers = db.relationship("Customer", secondary="rentals")
+
+    def video_dict(self):
+        return {
+            "id": self.id,
+            "title": self.title,
+            "release_date": self.release_date,
+            "total_inventory": self.total_inventory
+        }

app/routes/customer_routes.py

@@ -0,0 +1,74 @@
+from app import db
+from flask import Blueprint, jsonify, request
+from app.models.customer import Customer
+from app.models.rental import Rental
+from app.models.video import Video
+from app.models.helper import invalid_cust_data, invalid_customer
+
+customers_bp = Blueprint("customers", __name__, url_prefix="/customers")
+
+@customers_bp.route("", methods=["POST"])
+def post_customer():
+    request_body = request.get_json()
+    invalid_response = invalid_cust_data(request_body)
+    if not invalid_response:
+        new_customer = Customer(
+            name=request_body["name"],
+            postal_code=request_body["postal_code"],
+            phone=request_body["phone"]
+        )
+        db.session.add(new_customer)
+        db.session.commit()
+        return jsonify(new_customer.to_json()), 201
+    return jsonify(invalid_response), 400
+
+@customers_bp.route("", methods=["GET"])
+def get_customers():
+    customers = Customer.query.all()
+    customer_list = [customer.to_json() for customer in customers]
+    return jsonify(customer_list), 200
+
+
+@customers_bp.route("/<customer_id>", methods=["GET"])
+def get_one_customer(customer_id):
+    invalid_response = invalid_customer(customer_id)
+    if invalid_response:
+        return invalid_response
+    one_customer = Customer.query.get(customer_id)
+    return jsonify(one_customer.to_json()), 200
+
+@customers_bp.route("/<customer_id>/rentals", methods=["GET"])
+def get_cust_rental(customer_id):
+    if Customer.query.get(customer_id) is None:
+        return jsonify({"message": f"Customer {customer_id} was not found"}), 404
+    all_rentals = Rental.query.filter_by(customer_id = customer_id) # get rental by cust_id
+    response = []
+    for rental in all_rentals: # to access formatted rental information
+        response.append(rental.rental_by_title())
+    return jsonify(response), 200
+
+@customers_bp.route("/<customer_id>", methods=["PUT"])
+def update_customer(customer_id):
+    one_customer = Customer.query.get(customer_id)
+    invalid_cust = invalid_customer(customer_id)
+    if invalid_cust:
+        return invalid_cust
+    request_body = request.get_json()
+    invalid_response = invalid_cust_data(request_body)
+    if invalid_response:
+        return jsonify(invalid_response), 400
+    one_customer.name = request_body["name"]
+    one_customer.phone = request_body["phone"]
+    one_customer.postal_code = request_body["postal_code"]
+    db.session.commit()
+    return jsonify(one_customer.to_json()), 200
+
+@customers_bp.route("/<customer_id>", methods=["DELETE"])
+def delete_customer(customer_id):
+    one_customer = Customer.query.get(customer_id)
+    invalid_cust = invalid_customer(customer_id)
+    if invalid_cust:
+        return invalid_cust
+    db.session.delete(one_customer)
+    db.session.commit()
+    return jsonify(one_customer.to_json()), 200

app/routes/rental_routes.py

@@ -0,0 +1,79 @@
+from app import db
+from flask import Blueprint, jsonify, request
+from app.models.rental import Rental
+from app.models.customer import Customer
+from app.models.video import Video
+from datetime import timedelta, datetime
+from app.models.helper import invalid_rental_data
+
+rentals_bp = Blueprint("rentals", __name__, url_prefix="/rentals")
+
+@rentals_bp.route("/check-out", methods=["POST"])
+def post_rental():
+    request_body = request.get_json()
+    invalid_request = invalid_rental_data(request_body)
+    if invalid_request:
+        return jsonify(invalid_request), 400
+
+    one_video = Video.query.get(request_body["video_id"])
+    one_customer = Customer.query.get(request_body["customer_id"])
+    if one_video is None:
+        return jsonify({"Bad Request": f"Video not found."}), 404
+    elif one_customer is None:
+        return jsonify({"Bad Request": f"Customer not found."}), 404
+    video_current_checked_out = Rental.query.filter_by(checked_out=True, video_id=one_video.id).count()
+    available_inventory = one_video.total_inventory - video_current_checked_out
+    if available_inventory == 0:
+        return jsonify({"message": "Could not perform checkout"}), 400
+    new_rental = Rental(
+        customer_id=one_customer.id,
+        video_id = one_video.id,
+        due_date = (datetime.now(tz=None) + timedelta(days=7)),
+        checked_out = True
+    )
+    db.session.add(new_rental)
+    db.session.commit()
+    cust_current_checked_out = Rental.query.filter_by(customer_id=one_customer.id, checked_out=True).count()
+    available_inventory -=1
+    response_body = {
+        "customer_id": one_customer.id,
+            "video_id": one_video.id,
+            "due_date": new_rental.due_date,
+            "videos_checked_out_count": cust_current_checked_out,
+            "available_inventory": available_inventory
+    }
+    return jsonify(response_body), 200
+
+@rentals_bp.route("/check-in", methods=["POST"])
+def post_rental_return():
+    request_body = request.get_json()
+    invalid_request = invalid_rental_data(request_body)
+    if invalid_request:
+        return jsonify(invalid_request), 400
+
+    one_video = Video.query.get(request_body["video_id"])
+    one_customer = Customer.query.get(request_body["customer_id"])
+    if one_video is None:
+        return jsonify({"Bad Request": f"Video not found."}), 404
+    elif one_customer is None:
+        return jsonify({"Bad Request": f"Customer not found."}), 404
+    video_current_checked_in = Rental.query.filter_by(checked_out=True).count()
+    video_checked_out = Rental.query.filter_by(checked_out=False).count()
+    if video_current_checked_in == 0:
+        return jsonify({"message": f"No outstanding rentals for customer {one_customer.id} and video {one_video.id}"}), 400
+    returned_rental = Rental(
+        customer_id = one_customer.id,
+        video_id = one_video.id,
+        checked_out = False
+    )
+    db.session.add(returned_rental)
+    db.session.commit()
+    available_inventory = one_video.total_inventory - video_checked_out
+    response_body = {
+        "customer_id": one_customer.id,
+        "video_id": one_video.id,
+        "videos_checked_out_count": video_checked_out,
+        "available_inventory": available_inventory
+            }
+    return jsonify(response_body), 200
+

app/routes/video_routes.py

@@ -0,0 +1,86 @@
+from app import db
+from flask import Blueprint, jsonify, request
+from app.models.helper import invalid_video, invalid_video_data
+from app.models.video import Video
+from app.models.rental import Rental
+
+video_bp = Blueprint("video_bp", __name__, url_prefix="/videos")
+
+
+@video_bp.route("", methods=["GET"])
+def get_videos():
+    videos = Video.query.all()
+    video_list = [video.video_dict() for video in videos]
+    return jsonify(video_list), 200
+
+
+@video_bp.route("/<video_id>", methods=["GET"])
+def get_one_video(video_id):
+    invalid_response = invalid_video(video_id)
+    if invalid_response:
+        return invalid_response
+
+    one_video = Video.query.get(video_id)
+    return jsonify(one_video.video_dict()), 200
+
+
+@video_bp.route("/<video_id>/rentals", methods=["GET"])
+def get_vid_rental(video_id):
+    if Video.query.get(video_id) is None:
+        return jsonify({"message": f"Video {video_id} was not found"}), 404
+    all_videos = Rental.query.filter_by(id=video_id)  # get rental by cust_id
+    response = []
+    for video in all_videos:  # to access formatted rental information
+        response.append(video.cust_by_name())
+    return jsonify(response), 200
+
+
+@video_bp.route("", methods=["POST"])
+def post_video():
+    request_body = request.get_json()
+    invalid_response = invalid_video_data(request_body)
+    if not invalid_response:
+        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.video_dict()), 201
+
+    return jsonify(invalid_response), 400
+
+
+@video_bp.route("/<video_id>", methods=["PUT"])
+def update_video(video_id):
+    one_video = Video.query.get(video_id)
+    invalid_vid = invalid_video(video_id)
+    if invalid_vid:
+        return invalid_vid
+
+    request_body = request.get_json()
+    invalid_response = invalid_video_data(request_body)
+
+    if invalid_response:
+        return jsonify(invalid_response), 400
+
+    one_video.title = request_body["title"]
+    one_video.release_date = request_body["release_date"]
+    one_video.total_inventory = request_body["total_inventory"]
+
+    db.session.commit()
+    return jsonify(one_video.video_dict()), 200
+
+
+@video_bp.route("/<video_id>", methods=["DELETE"])
+def delete_video(video_id):
+    one_video = Video.query.get(video_id)
+    invalid_vid = invalid_video(video_id)
+    if invalid_vid:
+        return invalid_vid
+
+    db.session.delete(one_video)
+    db.session.commit()
+    return jsonify(one_video.video_dict()), 200

migrations/README

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