Scissors: Mai Truong

Procfile

@@ -0,0 +1 @@
+web: gunicorn 'app:create_app()'

app/__init__.py

@@ -1,7 +1,40 @@
 from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+from flask_migrate import Migrate
+import os
+from dotenv import load_dotenv
 
 
+db = SQLAlchemy()
+migrate = Migrate()
+load_dotenv()
+
 def create_app(test_config=None):
     app = Flask(__name__)
+    app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
+
+    if test_config is None:
+        app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get(
+            "SQLALCHEMY_DATABASE_URI")
+    else:
+        app.config["TESTING"] = True
+        app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get(
+            "SQLALCHEMY_TEST_DATABASE_URI")
+
+    # Import models here for Alembic setup
+    from app.models.customer import Customer
+    from app.models.video import Video
+    from app.models.rental import Rental
+
+    db.init_app(app)
+    migrate.init_app(app, db)
+
+    # Register Blueprints here
+    from .routes import video_bp
+    app.register_blueprint(video_bp)
+    from .routes import customer_bp
+    app.register_blueprint(customer_bp)
+    from .routes import rental_bp
+    app.register_blueprint(rental_bp)
 
     return app

app/models/customer.py

@@ -0,0 +1,25 @@
+from flask import current_app
+from app import db
+from sqlalchemy.orm import relationship
+
+
+class Customer(db.Model):
+    __tablename__ = "customer"
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    name = db.Column(db.String)
+    postal_code = db.Column(db.Integer)
+    phone_number = db.Column(db.String)
+    registered_at = db.Column(db.DateTime, nullable=True)
+    videos_checked_out_count = db.Column(db.Integer, default=0)
+    video = relationship("Rental", back_populates="customer")
+
+    def to_json(self):
+        customer_dict = {
+            "id": self.id,
+            "name": self.name,
+            "postal_code": self.postal_code,
+            "phone": self.phone_number,
+            "registered_at": self.registered_at,
+            "videos_checked_out_count": self.videos_checked_out_count
+        }
+        return customer_dict

app/models/rental.py

@@ -0,0 +1,23 @@
+from flask import current_app
+from app import db
+from sqlalchemy.orm import relationship
+from datetime import datetime, timedelta
+
+
+class Rental(db.Model):
+    __tablename__ = "rental"
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    customer_id = db.Column(db.Integer, db.ForeignKey('customer.id'), primary_key=True)
+    video_id = db.Column(db.Integer, db.ForeignKey('video.id'), primary_key=True)
+    due_date = db.Column(db.DateTime, nullable=True)
+    video = relationship("Video", back_populates="customer")
+    customer = relationship("Customer", back_populates="video")
+
+
+    # def to_json(self):
+    #     rental_dict = {
+    #         "customer_id": self.customer_id,
+    #         "video_id": self.video_id,
+    #         "due_date": self.due_date,
+    #     }
+    #     return rental_dict

app/models/video.py

@@ -0,0 +1,22 @@
+from flask import current_app
+from app import db
+from sqlalchemy.orm import relationship
+
+
+class Video(db.Model):
+    __tablename__ = "video"
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    title = db.Column(db.String)
+    release_date = db.Column(db.DateTime, nullable=True)
+    total_inventory = db.Column(db.Integer)
+    available_inventory = db.Column(db.Integer)
+    customer = relationship("Rental", back_populates="video")
+
+    def to_json(self):
+        video_dict = {
+            "video_id": self.id,
+            "title": self.title,
+            "release_date": self.release_date,
+            "total_inventory": self.total_inventory,
+        }
+        return video_dict