Ocelots-Jennifer and Cindy

app/__init__.py

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

app/models/customer.py

@@ -1,4 +1,32 @@
 from app import db
+from flask_sqlalchemy import SQLAlchemy
+from sqlalchemy.sql.functions import now
 
 class Customer(db.Model):
     id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String)
+    registered_at = db.Column(db.Date, nullable=True)
+    postal_code = db.Column(db.String)
+    phone = db.Column(db.String)
+
+    def to_dict(self):
+        customer_as_dict = {}
+        customer_as_dict["id"] = self.id
+        customer_as_dict["name"] = self.name
+        #customer_as_dict["registered_at"] = self.registered_at
+        customer_as_dict["postal_code"] = self.postal_code
+        customer_as_dict["phone"] = self.phone        
+
+        return customer_as_dict
+
+    @classmethod
+    def from_dict(cls, customer_data):
+        new_customer = Customer(name = customer_data["name"],
+                        #registered_at=customer_data["registered_at"],
+                        postal_code = customer_data["postal_code"],
+                        phone = customer_data["phone"])
+        return new_customer
+
+    @classmethod
+    def get_id(cls, id):
+        return Customer.query.get(id)

app/models/rental.py

@@ -1,4 +1,22 @@
 from app import db
+from app.models.video import Video
+from app.models.customer import Customer
+from sqlalchemy import func
+from datetime import timedelta
+from .video import Video
+from .customer import Customer
+from datetime import datetime, timedelta, date
 
 class Rental(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
+    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
+    due_date = db.Column(db.Date, nullable=True)
+    #is_checked_out = db.Column(db.Boolean, default=True)
+    video_id = db.Column(db.Integer, db.ForeignKey('video.id'))
+    customer_id = db.Column(db.Integer, db.ForeignKey('customer.id'))
+
+'''
+    @staticmethod
+    def due_date():
+        due_date = datetime.today() + timedelta(days=7)
+        return due_date
+'''

app/models/video.py

@@ -1,4 +1,27 @@
 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)
+    release_date = db.Column(db.Date, nullable=True)
+    total_inventory = db.Column(db.Integer, default=0)
+    customer = db.relationship("Customer", secondary="rental", backref="video")
+
+    def to_dict(self):
+        video_as_dict = {}
+        video_as_dict["id"] = self.id
+        video_as_dict["title"] = self.title
+        video_as_dict["release_date"] = self.release_date
+        video_as_dict["total_inventory"] = self.total_inventory     
+        return video_as_dict
+
+    @classmethod
+    def from_dict(cls, video_data):
+        new_video = Video(title = video_data["title"],
+                        release_date=video_data["release_date"],
+                        total_inventory = video_data['total_inventory'])
+        return new_video
+
+    @classmethod
+    def get_id(cls, id):
+        return Video.query.get(id)