Foreign Key Relationships

Author: techwithsam

website/__init__.py

@@ -18,7 +18,7 @@ def create_app():
  app.register_blueprint(views, url_prefix='/')
  app.register_blueprint(auth, url_prefix='/')
 
- from .models import User
+ from .models import User, Post
 
  create_database(app)
  login_manager = LoginManager()

website/models.py

@@ -8,9 +8,10 @@ class User(db.Model, UserMixin):
  email = db.Column(db.String(150), unique=True)
  password = db.Column(db.String(128))
  date_created = db.Column(db.DateTime(timezone=True), default=func.now())
+ posts = db.relationship('Post', backref="user", passive_deletes=True)
 
 class Post(db.Model):
  id = db.Column(db.Integer, primary_key=True)
  text = db.Column(db.Text, nullable=False)
  date_created = db.Column(db.DateTime(timezone=True), default=func.now())
- author = db.Column(db.Integer, db.ForeignKey('user.id'))
+ author = db.Column(db.Integer, db.ForeignKey('user.id', ondelete="CASCADE"), nullable=False)