class Post(Model): """ Post Model for storing post related details """ # Basic details id = Column(db.Integer, primary_key=True) public_id = Column(db.String(15)) owner_id = Column(db.Integer, db.ForeignKey("user.id")) creator_public_id = Column(db.String(15)) # Post content and details content = Column(db.Text) image_file = Column(db.String(40), default=None, nullable=True) status = Column(db.String(10)) created = Column(db.DateTime, default=datetime.utcnow) edited = Column(db.Boolean, default=False) likes = db.relationship("PostLike", backref="post", cascade="all, delete-orphan") comments = db.relationship("Comment", backref="post", cascade="all, delete-orphan") def __repr__(self): return f"<Post '{self.id}'>"
class Role(Model): """ Role Model for storing role related details """ __tablename__ = "role" id = Column(db.Integer, primary_key=True) name = Column(db.String(20), unique=True) description = Column(db.String(50)) def __repr__(self): return f"{self.name} - {self.id}"
class Notification(Model): id = Column(db.Integer, primary_key=True) owner_id = Column(db.Integer, db.ForeignKey("user.id")) # Example actions: 'liked', 'replied', 'commented', etc. action = Column(db.String(10)) timestamp = Column(db.DateTime) read = Column(db.Boolean, default=False) # Object names: post, comment, reply ... object_name = Column(db.String(20)) object_public_id = Column(db.String(15)) def is_read(self, read): self.read = True
class Notification(Model): id = Column(db.Integer, primary_key=True) # The user that committed the action, takes public_id. actor = Column(db.String(15)) # Target owner is the user receiving the notification. target_owner = Column(db.Integer, db.ForeignKey("user.id")) # Example actions: 'liked', 'replied', 'commented', etc. action = Column(db.String(10)) timestamp = Column(db.DateTime) read = Column(db.Boolean, default=False) # Object type: post, comment, reply ... object_type = Column(db.String(20)) object_public_id = Column(db.String(15)) def is_read(self, read): self.read = True
class Reply(Model): """ Reply Model for storing reply related details """ # Basic details id = Column(db.Integer, primary_key=True) public_id = Column(db.String(15)) creator_public_id = Column(db.String(15)) on_comment = Column(db.Integer, db.ForeignKey("comment.id")) # Reply content and details content = Column(db.Text) image_file = Column(db.String(40), default=None, nullable=True) created = Column(db.DateTime, default=datetime.utcnow) edited = Column(db.Boolean, default=False) likes = db.relationship("ReplyLike", backref="reply", cascade="all, delete-orphan") def __repr__(self): return f"<Reply '{self.id}'>"
class User(Model): """ User Model for storing user related details """ # Basic details id = Column(db.Integer, primary_key=True) public_id = Column(db.String(15), unique=True) email = Column(db.String(255), unique=True, nullable=False) username = Column(db.String(15), unique=True) full_name = Column(db.String(50), nullable=True) password_hash = Column(db.String(255)) bio = Column(db.String(150), nullable=True) # Media profile_picture = Column(db.String(40), nullable=True) background_cover = Column(db.String(40), nullable=True) # Relationships notifications = db.relationship("Notification", backref="user") posts = db.relationship("Post", backref="user") # Add Favorites (post) post_likes = db.relationship("PostLike", backref="user") comment_likes = db.relationship("CommentLike", backref="user") reply_likes = db.relationship("ReplyLike", backref="user") # Status joined_date = Column(db.DateTime) roles = db.relationship("Role", secondary=roles_users, backref=db.backref("users"), lazy="dynamic") @property def password(self): raise AttributeError("Password: Write-Only field") @password.setter def password(self, password): self.password_hash = bcrypt.generate_password_hash(password).decode( "utf-8") def check_password(self, password): return bcrypt.check_password_hash(self.password_hash, password) def __repr__(self): return f"<User '{self.username}'>"