Ejemplo n.º 1
0
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}'>"
Ejemplo n.º 2
0
class Comment(Model):
    """ Comment Model for storing comment 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_post = Column(db.Integer, db.ForeignKey("post.id"))

    # Comment 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("CommentLike",
                            backref="comment",
                            cascade="all, delete-orphan")
    replies = db.relationship("Reply",
                              backref="comment",
                              cascade="all, delete-orphan")

    def __repr__(self):
        return f"<Comment '{self.id}'>"
Ejemplo n.º 3
0
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}'>"