Newer
Older
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
from datetime import datetime
from blog import db, login_manager
from werkzeug.security import generate_password_hash, check_password_hash
from flask_login import UserMixin
class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
date = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
title = db.Column(db.Text, nullable=False)
content = db.Column(db.Text, nullable=False)
image_file = db.Column(db.String(40), nullable=False, default='default.jpg')
author_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
def __repr__(self):
return f"Post('{self.date}', '{self.title}', '{self.content}')"
class User(UserMixin,db.Model):
id=db.Column(db.Integer,primary_key=True)
username=db.Column(db.String(15),unique=True,nullable=False)
email=db.Column(db.String(120),unique=True,nullable=False)
password_hash=db.Column(db.String(128))
password=db.Column(db.String(60),nullable=False)
post=db.relationship('Post',backref='user',lazy=True)
comment=db.relationship('Comment',backref='user',lazy=True)
is_admin=db.Column(db.Boolean,nullable=False,default=False)
def __repr__(self):
return f"User('{self.username}','{self.email}')"
@property
def password(self):
raise AttributeError('password is not a readable attribute')
@password.setter
def password(self,password):
self.password_hash=generate_password_hash(password)
def verify_password(self,password):
return check_password_hash(self.password_hash,password)
@login_manager.user_loader
def load_user(user_id):
return User.query.get(int(user_id))
class Comment(db.Model):
id = db.Column(db.Integer,primary_key=True)
date = db.Column(db.DateTime,nullable=False,default=datetime.utcnow)
content = db.Column(db.Text,nullable=False)
parent_id = db.Column(db.Integer,db.ForeignKey('comment.id'),nullable=True)
post_id = db.Column(db.Integer,db.ForeignKey('post.id'),nullable=False)
author_id = db.Column(db.Integer,db.ForeignKey('user.id'),nullable=False)
parent = db.relationship('Comment',backref='comment_parent',remote_side=id,lazy=True)
def __repr__(self):
return f"Post('{self.date}','{self.content}')"
class Rate(db.Model):
# id = db.Column(db.Integer,primary_key=True)
date = db.Column(db.DateTime,nullable=False,default=datetime.utcnow)
rateValue = db.Column(db.Integer,nullable=False)
post_id = db.Column(db.Integer,db.ForeignKey('post.id'),nullable=False, primary_key=True)
author_id = db.Column(db.Integer,db.ForeignKey('user.id'),nullable=False, primary_key=True)
def __repr__(self):
return f"Post('{self.date}','{self.value}')"