-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmodel.py
39 lines (34 loc) · 1.65 KB
/
model.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from flask_sqlalchemy import SQLAlchemy
from flask_login import UserMixin, login_user, LoginManager, login_required, current_user, logout_user
from sqlalchemy import ForeignKey
from sqlalchemy.orm import relationship
login_manager = LoginManager()
db = SQLAlchemy()
# CONFIGURE TABLE
class BlogPost(db.Model):
__tablename__ = 'blogpost'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(250), unique=True, nullable=False)
subtitle = db.Column(db.String(250), nullable=False)
date = db.Column(db.String(250), nullable=False)
body = db.Column(db.Text, nullable=False)
author = relationship('User', back_populates='posts')
img_url = db.Column(db.String(250), nullable=False)
author_id = db.Column(db.Integer, db.ForeignKey('users.id'))
comments = relationship('Comment', back_populates='parent_post')
class User(UserMixin, db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(250), nullable=False)
email = db.Column(db.String(250), nullable=False, unique=True)
password = db.Column(db.String(250), nullable=False)
posts = relationship('BlogPost', back_populates='author')
comments = relationship("Comment", back_populates="author")
class Comment(db.Model):
__tablename__ = 'comments'
id = db.Column(db.Integer, primary_key=True)
text = db.Column(db.Text, nullable=False)
author_id = db.Column(db.Integer, db.ForeignKey('users.id'))
author = relationship('User', back_populates='comments')
post_id = db.Column(db.Integer, db.ForeignKey("blogpost.id"))
parent_post = relationship('BlogPost', back_populates='comments')