Spruce - C16 - Symone & Vange

app/__init__.py

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

app/models/customer.py

@@ -1,4 +1,10 @@
 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)
+    registered_at = db.Column(db.DateTime())
+    videos = db.relationship("Video", secondary="rental", backref="customers")

app/models/rental.py

@@ -1,4 +1,10 @@
 from app import db
 
+# join table 
 class Rental(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
+    id = db.Column(db.Integer, primary_key=True, autoincrement=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()) 
+    video = db.relationship('Video', backref='rentals')
+    customer = db.relationship('Customer', backref='rentals')

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, autoincrement=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,
+            "release_date" : self.release_date,
+            "total_inventory" : self.total_inventory
+        }       
+
+    @classmethod
+    def from_dict(cls, values):
+        return cls(**values)